auth added

Author: Bishwas-py

djapy/core/auth/__init__.py

@@ -0,0 +1,9 @@
+__all__ = [
+    "BaseAuthMechanism",
+    "SessionAuth",
+    "djapy_auth",
+    "djapy_method"
+]
+
+from djapy.core.auth.auth import BaseAuthMechanism, SessionAuth
+from djapy.core.auth.dec import djapy_auth, djapy_method

djapy/core/auth/__pycache__/__init__.cpython-311.pyc

@@ -0,0 +1,58 @@
+from django.http import HttpRequest
+
+
+class BaseAuthMechanism:
+    def __init__(self, *args, **kwargs):
+        self.message_response = None
+
+    def authenticate(self, request: HttpRequest, *args, **kwargs):
+        pass
+
+    def authorize(self, request: HttpRequest, *args, **kwargs):
+        pass
+
+    def schema(self):
+        return {}
+
+    def required_headers(self):
+        pass
+
+    def app_schema(self):
+        return {}
+
+    def set_message_response(self, message_response: dict):
+        self.message_response = message_response
+
+
+class SessionAuth(BaseAuthMechanism):
+    def __init__(self, *args, **kwargs):
+        self.message_response = {"message": "You are not authenticated"}
+        super().__init__(*args, **kwargs)
+
+    def authenticate(self, request: HttpRequest, *args, **kwargs):
+        return self.message_response
+
+    def required_headers(self):
+        return [
+            "Cookie"
+        ]
+
+    def schema(self):
+        return {
+            "SessionAuth": {
+                "type": "apiKey",
+                "in": "cookie",
+                "name": "sessionid"
+            },
+            "CSRFTokenAuth": {
+                "type": "apiKey",
+                "in": "cookie",
+                "name": "csrftoken"
+            }
+        }
+
+    def app_schema(self):
+        return {
+            "SessionAuth": [],
+            "CSRFTokenAuth": []
+        }

djapy/core/auth/__pycache__/auth.cpython-311.pyc

@@ -0,0 +1,50 @@
+import inspect
+import json
+from functools import wraps
+from typing import Dict, Callable, List, Type
+
+from django.http import HttpRequest, JsonResponse
+
+from . import BaseAuthMechanism
+from ..defaults import ALLOW_METHODS_LITERAL, DEFAULT_METHOD_NOT_ALLOWED_MESSAGE
+
+
+def djapy_auth(auth_mechanism_class: Type[BaseAuthMechanism],
+               permissions: List[str] = None) -> Callable:
+    def decorator(view_func):
+        @wraps(view_func)
+        def _wrapped_view(request: HttpRequest, *args, **kwargs):
+            return view_func(request, *args, **kwargs)
+
+        _wrapped_view.auth_mechanism = auth_mechanism_class()
+        _wrapped_view.permissions = permissions
+        return _wrapped_view
+
+    if inspect.isfunction(auth_mechanism_class):
+        return decorator(auth_mechanism_class)
+
+    return decorator
+
+
+def djapy_method(allowed_method_or_list: ALLOW_METHODS_LITERAL | List[ALLOW_METHODS_LITERAL],
+                 message_response: Dict[str, str] = None) -> Callable:
+    message_response = message_response or DEFAULT_METHOD_NOT_ALLOWED_MESSAGE
+    try:
+        json.dumps(message_response)
+    except TypeError as e:
+        raise TypeError(f"Invalid message_response: {message_response}") from e
+
+    def decorator(view_func):
+        @wraps(view_func)
+        def _wrapped_view(request: HttpRequest, *args, **kwargs):
+            if request.method not in allowed_method_or_list:
+                view_func.djapy_message_response = message_response
+            return view_func(request, *args, **kwargs)
+
+        if isinstance(allowed_method_or_list, str):
+            _wrapped_view.djapy_allowed_method = [allowed_method_or_list]
+        else:
+            _wrapped_view.djapy_allowed_method = allowed_method_or_list
+        return _wrapped_view
+
+    return decorator