add_remove_labels.py

add-remove_labels.py · 1.2 KiB · Python Raw

import asyncio

from lightkube import AsyncClient
from lightkube.resources.core_v1 import Node
from lightkube.types import PatchType

async def set_feature_label(
    client: AsyncClient, node_name: str, kind: str, variant: str
):
    """Add label with feature to the node"""
    # create patch
    patch_metadata = [
        {
            "op": "add",
            "path": f"/metadata/labels/node-feature.k8s.dpc24.ru~1{kind}",
            "value": variant,
        }
    ]
    # run patch
    await client.patch(
        Node, name=node_name, obj=patch_metadata, patch_type=PatchType.JSON
    )

async def unset_feature_label(
    client: AsyncClient, node_name: str, kind: str, variant: str
):
    """Remove label with feature to the node"""
    # delete patch
    patch_metadata = [
        {
            "op": "remove",
            "path": f"/metadata/labels/node-feature.k8s.dpc24.ru~1{kind}",
        }
    ]
    # run patch
    await client.patch(
        Node, name=node_name, obj=patch_metadata, patch_type=PatchType.JSON
    )

async def main():
    lk = AsyncClient()
    await set_feature_label(lk, "feature-core-k8s2-w9-dpc", "test", "nvidia")

if __name__ == "__main__":
    asyncio.run(main())

1	import asyncio
2
3	from lightkube import AsyncClient
4	from lightkube.resources.core_v1 import Node
5	from lightkube.types import PatchType
6
7
8	async def set_feature_label(
9	client: AsyncClient, node_name: str, kind: str, variant: str
10	):
11	"""Add label with feature to the node"""
12	# create patch
13	patch_metadata = [
14	{
15	"op": "add",
16	"path": f"/metadata/labels/node-feature.k8s.dpc24.ru~1{kind}",
17	"value": variant,
18	}
19	]
20	# run patch
21	await client.patch(
22	Node, name=node_name, obj=patch_metadata, patch_type=PatchType.JSON
23	)
24
25
26	async def unset_feature_label(
27	client: AsyncClient, node_name: str, kind: str, variant: str
28	):
29	"""Remove label with feature to the node"""
30	# delete patch
31	patch_metadata = [
32	{
33	"op": "remove",
34	"path": f"/metadata/labels/node-feature.k8s.dpc24.ru~1{kind}",
35	}
36	]
37	# run patch
38	await client.patch(
39	Node, name=node_name, obj=patch_metadata, patch_type=PatchType.JSON
40	)
41
42	async def main():
43	lk = AsyncClient()
44	await set_feature_label(lk, "feature-core-k8s2-w9-dpc", "test", "nvidia")
45
46
47	if __name__ == "__main__":
48	asyncio.run(main())
49