feat: add support for StrimziPodSet

.github/actions/get-python-version.yaml

@@ -0,0 +1,14 @@
+name: "Get Python Version"
+description: "Read the python version from .python-version file"
+outputs:
+  python-version:
+    description: "Python version"
+    value: ${{ steps.get_version.outputs.python-version }}
+runs:
+  using: "composite"
+  steps:
+    - name: get python-version
+      shell: bash
+      id: get_version
+      run: |
+        echo "python-version=$(cat .python-version)" >> $GITHUB_OUTPUT

.github/workflows/build-on-release.yml

@@ -14,12 +14,15 @@ jobs:
     runs-on: ${{ matrix.os }}
 
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v4
+
+    - uses: ./.github/actions/get-python-version
+      id: get_version
 
     - name: Set up Python
-      uses: actions/setup-python@v2
+      uses: actions/setup-python@v5
       with:
-        python-version: '3.11'
+        python-version: "${{ steps.get_version.outputs.python-version }}"
 
     - name: Install dependencies
       run: |
@@ -157,7 +160,7 @@ jobs:
       MAC_BUILD_HASH: ${{ steps.calc-hash.outputs.MAC_BUILD_HASH }}
     steps:
       - name: Checkout Repository
-        uses: actions/checkout@v2
+        uses: actions/checkout@v4
       - name: Download MacOS artifact
         uses: actions/download-artifact@v4
         with:
@@ -175,7 +178,7 @@ jobs:
       LINUX_BUILD_HASH: ${{ steps.calc-hash.outputs.LINUX_BUILD_HASH }}
     steps:
       - name: Checkout Repository
-        uses: actions/checkout@v2
+        uses: actions/checkout@v4
       - name: Download Linux artifact
         uses: actions/download-artifact@v4
         with:
@@ -190,7 +193,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Checkout homebrew-krr repository
-        uses: actions/checkout@v2
+        uses: actions/checkout@v4
         with:
           repository: robusta-dev/homebrew-krr
           token: ${{ secrets.MULTIREPO_GITHUB_TOKEN }}

.github/workflows/docker-build-on-tag.yml

@@ -11,7 +11,7 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v4
 
     - name: Login to Docker Hub
       uses: docker/#-action@v1
@@ -20,17 +20,21 @@ jobs:
         password: ${{ secrets.DOCKER_PASSWORD }}
 
     - name: Set up Docker Buildx
-      uses: docker/setup-buildx-action@v1
+      uses: docker/setup-buildx-action@v3
+
+    - uses: ./.github/actions/get-python-version
+      id: get_version
 
     - name: Build and push Docker images
-      uses: docker/build-push-action@v2
+      uses: docker/build-push-action@v6
       with:
         context: .
         platforms: linux/arm64,linux/amd64
         push: true
         tags: robustadev/krr:${{ github.ref_name }}
         build-args: |
           BUILDKIT_INLINE_CACHE=1
+          PYTHON_VERSION=${{ steps.get_version.outputs.python-version }}
 
     - name: Set up gcloud CLI
       uses: google-github-actions/setup-gcloud@v0.2.0

.github/workflows/pytest-on-push.yml

@@ -8,12 +8,14 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-    - uses: actions/checkout@v2
+    - uses: actions/checkout@v4
+    - uses: ./.github/actions/get-python-version
+      id: get_version
 
     - name: Set up Python
-      uses: actions/setup-python@v2
+      uses: actions/setup-python@v5
       with:
-        python-version: '3.9'
+        python-version: "${{ steps.get_version.outputs.python-version }}"
 
     - name: Install dependencies
       run: |

Dockerfile

@@ -1,5 +1,6 @@
-# Use the official Python 3.9 slim image as the base image
-FROM python:3.12-slim as builder
+# PYTHON_VERSION is passed as a build argument which uses .python-version 
+ARG PYTHON_VERSION
+FROM python:${PYTHON_VERSION}-slim as builder
 ENV LANG=C.UTF-8
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1

build_release.sh

@@ -1,5 +1,10 @@
+#!/bin/bash
+
+python_version=$(cat .python-version)
+
 docker buildx build \
   --build-arg BUILDKIT_INLINE_CACHE=1 \
