Fix database connection closed (#4111)

kartoza · Jul 25, 2024 · 30f6eba · 30f6eba
1 parent f77cf86
commit 30f6eba
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 1 deletion.
diff --git a/bims/tasks/email_csv.py b/bims/tasks/email_csv.py
@@ -3,6 +3,7 @@
 import zipfile
 
 from celery import shared_task
+from django.db import DatabaseError, connection, InterfaceError
 from openpyxl import load_workbook
 
 from bims.utils.domain import get_current_domain
@@ -49,7 +50,12 @@ def send_csv_via_email(
  if download_request and not download_request.request_file:
  download_request.request_category = file_name
  download_request.request_file = csv_file
- download_request.save()
+ try:
+ download_request.save()
+ except (DatabaseError, InterfaceError):
+ # Attempt to reconnect and save again
+ connection.connect()
+ download_request.save()
 
  email_template = 'csv_download/csv_created'
  if download_request and download_request.resource_type == DownloadRequest.PDF:

diff --git a/core/settings/contrib.py b/core/settings/contrib.py
@@ -40,6 +40,7 @@
  'grappelli',
  'django_tenants',
  'tenants',
+ 'django_dbconn_retry',
 
  'colorfield',
  'polymorphic',

diff --git a/deployment/docker/REQUIREMENTS.txt b/deployment/docker/REQUIREMENTS.txt
@@ -97,3 +97,4 @@ celery==5.3.6
 django-celery-results==2.5.1
 django-tenants==3.6.1
 tenant-schemas-celery==2.2.0
+django-dbconn-retry==0.1.7
diff --git a/deployment/production/REQUIREMENTS.txt b/deployment/production/REQUIREMENTS.txt
@@ -97,3 +97,4 @@ celery==5.3.6
 django-celery-results==2.5.1
 django-tenants==3.6.1
 tenant-schemas-celery==2.2.0
+django-dbconn-retry==0.1.7