aboutsummaryrefslogtreecommitdiffstats
path: root/arthur/apis/kubernetes/nodes.py
diff options
context:
space:
mode:
authorGravatar Chris Lovering <[email protected]>2024-02-02 16:12:08 +0000
committerGravatar Chris Lovering <[email protected]>2024-02-02 16:12:08 +0000
commit5fd87154cbf326068b181f366286a635a7eaa119 (patch)
tree709446f42d8981e83bca8b17f4fda280d477c7e2 /arthur/apis/kubernetes/nodes.py
parentAdd ruff isort configuration (diff)
Enable all ruff rules
Diffstat (limited to 'arthur/apis/kubernetes/nodes.py')
-rw-r--r--arthur/apis/kubernetes/nodes.py14
1 files changed, 7 insertions, 7 deletions
diff --git a/arthur/apis/kubernetes/nodes.py b/arthur/apis/kubernetes/nodes.py
index ed45c26..3221a15 100644
--- a/arthur/apis/kubernetes/nodes.py
+++ b/arthur/apis/kubernetes/nodes.py
@@ -7,14 +7,14 @@ from kubernetes_asyncio.client.models import V1NodeList
async def list_nodes() -> V1NodeList:
"""List Kubernetes nodes."""
- async with ApiClient() as api:
- api = client.CoreV1Api(api)
+ async with ApiClient() as api_client:
+ api = client.CoreV1Api(api_client)
return await api.list_node()
-async def _change_cordon(node: str, cordon: bool) -> None:
- async with ApiClient() as api:
- api = client.CoreV1Api(api)
+async def _change_cordon(node: str, *, cordon: bool) -> None:
+ async with ApiClient() as api_client:
+ api = client.CoreV1Api(api_client)
await api.patch_node(
node,
body={"spec": {"unschedulable": cordon}},
@@ -23,9 +23,9 @@ async def _change_cordon(node: str, cordon: bool) -> None:
async def cordon_node(node: str) -> None:
"""Cordon a Kubernetes node."""
- await _change_cordon(node, True)
+ await _change_cordon(node, cordon=True)
async def uncordon_node(node: str) -> None:
"""Uncordon a Kubernetes node."""
- await _change_cordon(node, False)
+ await _change_cordon(node, cordon=False)