This commit is contained in:
Egor Matveev 2022-05-09 08:57:33 +03:00
parent b7747ef60b
commit bb3a153ced

View File

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