1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
# -*- coding: utf-8 -*-
"""File for connecting to Active Collab."""
import json
import os
import urllib
from typing import Dict
import requests
class AuthenticationException(Exception):
"""Exception for authentication errors."""
pass
class Config():
"""Configuration for ActiveCollab Session."""
def __init__(self, filename: str='config.json') -> None:
"""Initialize config."""
self.filename = filename
self.url = None
self.user = None
self.password = None
self.client_name = None
self.client_vendor = None
def load(self) -> None:
"""Load the config file."""
with open(os.path.expanduser(self.filename), 'r') as f:
json_data = json.load(f)
self.__dict__.update(json_data)
class ActiveCollab(object):
"""Active Collab API."""
def __init__(self, config: Config) -> None:
"""Initialize the class."""
self.config = config
self.session = requests.session()
def authenticate(self) -> None:
"""Authenticate against the api."""
data = {
'username': self.config.user,
'password': self.config.password,
'client_name': self.config.client_name,
'client_vendor': self.config.client_vendor
}
auth_url = '{}/issue-token'.format(self.config.url)
r = self.session.post(auth_url, data=data)
try:
token = r.json()['token']
except json.decoder.JSONDecodeError:
raise AuthenticationException
else:
self.token = token
def get(self, url: str) -> Dict:
"""Make a get call to the API."""
if not getattr(self, 'token'):
raise AuthenticationException
headers = {'X-Angie-AuthApiToken': self.token}
url = '{}{}'.format(self.config.url, url)
r = self.session.get(url, headers=headers)
return r.json()
def post(self, url: str, data: Dict) -> Dict:
"""Make post call to the API."""
if not getattr(self, 'token'):
raise AuthenticationException
headers = {'X-Angie-AuthApiToken': self.token}
url = '{}{}'.format(self.config.url, url)
r = self.session.post(url, data=data, headers=headers)
return r.json()
def get_info(self) -> Dict:
"""Get info about the system information."""
return self.get('/info')
def get_job_types(self) -> Dict:
"""Get the available job types."""
return self.get('/job-types')
def get_projects(self) -> Dict:
"""Get the projects."""
return self.get('/projects')
def get_users(self) -> Dict:
"""Get the projects."""
return self.get('/users')
def get_time_records(self, user_id: str) -> Dict:
"""Get the time records for a user."""
return self.get('/users/{}/time-records'.format(user_id))
|
