88baa30d04
Introduce logging to handle cases where the "object" key is missing from the input data. This helps with debugging by printing the data in JSON format before returning it unchanged.
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
import json
|
|
|
|
from api import Api, ApiAuthType
|
|
from typing import Tuple
|
|
|
|
|
|
class PelicanApi(Api):
|
|
def __init__(self, base_url, auth_token, client_auth_token):
|
|
self.base_url = base_url
|
|
self.auth_token = auth_token
|
|
self.client_auth_token = client_auth_token
|
|
|
|
def apply_authentication(self, use_alt_auth: bool = False) -> Tuple[ApiAuthType, dict]:
|
|
if use_alt_auth:
|
|
return ApiAuthType.Header, {
|
|
"Authorization": "Bearer " + self.client_auth_token
|
|
}
|
|
return ApiAuthType.Header, {
|
|
"Authorization": "Bearer " + self.auth_token
|
|
}
|
|
|
|
def transform(self, data):
|
|
if "object" not in data:
|
|
print("No object in data")
|
|
print(json.dumps(data))
|
|
return data
|
|
type = data["object"]
|
|
if type == "list":
|
|
return [self.transform(x) for x in data["data"]]
|
|
|
|
if type in ["allocation", "server", "node", "stats"]:
|
|
return data["attributes"]
|
|
|
|
return data
|
|
|
|
def nodes(self):
|
|
return self._get("/application/nodes")
|
|
|
|
def servers(self):
|
|
return self._get("/application/servers")
|
|
|
|
def allocations(self, node_id: int):
|
|
return self._get("/application/nodes/" + str(node_id) + "/allocations")
|
|
|
|
def server_allocations(self, identifier):
|
|
return self._get("/client/servers/" + str(identifier) + "/network/allocations")
|
|
|
|
def resources(self, identifier):
|
|
return self._get("/client/servers/" + str(identifier) + "/resources")
|