Create policy/kill_switch.py

policy/kill_switch.py

@@ -1,26 +1,9 @@
-import time
-
 class KillSwitch:
     def __init__(self):
         self.disabled = {}
-        self.cooldown = 3600  # 1 hour
 
     def disable(self, agency, reason):
-        self.disabled[agency] = {
-            "reason": reason,
-            "time": time.time()
-        }
-
-    def is_enabled(self, agency):
-        if agency not in self.disabled:
-            return True
-
-        elapsed = time.time() - self.disabled[agency]["time"]
-        if elapsed > self.cooldown:
-            del self.disabled[agency]
-            return True
-
-        return False
+        self.disabled[agency] = reason
 
-    def reason(self, agency):
-        return self.disabled.get(agency, {}).get("reason", "")
+    def enabled(self, agency):
+        return agency not in self.disabled