SIP on filling influxdb

rescuetime/__init__.py

@@ -77,3 +77,20 @@ def get_groups(gap=timedelta(hours=3)):
     data = get_rescuetime()
     return group_by_cmp(data, lambda a, b: (b.dt - a.dt) <= gap, dist=1)
 
+
+
+def fill_influxdb():
+    from influxdb import InfluxDBClient # type: ignore
+    client = InfluxDBClient()
+    # client.delete_series(database='lastfm', measurement='phone')
+    db = 'test'
+    client.drop_database(db)
+    client.create_database(db)
+    jsons = [{
+        "measurement": 'phone',
+        "tags": {},
+        "time": str(e.dt),
+        "fields": {"name": e.activity},
+    } for e in get_rescuetime()]
+    client.write_points(jsons, database=db) # TODO??
+

rescuetime/__main__.py

@@ -1,14 +1,15 @@
 from kython.klogging import setup_logzero
 
-from . import get_logger, get_groups, get_rescuetime
+from . import get_logger, get_groups, get_rescuetime, fill_influxdb
 
 logger = get_logger()
 setup_logzero(logger)
 
 # for gr in get_groups():
 #     print(f"{gr[0].dt}--{gr[-1].dt}")
-for e in get_rescuetime(latest=2):
-    print(e)
+# for e in get_rescuetime(latest=2):
+#     print(e)
+fill_influxdb()
 
 # TODO merged db?
 # TODO ok, it summarises my sleep intervals pretty well. I guess should adjust it for the fact I don't sleep during the day, and it would be ok!