fix: objects exception if not exists

Signed-off-by: djerfy <[email protected]>
djerfy · Jan 13, 2024 · c3ed0cc · c3ed0cc
1 parent d7c5e31
commit c3ed0cc
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/src/modules/kubernetes/openebs/cstorpoolclusters.py b/src/modules/kubernetes/openebs/cstorpoolclusters.py
@@ -17,7 +17,12 @@ def openebsGetCstorpoolclusters(config):
     if config['monitoring']['openebs']['engine'] != "cstor":
         return cstorpoolclusters
 
-    for cstorpoolcluster in rawObjects(kubernetes.list_cluster_custom_object(group="cstor.openebs.io", version="v1", plural="cstorpoolclusters")):
+    try:
+        objects = kubernetes.list_cluster_custom_object(group="cstor.openebs.io", version="v1", plural="cstorpoolclusters")
+    except Exception:
+        return cstorpoolclusters
+
+    for cstorpoolcluster in rawObjects(objects):
         json = {
             "name": cstorpoolcluster['metadata']['name'],
             "namespace": cstorpoolcluster['metadata']['namespace'],

diff --git a/src/modules/kubernetes/openebs/cstorpoolinstances.py b/src/modules/kubernetes/openebs/cstorpoolinstances.py
@@ -16,8 +16,13 @@ def openebsGetCstorpoolinstances(config):
 
     if config['monitoring']['openebs']['engine'] != "cstor":
         return cstorpoolinstances
+
+    try:
+        objects = kubernetes.list_cluster_custom_object(group="cstor.openebs.io", version="v1", plural="cstorpoolinstances")
+    except Exception:
+        return cstorpoolinstances
 
-    for cstorpoolinstance in rawObjects(kubernetes.list_cluster_custom_object(group="cstor.openebs.io", version="v1", plural="cstorpoolinstances")):
+    for cstorpoolinstance in rawObjects(objects):
         json = {
             "name": cstorpoolinstance['metadata']['name'],
             "namespace": cstorpoolinstance['metadata']['namespace'],

diff --git a/src/modules/kubernetes/trivy/vulnerabilityreports.py b/src/modules/kubernetes/trivy/vulnerabilityreports.py
@@ -14,6 +14,11 @@ def trivyGetVulnerabilityreports(config=None):
 
     reports = []
 
-    for vuln in rawObjects(kubernetes.list_cluster_custom_object(group="aquasecurity.github.io", version="v1alpha1", plural="vulnerabilityreports")):
+    try:
+        objects = kubernetes.list_cluster_custom_object(group="aquasecurity.github.io", version="v1alpha1", plural="vulnerabilityreports")
+    except Exception:
+        return reports
+
+    for vuln in rawObjects(objects):
         print(vuln['metadata']['name'])
         print(vuln['report']['summary'])