97 lines
No EOL
2.7 KiB
Python
97 lines
No EOL
2.7 KiB
Python
from urllib.request import urlopen, Request, urljoin
|
|
from base64 import b64decode
|
|
|
|
import json
|
|
|
|
from classes.exceptions import AdonisAuthenticationDeniedError, AdonisError
|
|
from classes.crew import Crew
|
|
|
|
from const import USER_AGENT
|
|
|
|
from PIL.Image import Image
|
|
|
|
|
|
class Adonis:
|
|
def __init__(self, crew_portal_base_url: str, login: str, password: str):
|
|
self.crew_portal_base_url = crew_portal_base_url
|
|
self.login = login
|
|
self.password = password
|
|
|
|
@classmethod
|
|
def fromConfig(cls, config):
|
|
return cls(config.crew_portal_base_url, config.login, config.password)
|
|
|
|
def requestCrewPortal(self, endpoint: str, payload: dict, add_token: bool = True, wrap_request: bool = True, extract_response: bool = True):
|
|
url = urljoin(self.crew_portal_base_url, endpoint)
|
|
|
|
req = Request(url)
|
|
req.add_header("User-Agent", USER_AGENT)
|
|
req.add_header("Content-Type", "application/json")
|
|
req.add_header("Accept", "application/json")
|
|
|
|
if add_token:
|
|
payload["Authentication_Token"] = self.getTokenCrewPortal()
|
|
|
|
if wrap_request:
|
|
payload = {"request": payload}
|
|
|
|
req.data = json.dumps(payload).encode()
|
|
|
|
res = json.loads(urlopen(req).read())
|
|
|
|
if extract_response:
|
|
res = res[endpoint + "Result"]
|
|
|
|
if not res["Authentication_Approved"]:
|
|
raise AdonisAuthenticationDeniedError(
|
|
res["Authentication_ReasonDenied"])
|
|
|
|
if res["ErrorText"]:
|
|
raise AdonisError(res["ErrorText"])
|
|
|
|
return res
|
|
|
|
def getTokenCrewPortal(self, lifetime: int = 60):
|
|
data = {
|
|
"credentials": {
|
|
"Login": self.login,
|
|
"Password": self.password,
|
|
"LifeTime": lifetime
|
|
}
|
|
}
|
|
|
|
result = self.requestCrewPortal("GNL_API_AUTHENTICATION", data, False, False)
|
|
|
|
return result["Authentication_Token"]
|
|
|
|
def getCrewPersonalData(self, pin):
|
|
data = {
|
|
"Pin": pin
|
|
}
|
|
|
|
result = self.requestCrewPortal("DG_PersonalInformationRead", data)
|
|
|
|
return result
|
|
|
|
def getCrewProfilePicture(self, pin):
|
|
data = {
|
|
"Pin": pin
|
|
}
|
|
|
|
result = self.requestCrewPortal("DG_ProfilePictureRead", data)
|
|
|
|
try:
|
|
return result["FileBase64"]
|
|
except AdonisError:
|
|
return None
|
|
|
|
def getCrew(self, pin, personal_data=True, profile_picture=True):
|
|
crew = Crew(pin)
|
|
|
|
if personal_data:
|
|
crew._personal_data = self.getCrewPersonalData(pin)
|
|
|
|
if profile_picture:
|
|
crew._profile_picture = self.getCrewProfilePicture(pin)
|
|
|
|
return crew |