Add get projects

This commit is contained in:
2024-02-24 17:42:26 +00:00
parent 4be5d4c325
commit 3d6fc32030
5 changed files with 92 additions and 14 deletions
+5 -3
View File
@@ -3,6 +3,8 @@ import json
import requests
from ActiveCollabAPI import AC_USER_AGENT, AC_API_VERSION
from ActiveCollabAPI.AcAccount import AcAccount
from ActiveCollabAPI.AcToken import AcToken
class ACClient:
@@ -10,17 +12,17 @@ class ACClient:
account = None
token = None
def __init__(self, account, token):
def __init__(self, account: AcAccount, token: AcToken):
self.account = account
self.token = token
#
self.base_url = self.account['url'] + '/api/v%d' % AC_API_VERSION
self.base_url = self.account.url + '/api/v%d' % AC_API_VERSION
def headers(self):
return {
'Content-Type': 'application/json; charset=utf8',
'Accept': 'application/json',
'X-Angie-AuthApiToken': self.token,
'X-Angie-AuthApiToken': self.token.token,
'User-Agent': AC_USER_AGENT
}
+10
View File
@@ -0,0 +1,10 @@
from dataclasses import dataclass
from ActiveCollabAPI.AcUser import AcUser
from ActiveCollabAPI.AcAccount import AcAccount
@dataclass
class AcLoginResponse:
user: AcUser
accounts: list[AcAccount]
+54
View File
@@ -0,0 +1,54 @@
from dataclasses import dataclass
@dataclass
class AcProject:
based_on_id: int
based_on_type: str
body: str
body_formatted: str
budget: float
budget_type: str
budgeting_interval: str
category_id: int
class_: str
company_id: int
completed_by_id: int
completed_on: int
count_discussions: int
count_files: int
count_notes: int
count_tasks: int
created_by_email: str
created_by_id: int
created_by_name: str
created_on: int
currency_id: int
email: str
file_size: int
id: int
is_billable: bool
is_client_reporting_enabled: bool
is_completed: bool
is_estimate_visible_to_subcontractors: bool
is_sample: bool
is_tracking_enabled: bool
is_trashed: bool
label_id: int
last_activity_on: int
leader_id: int
members: list[int]
members_can_change_billable: bool
name: str
project_number: int
trashed_by_id: int
trashed_on: int
updated_by_id: int
updated_on: int
url_path: str
def project_from_json(json_obj: dict) -> AcProject:
json_obj["class_"] = json_obj["class"]
del json_obj["class"]
return AcProject(**json_obj)
+15 -8
View File
@@ -2,22 +2,23 @@ from dataclasses import dataclass
from ActiveCollabAPI import AC_API_VERSION
from ActiveCollabAPI.ACAuthenticator import ACAuthenticator
from ActiveCollabAPI.ACClient import ACClient
from ActiveCollabAPI.ACTokenAuthenticator import ACTokenAuthenticator
from ActiveCollabAPI.AcAccount import AcAccount
from ActiveCollabAPI.AcLoginResponse import AcLoginResponse
from ActiveCollabAPI.AcProject import AcProject, project_from_json
from ActiveCollabAPI.AcSession import AcSession
from ActiveCollabAPI.AcToken import AcToken
from ActiveCollabAPI.AcUser import AcUser
@dataclass
class AcLoginResponse:
user: AcUser
accounts: list[AcAccount]
class ActiveCollab:
base_url: str = ""
session: AcSession = None
def __init__(self, base_url: str):
self.base_url = base_url
@@ -25,8 +26,8 @@ class ActiveCollab:
login_res = self.user_login(email, password)
cur_account = self.select_first_account(login_res.accounts)
token = self.create_token(cur_account, login_res.user)
session = AcSession(login_res.user, login_res.accounts, cur_account, token)
return session
self.session = AcSession(login_res.user, login_res.accounts, cur_account, token)
return self.session
def user_login(self, email: str, password: str) -> AcLoginResponse:
auth = ACAuthenticator(self.base_url)
@@ -59,4 +60,10 @@ class ActiveCollab:
return AcToken(res_data["token"])
def get_projects(self):
pass
client = ACClient(self.session.cur_account, self.session.token)
res = client.get_projects()
if res.status_code != 200:
raise Exception("Error %d" % res.status_code)
res_data = res.json()
projects = list(map(lambda p: project_from_json(p), res_data))
return projects
@@ -13,10 +13,15 @@ class Test_Integration_Login(TestCase):
def test_login_success(self):
ac = ActiveCollab(config["base_url"])
session = ac.login_to_first_account(config["username"], config["password"])
pprint(session)
self.assertEqual(config["first_name"], session.user.first_name)
self.assertEqual(config["last_name"], session.user.last_name)
self.assertEqual(1, len(session.accounts))
self.assertGreater(len(session.token.token), 5)
def test_get_projects(self):
ac = ActiveCollab(config["base_url"])
ac.login_to_first_account(config["username"], config["password"])
projects = ac.get_projects()
self.assertGreater(len(projects), 0)
self.assertEqual('Project', projects[0].class_)
self.assertGreater(projects[0].id, 0)