diff options
author | 2024-02-02 16:12:08 +0000 | |
---|---|---|
committer | 2024-02-02 16:12:08 +0000 | |
commit | 5fd87154cbf326068b181f366286a635a7eaa119 (patch) | |
tree | 709446f42d8981e83bca8b17f4fda280d477c7e2 /arthur/apis/kubernetes/nodes.py | |
parent | Add ruff isort configuration (diff) |
Enable all ruff rules
Diffstat (limited to 'arthur/apis/kubernetes/nodes.py')
-rw-r--r-- | arthur/apis/kubernetes/nodes.py | 14 |
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) |