lightning-pose-app 1.8.1a5__py3-none-any.whl → 1.8.1a6__py3-none-any.whl

{lightning_pose_app-1.8.1a5.dist-info → lightning_pose_app-1.8.1a6.dist-info}/METADATA

@@ -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.dist-info → lightning_pose_app-1.8.1a6.dist-info}/RECORD

@@ -21,9 +21,9 @@ litpose_app/routes/files.py,sha256=HHyug226s_M4jMi8q7oI2dTWHTKu_J_c7vIbG0rdk9s,3
 litpose_app/routes/project.py,sha256=pQW7zFdGHMy_eE0Z1bIXs9QLkV1-_MwFHsZR51-H6uQ,1957
 litpose_app/routes/transcode.py,sha256=0p2GmQntbSDlBxhXiH9AeDXs44uRxfa1eiVNqt0LjUk,4848
 litpose_app/tasks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-litpose_app/tasks/management.py,sha256=jAit5aa8Jckj_3qbaRH_1Voe1aLkxebpU2ulG87U7wI,600
+litpose_app/tasks/management.py,sha256=Z_XYJsy4Mlz_0tQwITsnkPk-ugkMFa4ek6NcFrhTH74,831
 litpose_app/tasks/transcode_fine.py,sha256=bX8OblwMO_xaXXqZXJmmQPprCYIpPGHiboyEJli6FHs,209
 litpose_app/transcode_fine.py,sha256=YVrKg6IxrLEi0mWVWQqt89NHD6IAwYS512mw-4zkp54,6188
-lightning_pose_app-1.8.1a5.dist-info/METADATA,sha256=BGAX4WhTfKrYl4ZR3whh_m2T-WjLxJ6G-uN8fBSdhSw,945
-lightning_pose_app-1.8.1a5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-lightning_pose_app-1.8.1a5.dist-info/RECORD,,
+lightning_pose_app-1.8.1a6.dist-info/METADATA,sha256=BNsYWKmm_MrPjv7TS4NIfkM1DpAkhSHvwgIh9G4fejo,945
+lightning_pose_app-1.8.1a6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+lightning_pose_app-1.8.1a6.dist-info/RECORD,,

litpose_app/tasks/management.py

@@ -9,12 +9,25 @@ def setup_active_task_registry(app: FastAPI):
     # Get APScheduler instance
     scheduler: BaseScheduler = app.state.scheduler
 
-    subject = reactivex.subject.BehaviorSubject(len(scheduler.get_jobs()))
+    subject = reactivex.subject.BehaviorSubject(0)
 
     app.state.num_active_transcode_tasks = subject
 
+    _num_active_jobs = 0
+
     def my_listener(event):
-        jobs = scheduler.get_jobs()
-        subject.on_next(len(jobs))
+        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_ADDED | e.EVENT_JOB_REMOVED)
+    scheduler.add_listener(
+        my_listener,
+        e.EVENT_JOB_SUBMITTED
+        | e.EVENT_JOB_EXECUTED
+        | e.EVENT_JOB_ERROR
+        | e.EVENT_JOB_MISSED,
+    )