blob: 6a68b7046de09e8c236893172c684bea5eeb4ee7 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
"""APIs for interacting with Kubernetes nodes."""
from kubernetes_asyncio import client
from kubernetes_asyncio.client.api_client import ApiClient
from kubernetes_asyncio.client.models import V1NodeList
async def list_nodes() -> V1NodeList:
"""List Kubernetes nodes."""
async with ApiClient() as api:
api = client.CoreV1Api(api)
return await api.list_node()
async def _change_cordon(node: str, cordon: bool) -> None:
async with ApiClient() as api:
api = client.CoreV1Api(api)
await api.patch_node(
node,
body={"spec": {"unschedulable": cordon}},
)
async def cordon_node(node: str) -> None:
"""Cordon a Kubernetes node."""
await _change_cordon(node, True)
async def uncordon_node(node: str) -> None:
"""Uncordon a Kubernetes node."""
await _change_cordon(node, False)
|
