Post Black Linting

Server/scheduler_async.py

@@ -30,6 +30,7 @@ scheduled_jobs: Dict[str, asyncio.TimerHandle] = {}
 # Path to the JSON file for job persistence TODO - pin this to the correct place
 JOBS_FILE = os.path.join(os.getcwd(), "jobs.json")
 
+
 def register_function(name: str, func: Callable):
     """
     Register a function so it can be called by name later.
@@ -38,6 +39,7 @@ def register_function(name: str, func: Callable):
     """
     FUNCTION_MAP[name] = func
 
+
 def load_jobs() -> List[Dict[str, Any]]:
     """
     Load jobs from the JSON file, or return [] if none exist.
@@ -47,6 +49,7 @@ def load_jobs() -> List[Dict[str, Any]]:
     with open(JOBS_FILE, "r") as f:
         return json.load(f)
 
+
 def save_jobs(jobs: List[Dict[str, Any]]):
     """
     Save jobs to the JSON file (overwrite).
@@ -54,6 +57,7 @@ def save_jobs(jobs: List[Dict[str, Any]]):
     with open(JOBS_FILE, "w") as f:
         json.dump(jobs, f, indent=4)
 
+
 def cancel_job(job_id: str):
     """
     Cancel a scheduled job by ID and remove it from the registry and persistence.
@@ -66,7 +70,15 @@ def cancel_job(job_id: str):
     jobs = [j for j in load_jobs() if j.get("id") != job_id]
     save_jobs(jobs)
 
-def run_once_job(job_id: str, func_name: str, delay_seconds: float, args=None, kwargs=None, persist=True):
+
+def run_once_job(
+    job_id: str,
+    func_name: str,
+    delay_seconds: float,
+    args=None,
+    kwargs=None,
+    persist=True,
+):
     """
     Schedule a job to run once after a delay (in seconds).
     """
@@ -87,18 +99,25 @@ def run_once_job(job_id: str, func_name: str, delay_seconds: float, args=None, k
 
     if persist:
         jobs = [j for j in load_jobs() if j.get("id") != job_id]
-        jobs.append({
-            "id": job_id,
-            "type": "once",
-            "delay": delay_seconds,
-            "function": func_name,
-            "args": args,
-            "kwargs": kwargs
-        })
+        jobs.append(
+            {
+                "id": job_id,
+                "type": "once",
+                "delay": delay_seconds,
+                "function": func_name,
+                "args": args,
+                "kwargs": kwargs,
+            }
+        )
         save_jobs(jobs)
-        logger.info(f"Scheduled one-time job '{job_id}' to run in {delay_seconds} seconds.")
+        logger.info(
+            f"Scheduled one-time job '{job_id}' to run in {delay_seconds} seconds."
+        )
 
-def recurring_job(job_id: str, func_name: str, interval: float, args=None, kwargs=None, persist=True):
+
+def recurring_job(
+    job_id: str, func_name: str, interval: float, args=None, kwargs=None, persist=True
+):
     """
     Schedule a recurring job.
     """
@@ -121,17 +140,20 @@ def recurring_job(job_id: str, func_name: str, interval: float, args=None, kwarg
 
     if persist:
         jobs = [j for j in load_jobs() if j.get("id") != job_id]
-        jobs.append({
-            "id": job_id,
-            "type": "recurring",
-            "interval": interval,
-            "function": func_name,
-            "args": args,
-            "kwargs": kwargs
-        })
+        jobs.append(
+            {
+                "id": job_id,
+                "type": "recurring",
+                "interval": interval,
+                "function": func_name,
+                "args": args,
+                "kwargs": kwargs,
+            }
+        )
         save_jobs(jobs)
         logger.info(f"Scheduled recurring job '{job_id}' every {interval} seconds.")
 
+
 def reload_jobs():
     """
     Reload jobs from JSON and reschedule them.
@@ -145,7 +167,7 @@ def reload_jobs():
                 job["delay"],
                 job.get("args"),
                 job.get("kwargs"),
-                persist=False
+                persist=False,
             )
         elif job["type"] == "recurring":
             recurring_job(
@@ -154,9 +176,10 @@ def reload_jobs():
                 job["interval"],
                 job.get("args"),
                 job.get("kwargs"),
-                persist=False
+                persist=False,
             )
 
+
 async def start_scheduler():
     """
     Start the asynchronous scheduler loop.
@@ -189,4 +212,4 @@ async def start_scheduler():
         while True:
             await asyncio.sleep(3600)  # Sleep indefinitely; jobs run via call_later
     except asyncio.CancelledError:
-        logger.critical("Scheduler stopped.")
+        logger.critical("Scheduler stopped.")