fix: convert list items to strings before exclude check is performed (#9

)
thegeeklab · Jun 13, 2021 · 5892717 · 5892717
1 parent 75cfc01
commit 5892717
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/prometheuspvesd/config.py b/prometheuspvesd/config.py
@@ -75,7 +75,7 @@ class Config():
         },
         "exclude_vmid": {
             "default": [],
-            "env": "EXCLUDE_STATE",
+            "env": "EXCLUDE_VMID",
             "file": True,
             "type": environs.Env().list
         },

diff --git a/prometheuspvesd/discovery.py b/prometheuspvesd/discovery.py
@@ -124,10 +124,10 @@ def _exclude(self, pve_list):
             if obj["template"] == 1:
                 continue
 
-            if obj["status"] in self.config.config["exclude_state"]:
+            if obj["status"] in map(str, self.config.config["exclude_state"]):
                 continue
 
-            if obj["vmid"] in self.config.config["exclude_vmid"]:
+            if obj["vmid"] in map(str, self.config.config["exclude_vmid"]):
                 continue
 
             filtered.append(item.copy())