process exception for redis connection

abonapp/views.py

@@ -1,5 +1,6 @@
 from datetime import datetime
 from typing import Dict, Optional
+from kombu.exceptions import OperationalError
 
 from abonapp.tasks import customer_nas_command, customer_nas_remove
 from agent.commands.dhcp import dhcp_commit, dhcp_expiry, dhcp_release
@@ -194,7 +195,7 @@ class DelAbonDeleteView(LoginAdminMixin, PermissionRequiredMixin, DeleteView):
                 }).strip())
             messages.success(request, _('delete abon success msg'))
             return redirect('abonapp:people_list', gid=gid)
-        except NasNetworkError as e:
+        except (NasNetworkError, OperationalError) as e:
             messages.error(self.request, e)
         except NasFailedResult as e:
             messages.error(self.request, _("NAS says: '%s'") % e)
@@ -332,7 +333,11 @@ class AbonHomeUpdateView(LoginAdminMixin, PermissionRequiredMixin, UpdateView):
     def form_valid(self, form):
         r = super(AbonHomeUpdateView, self).form_valid(form)
         abon = self.object
+        try:
             customer_nas_command.delay(abon.pk, 'sync')
+        except OperationalError as e:
+            messages.error(self.request, str(e))
+        else:
             messages.success(self.request, _('edit abon success msg'))
         return r
 
@@ -442,9 +447,7 @@ def pick_tariff(request, gid: int, uname):
             messages.success(request, _('Tariff has been picked'))
             return redirect('abonapp:abon_services', gid=gid,
                             uname=abon.username)
-    except (lib.LogicError, NasFailedResult) as e:
-        messages.error(request, e)
-    except NasNetworkError as e:
+    except (lib.LogicError, NasFailedResult, NasNetworkError, OperationalError) as e:
         messages.error(request, e)
         return redirect('abonapp:abon_services', gid=gid, uname=abon.username)
     except Tariff.DoesNotExist:
@@ -485,7 +488,7 @@ def unsubscribe_service(request, gid: int, uname, abon_tariff_id: int):
         messages.success(request, _('User has been detached from service'))
     except NasFailedResult as e:
         messages.error(request, e)
-    except NasNetworkError as e:
+    except (NasNetworkError, OperationalError) as e:
         messages.warning(request, e)
     except lib.MultipleException as errs:
         for err in errs.err_list:
@@ -608,7 +611,11 @@ class IpUpdateView(LoginAdminPermissionMixin, UpdateView):
     def form_valid(self, form):
         r = super(IpUpdateView, self).form_valid(form)
         abon = self.object
+        try:
             customer_nas_command.delay(abon.pk, 'sync')
+        except OperationalError as e:
+            messages.error(self.request, str(e))
+        else:
             messages.success(self.request, _('Ip successfully updated'))
         return r
 
@@ -1193,12 +1200,15 @@ class UserSessionFree(LoginRequiredMixin, OnlyAdminsMixin, PermissionRequiredMix
     def post(self, request, *args, **kwargs):
         self.object = self.get_object()
         abon = self.object
+        try:
             if abon.ip_address:
                 abon.free_ip_addr()
                 customer_nas_command.delay(abon.pk, 'remove')
                 messages.success(request, _('Ip lease has been freed'))
             else:
                 messages.error(request, _('User not have ip'))
+        except OperationalError as e:
+            messages.error(request, e)
         return redirect(
             'abonapp:abon_home',
             gid=self.kwargs.get('gid'),

clientsideapp/views.py

@@ -1,3 +1,4 @@
+from kombu.exceptions import OperationalError
 from django.contrib.auth.decorators import login_required
 from django.shortcuts import render, get_object_or_404, redirect
 from django.contrib import messages
@@ -74,6 +75,8 @@ def buy_service(request, srv_id):
         messages.error(request, e)
     except NasFailedResult as e:
         messages.error(request, e)
+    except OperationalError:
+        pass
     return redirect('client_side:services')
 
 

devapp/views.py

@@ -1,6 +1,8 @@
 import re
 from ipaddress import ip_address
 
+from kombu.exceptions import OperationalError
+
 from django.conf import settings
 from django.contrib import messages
 from django.contrib.auth.decorators import login_required
@@ -93,7 +95,7 @@ class DeviceDeleteView(LoginAdminPermissionMixin, DeleteView):
             onu_register.delay(
                 tuple(dev.pk for dev in Device.objects.exclude(group=None).only('pk').iterator())
             )
-        except (DeviceDBException, PermissionError) as e:
+        except (DeviceDBException, PermissionError, OperationalError) as e:
             messages.error(request, e)
         messages.success(request, _('Device successfully deleted'))
         return res
@@ -152,7 +154,7 @@ class DeviceUpdate(LoginAdminPermissionMixin, UpdateView):
                 tuple(dev.pk for dev in Device.objects.exclude(group=None).only('pk').iterator())
             )
             messages.success(self.request, _('Device info has been saved'))
-        except PermissionError as e:
+        except (PermissionError, OperationalError) as e:
             messages.error(self.request, e)
         return r
 
@@ -226,7 +228,7 @@ class DeviceCreateView(LoginAdminMixin, PermissionRequiredMixin, CreateView):
                 tuple(dev.pk for dev in Device.objects.exclude(group=None).only('pk').iterator())
             )
             messages.success(self.request, _('Device info has been saved'))
-        except PermissionError as e:
+        except (PermissionError, OperationalError) as e:
             messages.error(self.request, e)
         return r
 
@@ -779,7 +781,7 @@ class OnDeviceMonitoringEvent(global_base_views.SecureApiView):
             return {
                 'text': 'notification successfully sent'
             }
-        except ValueError as e:
+        except (ValueError, OperationalError) as e:
             return {
                 'text': str(e)
             }

msg_app/models.py

@@ -1,3 +1,5 @@
+from kombu.exceptions import OperationalError
+
 from django.db import models
 from django.utils.translation import gettext_lazy as _
 from accounts_app.models import UserProfile
@@ -206,6 +208,7 @@ class Conversation(models.Model):
             text=text, conversation=self,
             attachment=attachment, author=author
         )
+        try:
             if with_status:
                 for participant in self.participants.filter(is_active=True):
                     if participant == author:
@@ -217,6 +220,8 @@ class Conversation(models.Model):
                             account_id=participant.pk
                         )
             return msg
+        except OperationalError as e:
+            raise MessageError(e)
 
     @staticmethod
     def remove_message(msg):

taskapp/handle.py

@@ -1,4 +1,5 @@
-# -*- coding: utf-8 -*-
+from kombu.exceptions import OperationalError
+
 from django.template.loader import render_to_string
 from django.utils.translation import gettext as _
 from djing.tasks import send_email_notify # , multicast_email_notify
@@ -31,6 +32,7 @@ def handle(task, author, recipients):
         'task_status': task_status
     })
 
+    try:
         if task.state == 'F' or task.state == 'C':
             # If task completed or failed than send one message to author
             send_email_notify.delay(fulltext, author.pk)
@@ -38,3 +40,5 @@ def handle(task, author, recipients):
         else:
             #multicast_email_notify.delay(fulltext, profile_ids)
             multicast_viber_notify.delay(None, profile_ids, fulltext)
+    except OperationalError as e:
+        raise TaskException(e)