JDM170/spotizerr-dev
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

I am still unworthy of SSE...

Browse Source
This commit is contained in:
cool.gitter.not.me.again.duh
2025-05-29 14:45:29 -06:00
parent 03f132a9f3
commit ba33e10afc
2 changed files with 180 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
routes/utils/celery_tasks.py
View File

@@ -440,6 +440,17 @@ class ProgressTrackingTask(Task):
# Store the processed status update
store_task_status(task_id, stored_data)
# Immediately delete task info from Redis after marking as complete
if stored_data.get("status") == ProgressState.COMPLETE:
logger.info(f"Task {task_id} completed. Deleting task data from Redis.")
try:
redis_client.delete(f"task:{task_id}:info")
redis_client.delete(f"task:{task_id}:status")
redis_client.delete(f"task:{task_id}:status:next_id") # Also delete the counter
logger.info(f"Successfully deleted Redis data for completed task {task_id}.")
except Exception as e:
logger.error(f"Error deleting Redis data for completed task {task_id}: {e}", exc_info=True)
def _handle_initializing(self, task_id, data, task_info):
"""Handle initializing status from deezspot"""
# Extract relevant fields
@@ -785,6 +796,20 @@ class ProgressTrackingTask(Task):
data["status"] = ProgressState.COMPLETE
data["message"] = "Download complete"
# Store the processed status update
store_task_status(task_id, data)
# Immediately delete task info from Redis after marking as complete
if data.get("status") == ProgressState.COMPLETE:
logger.info(f"Task {task_id} ({task_info.get('name', 'Unknown')}) completed. Deleting task data from Redis.")
try:
redis_client.delete(f"task:{task_id}:info")
redis_client.delete(f"task:{task_id}:status")
redis_client.delete(f"task:{task_id}:status:next_id") # Also delete the counter
logger.info(f"Successfully deleted Redis data for completed task {task_id}.")
except Exception as e:
logger.error(f"Error deleting Redis data for completed task {task_id}: {e}", exc_info=True)
# Celery signal handlers
@task_prerun.connect
def task_prerun_handler(task_id=None, task=None, *args, **kwargs):