upgraded authentication for all domain.

Author: ajitmourya7

MANIFEST.in

@@ -0,0 +1,2 @@
+recursive-include django_single_auth/templates *
+recursive-include django_single_auth/static *

build/lib/django_single_auth/__init__.py

@@ -0,0 +1,3 @@
+from django.contrib import admin
+
+# Register your models here.

build/lib/django_single_auth/admin.py

@@ -0,0 +1,5 @@
+from django.apps import AppConfig
+
+
+class DjangoSingleAuthConfig(AppConfig):
+    name = 'django_single_auth'

build/lib/django_single_auth/apps.py

@@ -0,0 +1,16 @@
+from django.utils.deprecation import MiddlewareMixin
+from .server import session
+import urllib.parse as urlparse
+from urllib.parse import parse_qs
+
+
+class SameDomainServer(MiddlewareMixin):
+
+    def process_response(self, request, response):
+        try:
+            parsed = urlparse.urlparse(response.url)
+            if parse_qs(parsed.query).get('token') is not None:
+                session.update(request)
+        except:
+            pass
+        return response

build/lib/django_single_auth/middleware.py

@@ -0,0 +1,3 @@
+from django.db import models
+
+# Create your models here.

build/lib/django_single_auth/migrations/__init__.py

@@ -0,0 +1,25 @@
+from django.conf import settings
+from django.contrib.sessions.models import Session
+from django.db import transaction
+
+
+def update(request):
+    try:
+        if request.user.is_authenticated and request.session.session_key is not None:
+            with transaction.atomic():
+                # print(request.session.session_data)
+                master_session = Session.objects.get(session_key=str(request.session.session_key))
+                print(master_session.session_key)
+                for i in settings.DATABASES:
+                    if i != 'default':
+                        try:
+                            sync_session = Session.objects.using(i).get(session_key=master_session.session_key)
+                            sync_session.session_data = master_session.session_data
+                            sync_session.expire_date = master_session.expire_date
+                            sync_session.save()
+                        except Session.objects.using(i).model.DoesNotExist:
+                            Session.objects.using(i).create(session_key=master_session.session_key,
+                                                            session_data=master_session.session_data,
+                                                            expire_date=master_session.expire_date)
+    except:
+        print('already have session key')

build/lib/django_single_auth/models.py

@@ -0,0 +1,38 @@
+from cryptography.fernet import Fernet
+import base64
+import logging
+import traceback
+from django.conf import settings
+
+
+# this is your "password/ENCRYPT_KEY". keep it in settings.py file
+# key = Fernet.generate_key()
+
+def encrypt(txt):
+    try:
+        # convert integer etc to string first
+        txt = str(txt)
+        # get the key from settings
+        cipher_suite = Fernet(settings.ENCRYPT_KEY)  # key should be byte
+        # #input should be byte, so convert the text to byte
+        encrypted_text = cipher_suite.encrypt(txt.encode('ascii'))
+        # encode to urlsafe base64 format
+        encrypted_text = base64.urlsafe_b64encode(encrypted_text).decode("ascii")
+        return encrypted_text
+    except Exception as e:
+        # log the error if any
+        logging.getLogger("error_logger").error(traceback.format_exc())
+        return None
+
+
+def decrypt(txt):
+    try:
+        # base64 decode
+        txt = base64.urlsafe_b64decode(txt)
+        cipher_suite = Fernet(settings.ENCRYPT_KEY)
+        decoded_text = cipher_suite.decrypt(txt).decode("ascii")
+        return decoded_text
+    except Exception as e:
+        # log the error
+        logging.getLogger("error_logger").error(traceback.format_exc())
+        return None