tt/util/parser.py

import argparse


def create_parser():
    # Create the top-level parser
    parser = argparse.ArgumentParser(description="Time tracker.")
    subparsers = parser.add_subparsers(dest="command", help="Sub-command help")

    # Create the parser for the "ls" command
    parser_ls = subparsers.add_parser("ls", help="List events.")
    parser_ls.add_argument("-s", "--span", choices=['d', 'w', 'lw'], help="Display only envent in a certain time span (d = current day; w = current week; lw = last week)")

    # Create the parser for the "ps" command
    _ = subparsers.add_parser("ps", help="Show the currently running event.")

    # Create the parser for the "add" command
    parser_add = subparsers.add_parser("add", help="Adding a new event.")
    parser_add.add_argument("name", help="Name of the event")
    parser_add.add_argument("-s", "--start", required=True, help="Start datetime (default now)")
    parser_add.add_argument("-e", "--end", required=True, help="End datetime")

    # Create the parser for the "start" command
    parser_start = subparsers.add_parser("start", help="Starting a new event.")
    parser_start.add_argument("name", help="Name of the event")
    parser_start.add_argument("-s", "--start", help="Start datetime (default now)")

    # Create the parser for the "end" command
    parser_end = subparsers.add_parser("end", help="Ending the current event.")
    parser_end.add_argument("-e", "--end", help="End datetime")

    # Create the parser for the "acc" command
    parser_end = subparsers.add_parser("acc", help="Accumulate the duration of events.")
    parser_end.add_argument("query", help="The regex matching for the name of the event.")
    parser_end.add_argument("-s", "--span", choices=['d', 'w', 'lw'], help="Display only envent in a certain time span (d = current day; w = current week; lw = last week)")

    return parser