kubernetes-watch 0.1.10__py3-none-any.whl → 0.1.12__py3-none-any.whl

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

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.dist-info → kubernetes_watch-0.1.12.dist-info}/METADATA

@@ -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.dist-info → kubernetes_watch-0.1.12.dist-info}/RECORD

@@ -9,7 +9,7 @@ kube_watch/models/common.py,sha256=FQktpX552zSCigMxEzm4S07SvrHv5RA7YwVJHgv7uuI,5
 kube_watch/models/workflow.py,sha256=ZFBMz_LmYgROcbz2amSvms38K770njnyZC6h1bpTXGU,1634
 kube_watch/modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kube_watch/modules/clusters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kube_watch/modules/clusters/kube.py,sha256=_skDVeKIQcZ0nC9JLY0oaFmaeEVRHS-Xbrxh1HR2t9Y,7582
+kube_watch/modules/clusters/kube.py,sha256=ZvWpLrdgftLHexuigil8aa23BjqTZDmyeb6ZCAW6Y7w,10205
 kube_watch/modules/database/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kube_watch/modules/database/model.py,sha256=MWG9UW6g0KuBzW6MjzPBtknAk7GmuncQrdAq6HHarTo,207
 kube_watch/modules/database/postgre.py,sha256=1Sq2YFwgCJM_FWKabV2S1bFAIl2GBwytTtPCuLfVhu8,8182
@@ -22,7 +22,7 @@ kube_watch/modules/logic/trasnform.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 kube_watch/modules/mock/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kube_watch/modules/mock/mock_generator.py,sha256=BKKQFCxxQgFW_GFgeIbkyIbuNU4328xTTaFfTwFLsS8,1262
 kube_watch/modules/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kube_watch/modules/providers/aws.py,sha256=oly88Bihn-vR0NdOoCT-P1MNgPw8FLNfS1Ha4hc_4bc,8375
+kube_watch/modules/providers/aws.py,sha256=qKL7oGKeKOUO5AQRQqcrR35kzOpq42uNXuXumeU-gtw,10224
 kube_watch/modules/providers/git.py,sha256=5d7vkeTbVD_XN-mpCoPTRmM6lwgrwkDwcCIsjqYwmaQ,3778
 kube_watch/modules/providers/github.py,sha256=eQY8sLy2U6bOWMpFxA73DFCPVuswhTXSG25KmYSuo5s,5212
 kube_watch/modules/providers/vault.py,sha256=etzzHbTrUDsTUpeUN-xg0Xh8ulqC0-1FA3tHRZinIOo,7193
@@ -30,7 +30,7 @@ kube_watch/standalone/metarecogen/ckan_to_gn.py,sha256=LWd7ikyxRIC1IGt6CtALnDOEo
 kube_watch/watch/__init__.py,sha256=9KE0Sf1nLUTNaFvXbiQCgf11vpG8Xgmb5ddeMAmak3Q,88
 kube_watch/watch/helpers.py,sha256=8BQnQ6AeLHs0JEq54iKYDvWURb1F-kROJxwIcl_nv_Y,6276
 kube_watch/watch/workflow.py,sha256=CaXHFuEWVsFjBv5dU4IfVMeTlGJWyKaE1But9-YzVWk,9769
-kubernetes_watch-0.1.10.dist-info/LICENSE,sha256=_H2QdL-2dXbivDmOpJ11DnqJewSFhSJwGpHx_WAE-CA,1075
-kubernetes_watch-0.1.10.dist-info/METADATA,sha256=Fof5Cg1A--c2oCQ8VCrNb_GAeqw1rKjHMltZzjXurLY,5607
-kubernetes_watch-0.1.10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-kubernetes_watch-0.1.10.dist-info/RECORD,,
+kubernetes_watch-0.1.12.dist-info/LICENSE,sha256=_H2QdL-2dXbivDmOpJ11DnqJewSFhSJwGpHx_WAE-CA,1075
+kubernetes_watch-0.1.12.dist-info/METADATA,sha256=zG9wEStvXgfYthz6Aa5i05Oyoh7ygmIMWtI8kdzRM00,5600
+kubernetes_watch-0.1.12.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+kubernetes_watch-0.1.12.dist-info/RECORD,,