33 lines
895 B
Python
33 lines
895 B
Python
|
from api import Api, ApiAuthType
|
||
|
|
from typing import Tuple
|
||
|
|
|
||
|
|
|
||
|
|
class PelicanApi(Api):
|
||
|
|
def __init__(self, base_url, auth_token):
|
||
|
|
self.base_url = base_url
|
||
|
|
self.auth_token = auth_token
|
||
|
|
|
||
|
|
def apply_authentication(self) -> Tuple[ApiAuthType, dict]:
|
||
|
|
return ApiAuthType.Header, {
|
||
|
|
"Authorization": "Bearer " + self.auth_token
|
||
|
|
}
|
||
|
|
|
||
|
|
def transform(self, data):
|
||
|
|
type = data["object"]
|
||
|
|
if type == "list":
|
||
|
|
return [self.transform(x) for x in data["data"]]
|
||
|
|
|
||
|
|
if type in ["allocation", "server", "node"]:
|
||
|
|
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")
|