Merge branch 'api-conf'

hamstertools/kimaiapi.py

@@ -1,8 +1,6 @@
 from datetime import datetime
 import pdb
 import requests
-import requests_cache
-import os
 
 from dataclasses import dataclass, field
 
@@ -12,16 +10,12 @@ class NotFound(Exception):
 
 
 class KimaiAPI(object):
-    # temporary hardcoded config
-    KIMAI_API_URL = "https://kimai.autonomic.zone/api"
-    # KIMAI_API_URL = "https://kimaitest.autonomic.zone/api"
-
-    KIMAI_USERNAME = "3wordchant"
-    KIMAI_API_KEY = os.environ["KIMAI_API_KEY"]
-
-    auth_headers = {"X-AUTH-USER": KIMAI_USERNAME, "X-AUTH-TOKEN": KIMAI_API_KEY}
-
-    def __init__(self):
+    def __init__(self, username=None, api_key=None, api_url=None):
+        self.auth_headers = {"X-AUTH-USER": username, "X-AUTH-TOKEN": api_key}
+        self.kimai_api_url = api_url
+        # NOTE: Uncomment the following line to enable requests_cache, which can make development a *lot* faster
+        # TODO: Add a config option or something for this 
+        # import requests_cache
         # requests_cache.install_cache("kimai", backend="sqlite", expire_after=1800)
         self.customers_json = self.get("customers", {"visible": 3})
         self.projects_json = self.get("projects", {"visible": 3, "ignoreDates": 1})
@@ -31,7 +25,7 @@ class KimaiAPI(object):
 
     def get(self, endpoint, params=None):
         result = requests.get(
-            f"{self.KIMAI_API_URL}/{endpoint}", params=params, headers=self.auth_headers
+            f"{self.kimai_api_url}/{endpoint}", params=params, headers=self.auth_headers
         ).json()
         try:
             if result["code"] != 200:
@@ -42,7 +36,7 @@ class KimaiAPI(object):
 
     def post(self, endpoint, data):
         return requests.post(
-            f"{self.KIMAI_API_URL}/{endpoint}", json=data, headers=self.auth_headers
+            f"{self.kimai_api_url}/{endpoint}", json=data, headers=self.auth_headers
         )