kattis.py

#!/usr/bin/env python3
import traceback
import sys
from commands.contest import contestCommand
from commands.debug import debugCommand
from helpers.webutils import submitError
from helpers.cli import yes
from commands.startup import startupCommand
from commands.work import workCommand
from commands.archive import archiveCommand
from commands.get import getCommand
from commands.submit import submitCommand
from commands.test import testCommand
from commands.watch import watchCommand
from commands.list import listCommand
from commands.read import readCommand
from commands.unarchive import unarchiveCommand
from commands.config import configCommand
from helpers.parser import parse

execCommand = {
    "archive":   archiveCommand,
    "config":    configCommand,
    "contest":   contestCommand,
    "debug":     debugCommand,
    "get":       getCommand,
    "list":      listCommand,
    "read":      readCommand,
    "startup":   startupCommand,
    "submit":    submitCommand,
    "test":      testCommand,
    "unarchive": unarchiveCommand,
    "watch":     watchCommand,
    "work":      workCommand,
}


def main():
    try:
        data = parse(sys.argv[1:])
        command = data.get("command")
        if command in execCommand:
            execCommand[command](data)
        else:
            print(
                'Welcome to the Kat CLI for Kattis. To use, write "kattis [COMMAND]" with one of the following commands:')
            print(", ".join(execCommand))
    except (KeyboardInterrupt):
        print("\nInterrupted.")
    except (Exception) as error:
        print()
        print(*traceback.format_exception(None, error, error.__traceback__))
        print(f"Error occurred:\n {error}\n")
        print("The program ran into a problem while running, do you want to create an issue on github?")
        if yes():
            submitError(error)


if __name__ == '__main__':
    main()