From dff97d90673f65328331d2d4bc7ae87a2453c038 Mon Sep 17 00:00:00 2001
From: SergeantPanda <chester.cullen@gmail.com>
Date: Tue, 7 Jul 2026 18:26:09 +0000
Subject: [PATCH] Enhancement: Ensure proper DB connection management in Celery
 tasks by checking for worker context before releasing connections, preventing
 potential ORM errors.

---
 apps/epg/tasks.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/apps/epg/tasks.py b/apps/epg/tasks.py
index 612e9804..6e742c63 100644
--- a/apps/epg/tasks.py
+++ b/apps/epg/tasks.py
@@ -36,6 +36,7 @@ from core.utils import (
     send_websocket_update,
     cleanup_memory,
     log_system_event,
+    _is_celery_worker_context,
 )
 
 logger = logging.getLogger(__name__)
@@ -100,6 +101,8 @@ _NON_TERMINAL_REFRESH_STATUSES = frozenset({
 
 def _release_task_db_connection():
     """Return the Celery worker's DB connection to a clean state after ORM errors."""
+    if not _is_celery_worker_context():
+        return
     from django.db import close_old_connections
     close_old_connections()