feat: config.in_cluster

knechtionscoding · Apr 5, 2023 · 35bb4f8 · 35bb4f8
1 parent a42c1aa
commit 35bb4f8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/main.py b/main.py
@@ -314,7 +314,10 @@ def delete_image(client: boto3.client, image: dict):  # pragma: no cover
 def main():  # pragma: no cover
     deletable_images = []
     keepable_images = []
-    config.load_kube_config()
+    try:
+        config.load_kube_config()
+    except config.config_exception.ConfigException:
+        config.load_incluster_config()
 
     client = boto3.client("ecr")
     if os.getenv("AWS_REGISTRY_ID"):