K8s container annotations

granulate_utils/containers/cri.py

@@ -17,8 +17,9 @@
 
 import json
 from contextlib import contextmanager, suppress
+from dataclasses import dataclass, field
 from datetime import datetime, timezone
-from typing import Any, List, Optional, Type, TypeVar
+from typing import Any, Dict, List, Optional, Type, TypeVar
 
 import grpc  # type: ignore # no types-grpc sadly
 import psutil
@@ -122,14 +123,15 @@ def _create_container(
             with suppress(psutil.NoSuchProcess):
                 process = psutil.Process(pid)
 
-        return Container(
+        return K8sContainer(
             runtime=self.runtime_name,
             name=self._reconstruct_name(container),
             id=container.id,
             labels=container.labels,
             running=container.state == self.api.api_pb2.CONTAINER_RUNNING,
             process=process,
             time_info=time_info,
+            annotations=container.annotations,
         )
 
 
@@ -182,3 +184,8 @@ def get_container(self, container_id: str, all_info: bool) -> Container:
 
     def get_runtimes(self) -> List[str]:
         return [client.runtime_name for client in self._clients]
+
+
+@dataclass
+class K8sContainer(Container):
+    annotations: Dict[str, str] = field(default_factory=dict)