From 05928b124454e6ceefb4c47520b671a5e08453dd Mon Sep 17 00:00:00 2001
From: 3wc <3wc@doesthisthing.work>
Date: Wed, 1 Nov 2023 19:32:00 +0000
Subject: [PATCH] Add (temporary?) scripts

---
 scripts/apitest.py | 30 ++++++++++++++++++
 scripts/upload.py  | 79 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 109 insertions(+)
 create mode 100644 scripts/apitest.py
 create mode 100644 scripts/upload.py

diff --git a/scripts/apitest.py b/scripts/apitest.py
new file mode 100644
index 0000000..1ffd641
--- /dev/null
+++ b/scripts/apitest.py
@@ -0,0 +1,30 @@
+import os
+import sys
+sys.path.append(os.path.join(os.path.dirname(__file__), ".."))
+
+from datetime import datetime, timedelta
+import logging
+
+from hamstertools.kimai import *
+
+logging.basicConfig()
+logging.getLogger().setLevel(logging.DEBUG)
+requests_log = logging.getLogger("requests.packages.urllib3")
+requests_log.setLevel(logging.DEBUG)
+requests_log.propagate = True
+
+api = KimaiAPI()
+
+# print(Timesheet.list(api))
+
+t = Timesheet(api,
+    activity=Activity.get_by_id(api, 613),
+    project=Project.get_by_id(api, 233),
+    begin=datetime.now() - timedelta(minutes=10),
+    end=datetime.now(),
+)
+
+# r = t.upload()
+# from pdb import set_trace; set_trace()
+
+print(Timesheet.get_by_id(api, 30683))
diff --git a/scripts/upload.py b/scripts/upload.py
new file mode 100644
index 0000000..4431632
--- /dev/null
+++ b/scripts/upload.py
@@ -0,0 +1,79 @@
+import os
+import sys
+sys.path.append(os.path.join(os.path.dirname(__file__), ".."))
+
+from datetime import datetime
+
+from peewee import JOIN
+
+from hamstertools.db import *
+from hamstertools.kimai import *
+
+api = KimaiAPI()
+
+DATE_FROM='2023-10-01'
+DATE_TO='2023-11-01'
+SEARCH='auto'
+
+facts = HamsterFact.select(
+    HamsterFact,
+    HamsterActivity,
+    HamsterCategory
+).join(
+    HamsterActivity, JOIN.LEFT_OUTER
+).join(
+    HamsterCategory, JOIN.LEFT_OUTER
+).where(
+    (HamsterFact.start_time > datetime.strptime(DATE_FROM, "%Y-%m-%d"))
+    & (HamsterFact.start_time <  datetime.strptime(DATE_TO, "%Y-%m-%d"))
+    & HamsterCategory.name.contains(SEARCH)
+)
+
+has_errors = False
+
+# check data
+for f in facts:
+    mappings = f.activity.mappings
+    if len(mappings) == 0:
+        print(f'fact {f.id}: @{f.activity.category.id} {f.activity.category.name} » @{f.activity.id} {f.activity.name} has no mapping')
+        has_errors = True
+        continue
+    if len(mappings) > 1:
+        print(f'fact {f.id}: activity @{f.activity.id} {f.activity.name} has multiple mappings')
+        has_errors = True
+        continue
+    if mappings[0].kimai_activity.project is None and not mappings[0].kimai_project.allow_global_activities:
+        print(f'fact {f.id}: project @{mappings[0].kimai_project.id} {mappings[0].kimai_project.name} does not allow global activity {mappings[0].kimai_activity}')
+        has_errors = True
+        continue
+
+# if has_errors:
+#     sys.exit(1)
+
+# upload data
+for f in facts:
+    try:
+        mapping = f.activity.mappings[0]
+    except IndexError:
+        print(f"no mapping, skipping {f.id} ({f.activity.category.name} » {f.activity.name})")
+        continue
+    t = Timesheet(api,
+        activity=mapping.kimai_activity,
+        project=mapping.kimai_project,
+        begin=f.start_time,
+        end=f.end_time,
+        description=f.description if f.description != '' else mapping.kimai_description,
+        # tags=f.tags if f.tags != '' else mapping.kimai_tags
+    )
+    r = t.upload().json()
+    if r.get("code", 200) != 200:
+        print(r)
+        print(f"{f.id} ({f.activity.category.name} » {f.activity.name})")
+        from pdb import set_trace; set_trace()
+
+    else:
+        HamsterFactKimaiImport.create(
+            hamster_fact=f,
+            kimai_id=r['id']
+        ).save()
+        print(f'Created Kimai timesheet {r["id"]}')