From bb3a153ced0828c6c0fd4dec2fdb672db0d88990 Mon Sep 17 00:00:00 2001
From: Egor Matveev <emmtvv@icloud.com>
Date: Mon, 9 May 2022 08:57:33 +0300
Subject: [PATCH] queue

---
 SprintLib/queue.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/SprintLib/queue.py b/SprintLib/queue.py
index d058d49..89400bb 100644
--- a/SprintLib/queue.py
+++ b/SprintLib/queue.py
@@ -3,7 +3,8 @@ import json
 import pika
 from django.contrib.auth.models import User
 from django.core.management import BaseCommand
-from psycopg2 import OperationalError
+import psycopg2
+import django.db
 from pika.adapters.utils.connection_workflow import AMQPConnectorException
 
 from Sprint import settings
@@ -34,7 +35,7 @@ class MessagingSupport(BaseCommand):
         try:
             self.process(data)
             print("Process finished successfully")
-        except OperationalError:
+        except (psycopg2.OperationalError, django.db.OperationalError):
             print("Failed to connect to database, restarting...")
             send_to_queue(self.queue_name, data)
             raise