#!/usr/bin/python3

import argparse
from lib.DataStore import DataStore
from lib.TimeSlot import TimeSlot
from lib.ui import convert_to_table


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Time tracker')
    parser.add_argument('command', type=str, help='Manage time slots.', choices=['start', 'stop', 'add', 'ps', 'ls'])
    parser.add_argument('-b', '--begin', type=str, help="The start time of a time slot.")
    parser.add_argument('-e', '--end', type=str, help="The end time of a time slot.")

    args = parser.parse_args()

    ds = DataStore("data.json")

    if args.command == "ls":
        print(convert_to_table(ds.get_all_time_slots()))