Post Black Linting

models/agent.py

@@ -38,23 +38,19 @@ class Agent:
     status_text: Optional[str] = field(default=None)
 
     # Class-level status map
-    status_map: ClassVar[dict] = {
-        0: "Offline",
-        1: "Online",
-        2: "Hidden",
-        3: "Safemode"
-    }
-
+    status_map: ClassVar[dict] = {0: "Offline", 1: "Online", 2: "Hidden", 3: "Safemode"}
 
     def enrich_with_policies(self, policies: List[Policy]):
-            """Enrich the agent with groupname and human-readable status."""
-            self.status_text = self.status_map.get(self.status, "Unknown")
-            for policy in policies:
-                if policy.groupid == self.groupid:
-                    self.groupname = policy.name
-                    break
-            if not self.groupname:
-                self.groupname = "Unknown"
+        """Enrich the agent with groupname and human-readable status."""
+        self.status_text = self.status_map.get(self.status, "Unknown")
+        for policy in policies:
+            if policy.groupid == self.groupid:
+                self.groupname = policy.name
+                break
+        if not self.groupname:
+            self.groupname = "Unknown"
+
+
 """
 
 from models.agent import Agent