feat: add JSON support to export

src/hamster-cli.py

@@ -237,7 +237,7 @@ def assist(self, *args):
         if assist_command == "start":
             hamster_client._activities(sys.argv[-1])
         elif assist_command == "export":
-            formats = "html tsv xml ical".split()
+            formats = "html tsv xml ical json".split()
             chosen = sys.argv[-1]
             formats = [f for f in formats if not chosen or f.startswith(chosen)]
             print("\n".join(formats))
@@ -422,7 +422,7 @@ def version(self):
     * list [start-date [end-date]]: List activities
     * search [terms] [start-date [end-date]]: List activities matching a search
       term
-    * export [html|tsv|ical|xml] [start-date [end-date]]: Export activities with
+    * export [html|tsv|ical|xml|json] [start-date [end-date]]: Export activities with
       the specified format
     * current: Print current activity
     * activities: List all the activities names, one per line.

src/hamster/reports.py

@@ -51,6 +51,8 @@ def simple(facts, start_date, end_date, format, path = None):
 
     if format == "tsv":
         writer = TSVWriter(report_path)
+    if format == "json":
+        writer = JSONWriter(report_path)
     elif format == "xml":
         writer = XMLWriter(report_path)
     elif format == "ical":
@@ -323,3 +325,24 @@ def _finish(self, facts):
         self.file.write(Template(self.main_template).safe_substitute(data))
 
         return
+
+class JSONWriter(ReportWriter):
+    def __init__(self, path):
+        ReportWriter.__init__(self, path)
+        self.activity_list = []
+
+    def _write_fact(self, fact):
+        self.activity_list.append(
+            {
+                "name": fact.activity,
+                "start_time": str(fact.start_time),
+                "end_time": str(fact.end_time),
+                "duration_minutes": stuff.duration_minutes(fact.delta),
+                "category": fact.category,
+                "description": fact.description,
+                "tags": ": ".join(fact.tags),
+            }
+        )
+
+    def _finish(self, facts):
+        self.file.write(json.dumps(self.activity_list))