django-chelseru 2.0.2__py3-none-any.whl → 2.0.4__py3-none-any.whl

{django_chelseru-2.0.2.dist-info → django_chelseru-2.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-chelseru
-Version: 2.0.2
+Version: 2.0.4
 Summary: Authentication system, online and real-time chat, SMS & BANK system for Iranian SMS services.
 Home-page: https://pipdjango.chelseru.com
 Author: Sobhan Bahman Rashnu

{django_chelseru-2.0.2.dist-info → django_chelseru-2.0.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-django_chelseru-2.0.2.dist-info/licenses/LICENSE,sha256=VupU5KV4NteHaNQb-WH31G_WZWezxXoomjiCIAHoQJo,1089
+django_chelseru-2.0.4.dist-info/licenses/LICENSE,sha256=VupU5KV4NteHaNQb-WH31G_WZWezxXoomjiCIAHoQJo,1089
 drfchelseru/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 drfchelseru/admin.py,sha256=uamcWPCCw9_dD00XL8Brp8A1NvfhB5FtUfF9DWQCK9c,1022
 drfchelseru/apps.py,sha256=hOTTzFGLXiTPZeN8p_LLcSECLtsR2Q0SUo8zJzgM-qQ,211
@@ -13,7 +13,7 @@ drfchelseru/signals.py,sha256=CI1mZR5RrdOddXBnPC1eMdt9iQkYFDypfgogtfTc2eQ,2714
 drfchelseru/tests.py,sha256=mrbGGRNg5jwbTJtWWa7zSKdDyeB4vmgZCRc2nk6VY-g,60
 drfchelseru/urls.py,sha256=qD3As2gSCEaNDxgU0ejIADLUt9t2EvoaTmqeiV1LRSA,898
 drfchelseru/validators.py,sha256=jAJASfG4kPcqrGxxIcoR8VjS4L4n_EzZuxJtQ1g9qmo,480
-drfchelseru/views.py,sha256=shi3cd19Zxk-nCFr-cK40PtArS3Howu7d6t7x_dxFlE,18079
+drfchelseru/views.py,sha256=KMEfY1HLQB9QK9wvJ-LOEU57BSwe8ZNBDJ9hqNDa9vs,19821
 drfchelseru/migrations/0001_initial.py,sha256=LfkWAeQuPXNW3_igM-w_pyKafCeHQF3aReYgMXfMzEE,1034
 drfchelseru/migrations/0002_otpcode_session_user.py,sha256=c3oyBTQo2kjsKr5r_XQrqaq-66F7pui-4H4mEyAXgJk,3126
 drfchelseru/migrations/0003_rename_mobile_otpcode_mobile_number.py,sha256=eiD7t6etg6nOeT7BhJrSfEcR1yTPePXtATgtyu30520,377
@@ -37,7 +37,7 @@ drfchelseru/migrations/0020_payment_paid_at.py,sha256=H6cr-tfjJVjN3tCsCD8ecg9JGJ
 drfchelseru/migrations/0021_alter_payment_message.py,sha256=g5qIvQ-wecdXGcMf7ZhyfL2u85QpgMUOADHE7t4QHog,397
 drfchelseru/migrations/0022_payment_pay_mode_wallet.py,sha256=241lZITO1jQHqXkJKYm8DKUh7w9rmROh65LzNifP3BA,1521
 drfchelseru/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-django_chelseru-2.0.2.dist-info/METADATA,sha256=XpDHKGThjY6g1xxY19GijeYm8hZTWBjjp1yMEM5F3g4,18242
-django_chelseru-2.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-django_chelseru-2.0.2.dist-info/top_level.txt,sha256=fsaO1F03W3j4AYi0TfDGv5Cjb_Qrh6RSkwkWqfqaMns,12
-django_chelseru-2.0.2.dist-info/RECORD,,
+django_chelseru-2.0.4.dist-info/METADATA,sha256=GsPMOCPssvGFooiGKka7EXRrhnM1n1vVUTZRWWq4o1c,18242
+django_chelseru-2.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+django_chelseru-2.0.4.dist-info/top_level.txt,sha256=fsaO1F03W3j4AYi0TfDGv5Cjb_Qrh6RSkwkWqfqaMns,12
+django_chelseru-2.0.4.dist-info/RECORD,,

drfchelseru/views.py

@@ -13,11 +13,18 @@ from .validators import mobile_number as mobile_validator
 from .serializers import MessageSerializer, OTPCodeSerializer, SessionSerializer, ChatRoomSerializer, MessageChatSerializer, PaymentSerializer
 from .models import User, ChatRoom
 from django.utils.timezone import now, timedelta
-from django.db import transaction
+from django.db import transaction, IntegrityError, OperationalError
 
 from django.conf import settings
 from django.core.exceptions import ImproperlyConfigured
 
+import traceback
+import time
+import logging
+
+
+logger = logging.getLogger(__name__)
+
 
 class MessageSend(APIView):
     permission_classes = (AllowAny, )
@@ -164,18 +171,14 @@ class Authentication(APIView):
         try:
             if 'mobile_number' not in request.data:
                 return Response({'error': 'mobile_number is required.'}, status=HTTP_204_NO_CONTENT)
-            # assert 'mobile_number' in request.data, 'mobile_number is required.'
             mobile_number = request.data['mobile_number']
             if not mobile_number:
                 return Response({'error': 'mobile_number may not be blank.'}, status=HTTP_204_NO_CONTENT)
-            # assert mobile_number, 'mobile_number may not be blank.'
             mobile_number_isvalid = mobile_validator(mobile_number)
             if mobile_number_isvalid != True:
                 return Response({'error': mobile_number_isvalid}, status=HTTP_204_NO_CONTENT)
-            # assert mobile_number_isvalid == True, mobile_number_isvalid
             if 'code' not in request.data:
                 return Response({'error': 'code is required.'}, status=HTTP_204_NO_CONTENT)
-            # assert 'code' in request.data, 'code is required.'
             
             icheck = auth_init_check()
             if icheck and isinstance(icheck, dict) and 'AUTH_SERVICE' in icheck and 'AUTH_METHOD' in icheck:
@@ -185,9 +188,36 @@ class Authentication(APIView):
                     return Response({'error': 'The code sent to this mobile number was not found.'}, status=HTTP_401_UNAUTHORIZED)
                 
                 if otp.check_code():
+                    user = None
                     # login / signup
-                    group = int(request.data['group']) if 'group' in request.data else 0
-                    user, created = User.objects.get_or_create(mobile=mobile_number, group=group)
+                    try:
+                        group = int(request.data.get('group'), 0)
+                    except (ValueError, TypeError):
+                        group = 0
+
+                    try:
+                        with transaction.atomic():
+                            user, created = User.objects.get_or_create(mobile=mobile_number, group=group)
+                    
+                    except IntegrityError as e:
+                        logger.warning("IntegrityError on get_or_create for mobile=%s, group=%s: %s", mobile_number, group, e)
+                        time.sleep(0.05)
+                        try:
+                            user = User.objects.get(mobile=mobile_number, group=group)
+                            created = False
+                        except User.DoesNotExist:
+                            logger.error("After IntegrityError, user still not exists for mobile=%s. Trace: %s", mobile_number, traceback.format_exc())
+                            return Response({'error': 'Server error while creating user.'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+                    except OperationalError as e:
+                        logger.error("OperationalError when accessing DB for mobile=%s: %s\n%s", mobile_number, e, traceback.format_exc())
+                        return Response({'error': 'Database operational error, try again later.'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+                    except Exception as e:
+                        logger.exception("Unexpected error in user get_or_create for mobile=%s: %s", mobile_number, e)
+                        return Response({'error': 'An internal error occurred.'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+                    
+
+                    # user, created = User.objects.get_or_create(mobile=mobile_number, group=group)
                     if user:
                         auth_method = icheck['AUTH_METHOD']
                         auth_service = icheck['AUTH_SERVICE']
@@ -206,9 +236,10 @@ class Authentication(APIView):
                                 raise   
         except AssertionError as e:
             return Response({'error': str(e)}, status=HTTP_204_NO_CONTENT)
-        except:
-            pass
-        return Response({'error': 'An error occurred while generating or sending the otpcode, please contact the www.chelseru.com support team.'}, status=HTTP_500_INTERNAL_SERVER_ERROR)
+        except Exception:
+            logger.exception("Error while creating tokens for user (mobile=%s). user object: %r", mobile_number, user)
+            return Response({'error': 'Token generation failed.'}, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
+        # return Response({'error': 'An error occurred while generating or sending the otpcode, please contact the www.chelseru.com support team.'}, status=HTTP_500_INTERNAL_SERVER_ERROR)
 
 
 class SessionList(ListAPIView):