+  --build-arg "PYTHON_VERSION=${python_version}" \
   --platform linux/arm64,linux/amd64 \
   --tag us-central1-docker.pkg.dev/genuine-flight-317411/devel/krr:${TAG} \
   --push \

pyproject.toml

@@ -23,7 +23,7 @@ plugins = "numpy.typing.mypy_plugin,pydantic.mypy"
 krr = "robusta_krr.main:run"
 
 [tool.poetry.dependencies]
-python = ">=3.9,<=3.12.3"
+python = ">=3.9,<=3.12.7"
 typer = { extras = ["all"], version = "^0.7.0" }
 pydantic = "^1.10.7"
 kubernetes = "^26.1.0"

robusta_krr/core/integrations/kubernetes/__init__.py

@@ -101,6 +101,7 @@ async def list_scannable_objects(self) -> list[K8sObjectData]:
         workload_object_lists = await asyncio.gather(
             self._list_deployments(),
             self._list_rollouts(),
+            self._list_strimzipodsets(),
             self._list_deploymentconfig(),
             self._list_all_statefulsets(),
             self._list_all_daemon_set(),
@@ -204,7 +205,10 @@ def __build_scannable_object(
         labels = {}
         annotations = {}
         if item.metadata.labels:
-            labels = item.metadata.labels
+            if type(item.metadata.labels) is ObjectLikeDict:
+                labels = item.metadata.labels.__dict__
+            else:  
+                labels = item.metadata.labels
 
         if item.metadata.annotations:
             if type(item.metadata.annotations) is ObjectLikeDict:
@@ -364,6 +368,30 @@ async def _extract_containers(item: Any) -> list[V1Container]:
             extract_containers=_extract_containers,
         )
 
+    def _list_strimzipodsets(self) -> list[K8sObjectData]:
+        # NOTE: Using custom objects API returns dicts, but all other APIs return objects
+        # We need to handle this difference using a small wrapper
+        return self._list_scannable_objects(
+            kind="StrimziPodSet",
+            all_namespaces_request=lambda **kwargs: ObjectLikeDict(
+                self.custom_objects.list_cluster_custom_object(
+                    group="core.strimzi.io",
+                    version="v1beta2",
+                    plural="strimzipodsets",
+                    **kwargs,
+                )
+            ),
+            namespaced_request=lambda **kwargs: ObjectLikeDict(
+                self.custom_objects.list_namespaced_custom_object(
+                    group="core.strimzi.io",
+                    version="v1beta2",
+                    plural="strimzipodsets",
+                    **kwargs,
+                )
+            ),
+            extract_containers=lambda item: item.spec.pods[0].spec.containers,
+        )
+
     def _list_deploymentconfig(self) -> list[K8sObjectData]:
         # NOTE: Using custom objects API returns dicts, but all other APIs return objects
         # We need to handle this difference using a small wrapper

robusta_krr/core/models/objects.py

@@ -8,7 +8,7 @@
 from robusta_krr.utils.batched import batched
 from kubernetes.client.models import V1LabelSelector
 
-KindLiteral = Literal["Deployment", "DaemonSet", "StatefulSet", "Job", "CronJob", "Rollout", "DeploymentConfig"]
+KindLiteral = Literal["Deployment", "DaemonSet", "StatefulSet", "Job", "CronJob", "Rollout", "DeploymentConfig", "StrimziPodSet"]
 
 
 class PodData(pd.BaseModel):

robusta_krr/main.py

@@ -98,7 +98,7 @@ def run_strategy(
                     None,
                     "--resource",
                     "-r",
-                    help="List of resources to run on (Deployment, StatefulSet, DaemonSet, Job, Rollout). By default, will run on all resources. Case insensitive.",
+                    help="List of resources to run on (Deployment, StatefulSet, DaemonSet, Job, Rollout, StrimziPodSet). By default, will run on all resources. Case insensitive.",
                     rich_help_panel="Kubernetes Settings",
                 ),
                 selector: Optional[str] = typer.Option(