81 lines
2.1 KiB
Python
81 lines
2.1 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()
|
|
|
|
|
|
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):
|
|
for value in api.projects_json:
|
|
if value['id'] == id:
|
|
return Project(api, value['id'], value['name'],
|
|
Customer.get_by_id(api, value['customer']))
|
|
raise NotFound()
|
|
|
|
def __repr__(self):
|
|
return f'Project (id={self.id}, name={self.name}, customer={self.customer})'
|
|
|
|
|
|
class Activity():
|
|
pass
|