79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
from .model import HSAPICall, formatTags
|
|
from .schemas import v1Route, v1Node
|
|
from typing import Optional, List
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class v1SetTagsNodeRequest(BaseModel):
|
|
tags: Optional[List[str]] = Field(alias="tags", default=None)
|
|
|
|
def model_post_init(self, ctx):
|
|
self.tags = formatTags(self.tags)
|
|
|
|
|
|
class v1MoveNodeRequest(BaseModel):
|
|
user: Optional[str] = Field(alias="user", default=None)
|
|
|
|
|
|
class v1BackfillNodeIPsRequest(BaseModel):
|
|
confirmed: Optional[bool] = Field(alias="confirmed", default=True)
|
|
|
|
|
|
class v1ListNodesResponse(BaseModel):
|
|
nodes: Optional[List[Optional[v1Node]]] = Field(
|
|
alias="nodes", default=None)
|
|
|
|
|
|
class v1NodeResponse(BaseModel):
|
|
node: Optional[v1Node] = Field(alias="node", default=None)
|
|
|
|
|
|
class v1GetNodeRoutesResponse(BaseModel):
|
|
routes: Optional[List[Optional[v1Route]]] = Field(
|
|
alias="routes", default=None)
|
|
|
|
|
|
class v1BackfillNodeIPsResponse(BaseModel):
|
|
changes: Optional[List[str]] = Field(alias="changes", default=None)
|
|
|
|
|
|
class Node(HSAPICall):
|
|
|
|
objectPath = "node"
|
|
|
|
def list(self) -> v1ListNodesResponse:
|
|
response = self.call('get')
|
|
return v1ListNodesResponse(**response.json())
|
|
|
|
def get(self, nodeId: str) -> v1NodeResponse:
|
|
response = self.call('get', call_path=nodeId)
|
|
return v1NodeResponse(**response.json())
|
|
|
|
def delete(self, nodeId: str) -> None:
|
|
self.call('delete', call_path=nodeId)
|
|
|
|
def expire(self, nodeId: str) -> None:
|
|
self.call('post', f'{nodeId}/expire')
|
|
|
|
def rename(self, nodeId: str, newName: str) -> v1NodeResponse:
|
|
response = self.call('post', f'{nodeId}/rename/{newName}')
|
|
return v1NodeResponse(**response.json())
|
|
|
|
def move(self, nodeId: str, data: v1MoveNodeRequest) -> v1NodeResponse:
|
|
response = self.call('post', f'{nodeId}/user', data)
|
|
return v1NodeResponse(**response.json())
|
|
|
|
def routes(self, nodeId: str) -> v1GetNodeRoutesResponse:
|
|
response = self.call('get', f'{nodeId}/routes')
|
|
return v1GetNodeRoutesResponse(**response.json())
|
|
|
|
def setTags(self, nodeId: str, data: v1SetTagsNodeRequest) -> v1NodeResponse:
|
|
response = self.call('post', f'{nodeId}/tags', data)
|
|
return v1NodeResponse(**response.json())
|
|
|
|
# Broken on server
|
|
def backfillips(self, confirmed: bool = True) -> v1BackfillNodeIPsResponse:
|
|
response = self.call(
|
|
'post', f'/backfillips?confirmed={confirmed}')
|
|
return v1BackfillNodeIPsResponse(**response.json())
|