From 7b2ff3627c51f5375982b632c5e587e4ccc0084d Mon Sep 17 00:00:00 2001
From: Dima Gerasimov <karlicoss@gmail.com>
Date: Sat, 6 Apr 2019 18:18:50 +0100
Subject: [PATCH] SIP on filling influxdb

---
 rescuetime/__init__.py | 17 +++++++++++++++++
 rescuetime/__main__.py |  7 ++++---
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/rescuetime/__init__.py b/rescuetime/__init__.py
index d40c83b..3f81d1a 100644
--- a/rescuetime/__init__.py
+++ b/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??
+
diff --git a/rescuetime/__main__.py b/rescuetime/__main__.py
index a50fc32..5568d64 100644
--- a/rescuetime/__main__.py
+++ b/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!