removing seconds from timestamp; wider column for name when printing ls/ ps

main.py

@@ -4,6 +4,7 @@ import argparse
 from lib.DataStore import DataStore
 from lib.TimeSlot import TimeSlot
 from lib.ui import convert_to_table
+from datetime import datetime
 
 
 def create_parser():
@@ -12,14 +13,15 @@ def create_parser():
     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 all events.")
+    _ = subparsers.add_parser("ls", help="List all events.")
 
     # Create the parser for the "ps" command
-    parser_ps = subparsers.add_parser("ps", help="Show the currently running event.")
+    _ = 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("-b", "--begin", help="Begin datetime")
+    parser_add.add_argument("name", help="Name of the event")
+    parser_add.add_argument("-s", "--start", help="Start datetime (default now)")
     parser_add.add_argument("-e", "--end", help="End datetime")
 
     # Create the parser for the "start" command
@@ -27,7 +29,7 @@ def create_parser():
     parser_start.add_argument("name", help="Name of the event")
 
     # Create the parser for the "end" command
-    parser_end = subparsers.add_parser("end", help="Ending the current event.")
+    _ = subparsers.add_parser("end", help="Ending the current event.")
 
     return parser
 
@@ -62,3 +64,22 @@ if __name__ == "__main__":
             ds.write_update()
             print(f"Ended event {running[0].name}.")
 
+    elif args.command == "ps":
+        time_slots = ds.get_all_time_slots()
+        running = list(filter(lambda x: x.end is None, time_slots))
+        print(convert_to_table(running))
+
+    elif args.command == "add":
+        time_slots = ds.get_all_time_slots()
+        running = list(filter(lambda x: x.end is None, time_slots))
+        if len(running) > 0 and args.end is None:
+            print("An event is already running.")
+        else:
+            ts = TimeSlot(args.name)
+            if args.start is not None:
+                ts.start = datetime.strptime(args.start, '%d.%m.%y %H:%M')
+            if args.end is not None:
+                ts.end = datetime.strptime(args.end, '%d.%m.%y %H:%M')
+            ds.add_time_slot(ts)
+            ds.write_update()
+            print("The event was added.")