__main__.py

#!venv/bin/python3


from timekeeper import TimeKeeper
import argparse
import os
import sys
import time
import threading


def get_args():
    parser = argparse.ArgumentParser("A timekeeper")
    parser.add_argument("-p", "--project", help="Which project to track")
    parser.add_argument("-t", "--task", help="Which task to track", nargs="+")
    parser.add_argument("-l", "--list", help="Show list --> projects || tasks")
    parser.add_argument("-c", "--count", help="Count of lines", type=int)
    parser.add_argument("-r", "--remove", help="timekeeper -r <task|tag> <project> <task_name|tag_name>", nargs="+")
    parser.add_argument("-a", "--add",
                        help="timekeeper -a <task|tag> <project_name> <tagname|taskname> "
                             "<start_time> <end_time>", nargs="+")
    parser.add_argument("-fu", "--force_upload", action="store_true")
    parser.add_argument("-fd", "--force_download", action="store_true")
    parser.add_argument("-s", "--script", help="Run script from folder", nargs="+")
    options = parser.parse_args()
    return vars(options)


def input_thread_fkt(w):
    input()
    w.append(None)


def main():
    w = []
    options = get_args()
    arg_project_name = options.get("project")
    if isinstance(arg_project_name, list):
        arg_project_name = ' '.join(arg_project_name)
    arg_task_name = options.get("task")
    if isinstance(arg_task_name, list):
        arg_task_name = ' '.join(arg_task_name)
    show_list = options.get("list")
    count = options.get("count")
    remove = options.get("remove")
    add = options.get("add")
    force_upload = options.get("force_upload")
    force_download = options.get("force_download")
    script = options.get("script")
    timekeeper = TimeKeeper()

    if script:
        command = os.path.join(os.path.dirname(os.path.realpath(__file__)), "scripts", script[0])
        """add all arguments"""
        for current_script in script:
            if current_script != script[0]:
                command += " " + current_script
        os.system(command)
    elif force_download:
        timekeeper.json_helper.server_sync_helper.download_from_server()
    elif force_upload:
        timekeeper.save()
    elif add:
        timekeeper.add(add)
    elif remove:
        timekeeper.remove(remove)
    elif show_list:
        if count is None:
            count = 15
        if str(show_list).startswith("p"):
            timekeeper.terminal_ui_helper.print_project_table()
        elif str(show_list).startswith("t"):
            timekeeper.terminal_ui_helper.print_task_table(lines=count)
    else:
        if arg_task_name is None or arg_project_name is None:
            exit("Project and taskname are required")

        """Always sync with server, before new task starts"""
        timekeeper.json_helper.server_sync_helper.download_from_server()
        timekeeper = TimeKeeper()
        thread = threading.Thread(target=timekeeper.start, args=(arg_project_name, arg_task_name))
        thread.setDaemon(True)
        thread.start()

        input_thread = threading.Thread(target=input_thread_fkt, args=(w, ))
        input_thread.setDaemon(True)
        input_thread.start()
        while True:
            if timekeeper.current_project is not None:
                sys.stdout.write(timekeeper.current_project.current_task.get_run_time())
                sys.stdout.flush()
                time.sleep(1)
                if w:
                    input_thread.join()
                    break
        timekeeper.stop()


if __name__ == '__main__':
    main()