lightning-pose-app 1.8.1a5__tar.gz → 1.8.1a6__tar.gz

{lightning_pose_app-1.8.1a5 → lightning_pose_app-1.8.1a6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: lightning-pose-app
-Version: 1.8.1a5
+Version: 1.8.1a6
 Summary: 
 Requires-Python: >=3.10
 Classifier: Programming Language :: Python :: 3

{lightning_pose_app-1.8.1a5 → lightning_pose_app-1.8.1a6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "lightning-pose-app"
-version = "1.8.1a5"
+version = "1.8.1a6"
 requires-python = ">=3.10"
 dependencies = [
     "apscheduler~=3.11.0",

lightning_pose_app-1.8.1a6/src/litpose_app/tasks/management.py

@@ -0,0 +1,33 @@
+import reactivex
+import reactivex.subject
+from apscheduler import events as e
+from apscheduler.schedulers.base import BaseScheduler
+from fastapi import FastAPI
+
+
+def setup_active_task_registry(app: FastAPI):
+    # Get APScheduler instance
+    scheduler: BaseScheduler = app.state.scheduler
+
+    subject = reactivex.subject.BehaviorSubject(0)
+
+    app.state.num_active_transcode_tasks = subject
+
+    _num_active_jobs = 0
+
+    def my_listener(event):
+        nonlocal _num_active_jobs
+        if event.code == e.EVENT_JOB_SUBMITTED:
+            _num_active_jobs += 1
+        else:
+            _num_active_jobs -= 1
+
+        subject.on_next(_num_active_jobs)
+
+    scheduler.add_listener(
+        my_listener,
+        e.EVENT_JOB_SUBMITTED
+        | e.EVENT_JOB_EXECUTED
+        | e.EVENT_JOB_ERROR
+        | e.EVENT_JOB_MISSED,
+    )

lightning_pose_app-1.8.1a5/src/litpose_app/tasks/management.py

@@ -1,20 +0,0 @@
-import reactivex
-import reactivex.subject
-from apscheduler import events as e
-from apscheduler.schedulers.base import BaseScheduler
-from fastapi import FastAPI
-
-
-def setup_active_task_registry(app: FastAPI):
-    # Get APScheduler instance
-    scheduler: BaseScheduler = app.state.scheduler
-
-    subject = reactivex.subject.BehaviorSubject(len(scheduler.get_jobs()))
-
-    app.state.num_active_transcode_tasks = subject
-
-    def my_listener(event):
-        jobs = scheduler.get_jobs()
-        subject.on_next(len(jobs))
-
-    scheduler.add_listener(my_listener, e.EVENT_JOB_ADDED | e.EVENT_JOB_REMOVED)