Add get and toggle methods to routes
This commit is contained in:
parent
69b6b0c4fb
commit
9462f236b4
|
@ -17,6 +17,13 @@ class Route(HSAPICall):
|
||||||
response = self.call('get')
|
response = self.call('get')
|
||||||
return v1ListRoutesResponse(**response.json())
|
return v1ListRoutesResponse(**response.json())
|
||||||
|
|
||||||
|
def get(self, routeId: int) -> Optional[v1Route | None]:
|
||||||
|
routes = self.list()
|
||||||
|
route = [r for r in routes.routes if r.id == routeId]
|
||||||
|
if route:
|
||||||
|
return route[0]
|
||||||
|
return None
|
||||||
|
|
||||||
def delete(self, routeId: str) -> None:
|
def delete(self, routeId: str) -> None:
|
||||||
self.call('delete', call_path=routeId)
|
self.call('delete', call_path=routeId)
|
||||||
|
|
||||||
|
@ -25,3 +32,10 @@ class Route(HSAPICall):
|
||||||
|
|
||||||
def disable(self, routeId: int) -> None:
|
def disable(self, routeId: int) -> None:
|
||||||
self.call('post', f'{routeId}/disable')
|
self.call('post', f'{routeId}/disable')
|
||||||
|
|
||||||
|
def toggle(self, routeId: int) -> None:
|
||||||
|
route = self.get(routeId)
|
||||||
|
if route and route.enabled:
|
||||||
|
self.disable(routeId)
|
||||||
|
else:
|
||||||
|
self.enable(routeId)
|
||||||
|
|
Loading…
Reference in New Issue