hamster-tools/hamstertools/kimai.py

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})'