jarvis

#!/usr/bin/python

from PyPDF2 import PdfFileWriter, PdfFileReader
from os.path import basename
import webbrowser
import os
import sys
import shutil


def seperate_pdf(file):
    print("Seperating pdf command triggered")

    print("Initializing new directory...")
    parent = os.getcwd()
    directory = "pdf-output"
    path = os.path.join(parent, directory)

    if os.path.exists(path):
        print("Directory '{}' already exists, deleting..".format(directory))
        shutil.rmtree(path)

    os.mkdir(path)
    print("Directory '{}' created".format(directory))

    filename = file[:-4]

    print("Seperating {}...".format(file))
    input = PdfFileReader(open(file, "rb"))

    for i in range(input.numPages):
        output = PdfFileWriter()
        output.addPage(input.getPage(i))
        with open("{}/{}-p{}.pdf".format(path, filename, i), "wb") as outputStream:
            output.write(outputStream)
    print("Done.")


def pdf(arguments):
    action = arguments[0]

    if action == "split":
        if len(arguments) != 2:
            raise Exception(
                "Invalid arguments. Usage: jarvis pdf split [filename.pdf]")

        seperate_pdf(arguments[1])
    else:
        raise Exception("Invalid command. Possible commands: jarvis pdf split")


def github(arguments):
    action = arguments[0]

    if action == "open":
        print("In progress")
        parent = basename(os.getcwd())
        webbrowser.open("http://github.com/Angela-UW/{}".format(parent))


if __name__ == "__main__":
    num_args = len(sys.argv)

    # Get command
    command = sys.argv[1]

    if command == "pdf":
        pdf(sys.argv[2:])
    elif command == "git":
        github(sys.argv[2:])
    else:
        print("Invalid command. Please try again")