Actually get it to work, including self

stefankeidel · Jan 5, 2025 · 688ca8e · 688ca8e
1 parent 12f32b9
commit 688ca8e
Show file tree

Hide file tree

Showing 3 changed files with 53 additions and 2 deletions.
diff --git a/kirkbot/tasks.py b/kirkbot/tasks.py
@@ -3,6 +3,7 @@
 import abc
 import subprocess
 import json
+from glom import glom
 from datetime import datetime, timedelta
 
 
@@ -49,9 +50,13 @@ async def run(self):
             data = json.loads(result.stdout)
             now = datetime.utcnow()
 
+            # devices, including self -- weird structure in JSON
+            devices = glom(data, 'Peer') # type: ignore
+            devices['self'] = glom(data, 'Self') # type: ignore
+
             # Check for key expiry in the output
             # if less than one month, then alert
-            for device in data.get("Peer", {}).values():
+            for device in devices.values():
                 if "KeyExpiry" in device:
                     expiry_date = datetime.strptime(device["KeyExpiry"], "%Y-%m-%dT%H:%M:%SZ")
 

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -11,6 +11,7 @@ python = ">=3.11,<3.13"
 click = "^8.1.8"
 discord-py = "^2.4.0"
 psutil = "^6.1.1"
+glom = "^24.11.0"
 
 
 [tool.poetry.group.dev.dependencies]