kubernetes-watch 0.1.10__tar.gz → 0.1.12__tar.gz

{kubernetes_watch-0.1.10 → kubernetes_watch-0.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kubernetes-watch
-Version: 0.1.10
+Version: 0.1.12
 Summary: 
 Author: bmotevalli
 Author-email: b.motevalli@gmail.com
@@ -15,7 +15,7 @@ Requires-Dist: boto3 (>=1.34.68,<2.0.0)
 Requires-Dist: humps (>=0.2.2,<0.3.0)
 Requires-Dist: hvac (>=2.1.0,<3.0.0)
 Requires-Dist: kubernetes (>=29.0.0,<30.0.0)
-Requires-Dist: prefect (>=3.4.17,<4.0.0)
+Requires-Dist: prefect (==3.4.18)
 Requires-Dist: psycopg2 (>=2.9.10,<3.0.0)
 Requires-Dist: pydantic (>=2.11.7,<3.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)

{kubernetes_watch-0.1.10 → kubernetes_watch-0.1.12}/kube_watch/modules/clusters/kube.py

@@ -1,7 +1,6 @@
 from prefect import get_run_logger
 from typing import List
-from kubernetes import config
-from kubernetes import client
+from kubernetes import config, client, utils
 from kubernetes.client.rest import ApiException
 import base64
 import datetime
@@ -183,4 +182,76 @@ def has_mismatch_image_digest(repo_digest, label_selector, namespace):
     logger.info("Images are in-sync.")
     logger.info(f"Repo digest: {repo_digest}")
     logger.info(f"Curr digest: {current_image_id.split('@')[-1]}")
-    return False
+    return False
+
+
+def update_deployment_image_if_needed(namespace, deployment_name, container_name, new_tag):
+    """
+    Updates the deployment's container image tag only if it differs from the current one.
+    The repository URI is extracted automatically from the current image reference.
+
+    Args:
+        namespace: Kubernetes namespace where the deployment resides.
+        deployment_name: Name of the deployment to patch.
+        container_name: Name of the container inside the deployment.
+        new_tag: The new image tag to set, e.g. 'v1.2.3'
+
+    Returns:
+        True if the image was updated and rollout triggered, False otherwise.
+    """
+    try:
+        # Load kube config (works both in-cluster and local)
+        try:
+            config.load_incluster_config()
+        except config.ConfigException:
+            config.load_kube_config()
+
+        apps_v1 = client.AppsV1Api()
+
+        # Read the current deployment
+        deployment = apps_v1.read_namespaced_deployment(deployment_name, namespace)
+
+        # Find the container we want to update
+        containers = deployment.spec.template.spec.containers
+        target_container = next((c for c in containers if c.name == container_name), None)
+        if not target_container:
+            logger.error(f"Container '{container_name}' not found in deployment '{deployment_name}'.")
+            return False
+
+        current_image = target_container.image
+        if ':' in current_image:
+            repo_uri, current_tag = current_image.rsplit(':', 1)
+        else:
+            repo_uri, current_tag = current_image, "latest"
+
+        if current_tag == new_tag:
+            logger.info(f"No update needed. Deployment '{deployment_name}' already using tag '{new_tag}'.")
+            return False
+
+        # Build the new image reference using same repo
+        new_image = f"{repo_uri}:{new_tag}"
+        patch = {
+            "spec": {
+                "template": {
+                    "spec": {
+                        "containers": [
+                            {"name": container_name, "image": new_image}
+                        ]
+                    }
+                }
+            }
+        }
+
+        logger.info(f"Updating '{deployment_name}' from '{current_image}' to '{new_image}'")
+        apps_v1.patch_namespaced_deployment(
+            name=deployment_name,
+            namespace=namespace,
+            body=patch
+        )
+
+        logger.info(f"Deployment '{deployment_name}' successfully updated to {new_image}")
+        return True
+
+    except Exception as e:
+        logger.error(f"Failed to update deployment '{deployment_name}': {e}")
+        return False

{kubernetes_watch-0.1.10 → kubernetes_watch-0.1.12}/kube_watch/modules/providers/aws.py

@@ -2,7 +2,9 @@
 # This class is deprecated. Please refer to aws.py
 #========================================================================
 import json
+import re
 import base64
+from packaging import version
 from datetime            import datetime   , timezone, timedelta
 import boto3
 from botocore.exceptions import ClientError
@@ -73,6 +75,50 @@ def task_get_latest_image_digest(session, resource, region, repository_name, tag
     
     raise ValueError('Unknown resource')
 
+
+def task_get_highest_version_tag(session, resource, region, repository_name, tag_pattern=r"^v?(?P<version>\d+\.\d+\.\d+([-+][0-9A-Za-z.-]+)?)$"):
+    """
+    Fetches the highest versioned tag from an ECR repository, similar to FluxCD's semver filter.
+    
+    Args:
+        session: boto3 session
+        resource: AwsResources enum
+        region: AWS region
+        repository_name: Name of the ECR repository
+        tag_pattern: Regex pattern to extract version component (default: semver)
+    
+    Returns:
+        The highest matching tag or None if not found.
+    """
+    if resource == AwsResources.ECR:
+        ecr_client = session.client('ecr', region_name=region)
+        try:
+            pattern = re.compile(tag_pattern)
+            paginator = ecr_client.get_paginator('describe_images')
+            matched_tags = []
+
+            for page in paginator.paginate(repositoryName=repository_name, filter={'tagStatus': 'TAGGED'}):
+                for image in page.get('imageDetails', []):
+                    for tag in image.get('imageTags', []):
+                        match = pattern.match(tag)
+                        if match:
+                            ver_str = match.group('version')
+                            matched_tags.append((tag, version.parse(ver_str)))
+
+            if not matched_tags:
+                logger.info(f"No tags matching pattern '{tag_pattern}' found in repository {repository_name}.")
+                return None
+
+            # Sort by semantic version and pick the highest
+            highest_tag, _ = max(matched_tags, key=lambda t: t[1])
+            return highest_tag
+
+        except Exception as e:
+            logger.error(f"Error fetching highest version tag from {repository_name}: {e}")
+            return None
+
+    raise ValueError('Unknown resource')
+
 #========================================================================================
 # IAM Cred update
 #========================================================================================

{kubernetes_watch-0.1.10 → kubernetes_watch-0.1.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kubernetes-watch"
-version = "0.1.10"
+version = "0.1.12"
 description = ""
 authors = ["bmotevalli <b.motevalli@gmail.com>"]
 packages = [{include = "kube_watch"}]
@@ -8,7 +8,7 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = ">=3.10,<3.14"
-prefect = "^3.4.17"
+prefect = "3.4.18"
 boto3 = "^1.34.68"
 hvac = "^2.1.0"
 humps = "^0.2.2"