fixed pyright lint errors

Author: CaptainAril

pyproject.toml

@@ -19,6 +19,7 @@ dependencies = [
   "pydantic[email]>=2.10.6",
   "django-ninja>=1.3.0",
   "pyjwt>=2.10.1",
+  "typing>=3.10.0.0",
 ]
 
 [dependency-groups]

src/api/middlewares/GateWayMiddleware.py

@@ -1,11 +1,9 @@
 import hmac
-import base64
 import hashlib
-from http import HTTPStatus
-from datetime import datetime
+from datetime import datetime, timedelta
 
 from django.http import HttpRequest
-from ninja.errors import HttpError
+from ninja.errors import AuthenticationError
 from ninja.security import APIKeyHeader
 from ninja.openapi.schema import OpenAPISchema
 
@@ -18,23 +16,24 @@ def __init__(self, logger: Logger) -> None:
         self.logger = logger
         super().__init__()
 
-    def authenticate(self, request: HttpRequest, key: str) -> str | None:
-        api_key = request.headers.get("X-API-GATEWAY-KEY")
-        api_timestamp = request.headers.get("X-API-GATEWAY-TIMESTAMP")
-        api_signature = request.headers.get("X-API-GATEWAY-SIGNATURE")
-        user_id = request.headers.get("X-USER-ID")
-        user_email = request.headers.get("X-USER-EMAIL")
+    def authenticate(self, request: HttpRequest, key: str | None ) -> str | None:
 
-        if not all([api_key, api_timestamp, api_signature, user_id, user_email]):
-            message = "Missing required headers!"
+        try:
+            api_key = request.headers["X-API-GATEWAY-KEY"]
+            api_timestamp = request.headers["X-API-GATEWAY-TIMESTAMP"]
+            api_signature = request.headers["X-API-GATEWAY-SIGNATURE"]
+            user_id = request.headers["X-USER-ID"]
+            user_email = request.headers["X-USER-EMAIL"]
+        except KeyError as e:
+            message = f"Missing required header: {e}"
             self.logger.error(
                 {
                     "activity_type": "Authenticate User",
                     "message": message,
                     "metadata": {"headers": request.headers},
                 }
             )
-            raise HttpError(HTTPStatus.UNAUTHORIZED, message)
+            raise AuthenticationError(message=message)
 
         valid_api_key = api_gateway["key"]
         if api_key != valid_api_key:
@@ -46,9 +45,9 @@ def authenticate(self, request: HttpRequest, key: str) -> str | None:
                     "metadata": {"headers": request.headers},
                 }
             )
-            raise HttpError(HTTPStatus.UNAUTHORIZED, message)
+            raise AuthenticationError(message=message)
 
-        self._verify_signature(valid_api_key, valid_api_key, api_timestamp)
+        self._verify_signature(valid_api_key, api_signature, api_timestamp)
 
         self.logger.debug(
             {
@@ -80,13 +79,12 @@ def _verify_signature(
                     "metadata": {"signature": signature},
                 }
             )
-            raise HttpError(HTTPStatus.UNAUTHORIZED, message)
+            raise AuthenticationError(message=message)
 
-        timestamp = int(timestamp)
-        current_time = datetime.now().timestamp() * 1000
-        lifespan = api_gateway["expires_at"] * 60 * 60 * 1000
-
-        if abs(current_time - timestamp) > lifespan:
+        initial_time = datetime.fromtimestamp(int(timestamp) / 1000)
+        valid_window = initial_time + timedelta(minutes=api_gateway["expires_at"])
+        if valid_window < datetime.now():
+        
             message = "Signature expired!"
             self.logger.error(
                 {
@@ -95,15 +93,15 @@ def _verify_signature(
                     "metadata": {"timestamp": timestamp},
                 }
             )
-            raise HttpError(HTTPStatus.UNAUTHORIZED, message)
+            raise AuthenticationError(message=message)
 
         return True
 
     def generate_signature(self, api_key: str, timestamp: str) -> str:
         signature = hmac.new(
             key=api_key.encode(), msg=timestamp.encode(), digestmod=hashlib.sha256
-        ).digest()
-        return base64.b64encode(signature).decode("utf-8")
+        ).hexdigest()
+        return signature
 
 
 def get_authentication() -> GateWayAuth:

src/api/routes/__init__.py

@@ -1,5 +1,3 @@
-from typing import _T
-
 from ninja import NinjaAPI
 from django.http import HttpRequest
 from ninja.openapi.schema import OpenAPISchema
@@ -17,7 +15,7 @@
 original_get_openapi_schema = api.get_openapi_schema
 
 
-def custom_openapi_schema(path_params: _T | None = None) -> OpenAPISchema:
+def custom_openapi_schema(path_params: dict| None =None) -> OpenAPISchema:
     schema = original_get_openapi_schema()
 
     schema["components"]["securitySchemes"] = {