113 lines
3.2 KiB
Python
113 lines
3.2 KiB
Python
import requests
|
|
import requests_cache
|
|
import os
|
|
|
|
|
|
class NotFound(Exception):
|
|
pass
|
|
|
|
|
|
class KimaiAPI(object):
|
|
# temporary hardcoded config
|
|
KIMAI_API_URL = "https://kimai.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):
|
|
requests_cache.install_cache("kimai", backend="sqlite", expire_after=1800)
|
|
self.customers_json = requests.get(
|
|
f"{self.KIMAI_API_URL}/customers?visible=3", headers=self.auth_headers
|
|
).json()
|
|
self.projects_json = requests.get(
|
|
f"{self.KIMAI_API_URL}/projects?visible=3", headers=self.auth_headers
|
|
).json()
|
|
self.activities_json = requests.get(
|
|
f"{self.KIMAI_API_URL}/activities?visible=3", headers=self.auth_headers
|
|
).json()
|
|
|
|
|
|
class BaseAPI(object):
|
|
def __init__(self, api, **kwargs):
|
|
for key, value in kwargs.items():
|
|
setattr(self, key, value)
|
|
|
|
|
|
class Customer(BaseAPI):
|
|
def __init__(self, api, id, name):
|
|
super().__init__(api, id=id, name=name)
|
|
|
|
@staticmethod
|
|
def list(api):
|
|
return [Customer(api, c["id"], c["name"]) for c in api.customers_json]
|
|
|
|
@staticmethod
|
|
def get_by_id(api, id):
|
|
for value in api.customers_json:
|
|
if value["id"] == id:
|
|
return Customer(api, value["id"], value["name"])
|
|
raise NotFound()
|
|
|
|
def __repr__(self):
|
|
return f"Customer (id={self.id}, name={self.name})"
|
|
|
|
|
|
class Project(BaseAPI):
|
|
def __init__(self, api, id, name, customer):
|
|
super().__init__(api, id=id, name=name, customer=customer)
|
|
|
|
@staticmethod
|
|
def list(api):
|
|
return [
|
|
Project(api, p["id"], p["name"], Customer.get_by_id(api, p["customer"]))
|
|
for p in api.projects_json
|
|
]
|
|
|
|
@staticmethod
|
|
def get_by_id(api, id, none=False):
|
|
for value in api.projects_json:
|
|
if value["id"] == id:
|
|
return Project(
|
|
api,
|
|
value["id"],
|
|
value["name"],
|
|
Customer.get_by_id(api, value["customer"]),
|
|
)
|
|
if not none:
|
|
raise NotFound()
|
|
|
|
def __repr__(self):
|
|
return f"Project (id={self.id}, name={self.name}, customer={self.customer})"
|
|
|
|
|
|
class Activity(BaseAPI):
|
|
def __init__(self, api, id, name, project):
|
|
super().__init__(api, id=id, name=name, project=project)
|
|
|
|
@staticmethod
|
|
def list(api):
|
|
return [
|
|
Activity(
|
|
api, a["id"], a["name"], Project.get_by_id(api, a["project"], none=True)
|
|
)
|
|
for a in api.activities_json
|
|
]
|
|
|
|
@staticmethod
|
|
def get_by_id(api, id, none=False):
|
|
for value in api.activities_json:
|
|
if value["id"] == id:
|
|
return Activity(
|
|
api,
|
|
value["id"],
|
|
value["name"],
|
|
Project.get_by_id(api, value["project"]),
|
|
)
|
|
if not none:
|
|
raise NotFound()
|
|
|
|
def __repr__(self):
|
|
return f"Activity (id={self.id}, name={self.name}, project={self.project})"
|