First working version

2024-06-27 16:31:55 +02:00
parent df5754518d
commit 737716f8eb
57 changed files with 1508 additions and 0 deletions
--- a/headscale_api_client/nodes.py
+++ b/headscale_api_client/nodes.py
@ -0,0 +1,48 @@
+from .model import HSAPICall
+from headscale_api_client.schemas import (v1ListNodesResponse,
+                                          v1GetNodeResponse,
+                                          v1DeleteNodeResponse,
+                                          v1ExpireNodeResponse, v1MoveNodeRequest,
+                                          v1RenameNodeResponse,
+                                          v1MoveNodeRequest,
+                                          v1MoveNodeResponse,
+                                          v1GetNodeRoutesResponse,
+                                          v1SetTagsResponse,
+                                          HeadscaleServiceSetTagsBody)
+
+
+class Node(HSAPICall):
+
+    objectPath = "node"
+
+    def list(self) -> v1ListNodesResponse:
+        response = self.call('get')
+        return v1ListNodesResponse(**response.json())
+
+    def get(self, nodeId: str) -> v1GetNodeResponse:
+        response = self.call('get', call_path=nodeId)
+        return v1GetNodeResponse(**response.json())
+
+    def delete(self, nodeId: str) -> v1DeleteNodeResponse:
+        response = self.call('delete', call_path=nodeId)
+        return v1DeleteNodeResponse(**response.json())
+
+    def expire(self, nodeId: str) -> v1ExpireNodeResponse:
+        response = self.call('post', f'{nodeId}/expire')
+        return v1ExpireNodeResponse()(**response.json())
+
+    def rename(self, nodeId: str, newName: str) -> v1RenameNodeResponse:
+        response = self.call('post', f'{nodeId}/rename/{newName}')
+        return v1RenameNodeResponse(**response.json())
+
+    def move(self, nodeId: str, data: v1MoveNodeRequest) -> v1MoveNodeResponse:
+        response = self.call('post', f'{nodeId}/user', data)
+        return v1MoveNodeResponse()(**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: HeadscaleServiceSetTagsBody) -> v1SetTagsResponse:
+        response = self.call('post', f'{nodeId}/tags', data)
+        return v1SetTagsResponse(**response.json())