Proxmox ve plugin additions

cmk/plugins/proxmox_ve/agent_based/proxmox_ve_backup_status.py

@@ -6,7 +6,7 @@
 import json
 from collections.abc import Mapping
 from datetime import datetime, UTC
-from typing import Any, TypedDict
+from typing import Any, cast, TypedDict
 
 from cmk.agent_based.v1 import check_levels as check_levels_v1
 from cmk.agent_based.v2 import (
@@ -100,7 +100,7 @@ def check_proxmox_ve_vm_backup_status(
     levels and define result status accordingly
     >>> for result in check_proxmox_ve_vm_backup_status(
     ...     datetime.strptime("2020-12-07 21:28:02+01:00", '%Y-%m-%d %H:%M:%S%z'),
-    ...     {'age_levels_upper': (93600, 180000)},
+    ...     {'age_levels_upper': (93600.0, 180000.0)},
     ...     parse_proxmox_ve_vm_backup_status([[
     ...     '  {"last_backup": {'
     ...     '     "started_time": "2020-12-06 21:28:02+0000",'
@@ -121,17 +121,23 @@ def check_proxmox_ve_vm_backup_status(
     Result(state=<State.OK: 0>, summary='Bandwidth: 91.6 MB/s')
     Metric('backup_avgspeed', 91625968.975, boundaries=(0.0, None))
     """
-    age_levels_upper = params.get("age_levels_upper")
-    duration_levels_upper = params.get("duration_levels_upper")
-    bandwidth_levels_lower_bytes = params.get("bandwidth_levels_lower")
-    bandwidth_levels_lower = (
-        (
-            bandwidth_levels_lower_bytes[0] * 1000 * 1000,
-            bandwidth_levels_lower_bytes[1] * 1000 * 1000,
-        )
-        if bandwidth_levels_lower_bytes
-        else None
+
+    age_levels_upper = (
+        int(cast(tuple, params.get("age_levels_upper"))[0]),
+        int(cast(tuple, params.get("age_levels_upper"))[1]),
     )
+    duration_levels_upper = None
+    if params.get("duration_levels_upper") is not None:
+        duration_levels_upper = (
+            int(cast(tuple, params.get("duration_levels_upper"))[0]),
+            int(cast(tuple, params.get("duration_levels_upper"))[1]),
+        )
+    bandwidth_levels_lower = None
+    if params.get("bandwidth_levels_lower") is not None:
+        bandwidth_levels_lower = (
+            cast(tuple, params.get("bandwidth_levels_lower"))[0] * 1000 * 1000,
+            cast(tuple, params.get("bandwidth_levels_lower"))[1] * 1000 * 1000,
+        )
     last_backup = section.get("last_backup")
     if not last_backup:
         yield (