Initial Commit

1 files changed, 97 insertions, 0 deletions

diff --git a/pyactivecollab.py b/pyactivecollab.py
new file mode 100644
index 0000000..d8ae306
--- /dev/null
+++ b/pyactivecollab.py
@@ -0,0 +1,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))