fix: lint and format

Author: VaibhavSingh8

todo/middlewares/jwt_auth.py

@@ -47,8 +47,7 @@ def __call__(self, request):
                     ],
                 )
                 return JsonResponse(
-                    data=error_response.model_dump(mode="json", exclude_none=True), 
-                    status=status.HTTP_401_UNAUTHORIZED
+                    data=error_response.model_dump(mode="json", exclude_none=True), status=status.HTTP_401_UNAUTHORIZED
                 )
 
         except (TokenMissingError, TokenExpiredError, TokenInvalidError) as e:
@@ -67,8 +66,7 @@ def __call__(self, request):
                 ],
             )
             return JsonResponse(
-                data=error_response.model_dump(mode="json", exclude_none=True), 
-                status=status.HTTP_401_UNAUTHORIZED
+                data=error_response.model_dump(mode="json", exclude_none=True), status=status.HTTP_401_UNAUTHORIZED
             )
 
     def _try_authentication(self, request) -> bool:
@@ -103,28 +101,28 @@ def _try_google_refresh(self, request) -> bool:
         """Try to refresh Google access token"""
         try:
             refresh_token = request.COOKIES.get("ext-refresh")
-            
+
             if not refresh_token:
                 return False
-                
+
             payload = validate_google_refresh_token(refresh_token)
-            
+
             user_data = {
                 "user_id": payload["user_id"],
                 "google_id": payload["google_id"],
                 "email": payload["email"],
                 "name": payload.get("name", ""),
             }
-            
+
             new_access_token = generate_google_access_token(user_data)
-            
+
             self._set_google_user_data(request, payload)
 
             request._new_access_token = new_access_token
             request._access_token_expires = settings.GOOGLE_JWT["ACCESS_TOKEN_LIFETIME"]
-            
+
             return True
-            
+
         except (GoogleRefreshTokenExpiredError, GoogleTokenInvalidError):
             return False
         except Exception:
@@ -161,13 +159,10 @@ def _try_rds_auth(self, request) -> bool:
 
     def _process_response(self, request, response):
         """Process response and set new cookies if Google token was refreshed"""
-        if hasattr(request, '_new_access_token'):
+        if hasattr(request, "_new_access_token"):
             config = self._get_cookie_config()
             response.set_cookie(
-                "ext-access",
-                request._new_access_token,
-                max_age=request._access_token_expires,
-                **config
+                "ext-access", request._new_access_token, max_age=request._access_token_expires, **config
             )
         return response
 
@@ -191,8 +186,7 @@ def _handle_rds_auth_error(self, exception):
             errors=[ApiErrorDetail(title=ApiErrors.AUTHENTICATION_FAILED, detail=str(exception))],
         )
         return JsonResponse(
-            data=error_response.model_dump(mode="json", exclude_none=True), 
-            status=status.HTTP_401_UNAUTHORIZED
+            data=error_response.model_dump(mode="json", exclude_none=True), status=status.HTTP_401_UNAUTHORIZED
         )
 
     def _handle_google_auth_error(self, exception):
@@ -202,8 +196,7 @@ def _handle_google_auth_error(self, exception):
             errors=[ApiErrorDetail(title=ApiErrors.AUTHENTICATION_FAILED, detail=str(exception))],
         )
         return JsonResponse(
-            data=error_response.model_dump(mode="json", exclude_none=True), 
-            status=status.HTTP_401_UNAUTHORIZED
+            data=error_response.model_dump(mode="json", exclude_none=True), status=status.HTTP_401_UNAUTHORIZED
         )
 
 
@@ -240,4 +233,4 @@ def get_current_user_info(request) -> dict:
             }
         )
 
-    return user_info
+    return user_info

todo/tests/integration/base_mongo_test.py

@@ -20,9 +20,7 @@ def setUpClass(cls):
         cls.db = cls.mongo_client.get_database("testdb")
 
         cls.override = override_settings(
-            MONGODB_URI=cls.mongo_url,
-            DB_NAME="testdb",
-            FRONTEND_URL="http://localhost:4000"
+            MONGODB_URI=cls.mongo_url, DB_NAME="testdb", FRONTEND_URL="http://localhost:4000"
         )
         cls.override.enable()
         DatabaseManager.reset()
@@ -77,4 +75,4 @@ def get_user_model(self) -> UserModel:
             name=self.user_data["name"],
             createdAt=datetime.now(timezone.utc),
             updatedAt=datetime.now(timezone.utc),
-        )
+        )

todo/tests/unit/middlewares/test_jwt_auth.py

@@ -1,7 +1,6 @@
 from unittest import TestCase
 from unittest.mock import Mock, patch
 from django.http import HttpRequest, JsonResponse
-from django.conf import settings
 from rest_framework import status
 import json
 

todo/tests/unit/views/test_auth.py

@@ -108,7 +108,7 @@ def setUp(self):
         self.url = reverse("google_callback")
         self.factory = APIRequestFactory()
         self.view = GoogleCallbackView.as_view()
-        
+
         self.test_user_data = users_db_data[0]
 
     def test_get_redirects_for_oauth_error(self):
@@ -148,7 +148,7 @@ def test_get_redirects_for_valid_code_and_state(self, mock_create_user, mock_han
             "email": self.test_user_data["email_id"],
             "name": self.test_user_data["name"],
         }
-        
+
         user_id = str(ObjectId())
         mock_user = Mock()
         mock_user.id = ObjectId(user_id)
@@ -241,8 +241,8 @@ def test_logout_clears_session(self):
     def test_logout_clears_sessionid_cookie(self):
         """Test that logout clears sessionid cookie"""
         self.client.cookies["sessionid"] = "test_session_id"
-        
+
         response = self.client.post(self.url)
 
         self.assertEqual(response.status_code, status.HTTP_200_OK)
-        self.assertEqual(response.cookies.get("sessionid").value, "")
+        self.assertEqual(response.cookies.get("sessionid").value, "")

todo/utils/google_jwt_utils.py

@@ -28,7 +28,9 @@ def generate_google_access_token(user_data: dict) -> str:
             "token_type": "access",
         }
 
-        token = jwt.encode(payload=payload, key=settings.GOOGLE_JWT["PRIVATE_KEY"], algorithm=settings.GOOGLE_JWT["ALGORITHM"])
+        token = jwt.encode(
+            payload=payload, key=settings.GOOGLE_JWT["PRIVATE_KEY"], algorithm=settings.GOOGLE_JWT["ALGORITHM"]
+        )
         return token
 
     except Exception as e:
@@ -50,7 +52,9 @@ def generate_google_refresh_token(user_data: dict) -> str:
             "email": user_data["email"],
             "token_type": "refresh",
         }
-        token = jwt.encode(payload=payload, key=settings.GOOGLE_JWT["PRIVATE_KEY"], algorithm=settings.GOOGLE_JWT["ALGORITHM"])
+        token = jwt.encode(
+            payload=payload, key=settings.GOOGLE_JWT["PRIVATE_KEY"], algorithm=settings.GOOGLE_JWT["ALGORITHM"]
+        )
 
         return token
 
@@ -60,7 +64,9 @@ def generate_google_refresh_token(user_data: dict) -> str:
 
 def validate_google_access_token(token: str) -> dict:
     try:
-        payload = jwt.decode(jwt=token, key=settings.GOOGLE_JWT["PUBLIC_KEY"], algorithms=[settings.GOOGLE_JWT["ALGORITHM"]])
+        payload = jwt.decode(
+            jwt=token, key=settings.GOOGLE_JWT["PUBLIC_KEY"], algorithms=[settings.GOOGLE_JWT["ALGORITHM"]]
+        )
 
         if payload.get("token_type") != "access":
             raise GoogleTokenInvalidError(AuthErrorMessages.GOOGLE_TOKEN_INVALID)
@@ -77,7 +83,9 @@ def validate_google_access_token(token: str) -> dict:
 
 def validate_google_refresh_token(token: str) -> dict:
     try:
-        payload = jwt.decode(jwt=token, key=settings.GOOGLE_JWT["PUBLIC_KEY"], algorithms=[settings.GOOGLE_JWT["ALGORITHM"]])
+        payload = jwt.decode(
+            jwt=token, key=settings.GOOGLE_JWT["PUBLIC_KEY"], algorithms=[settings.GOOGLE_JWT["ALGORITHM"]]
+        )
         if payload.get("token_type") != "refresh":
             raise GoogleTokenInvalidError(AuthErrorMessages.GOOGLE_TOKEN_INVALID)
 

todo/views/auth.py

@@ -11,6 +11,7 @@
 from todo.utils.google_jwt_utils import generate_google_token_pair
 from todo.constants.messages import AppMessages
 
+
 class GoogleLoginView(APIView):
     @extend_schema(
         operation_id="google_login",
@@ -94,7 +95,7 @@ def get(self, request: Request):
         code = request.query_params.get("code")
         state = request.query_params.get("state")
         error = request.query_params.get("error")
-        
+
         if error:
             frontend_callback = f"{settings.FRONTEND_URL}/auth/callback"
             return HttpResponseRedirect(f"{frontend_callback}?error={error}")
@@ -137,7 +138,6 @@ def get(self, request: Request):
             frontend_callback = f"{settings.FRONTEND_URL}/auth/callback"
             return HttpResponseRedirect(f"{frontend_callback}?error=auth_failed")
 
-
     def _get_cookie_config(self):
         return {
             "path": "/",
@@ -154,6 +154,7 @@ def _set_auth_cookies(self, response, tokens):
             "ext-refresh", tokens["refresh_token"], max_age=settings.GOOGLE_JWT["REFRESH_TOKEN_LIFETIME"], **config
         )
 
+
 class GoogleLogoutView(APIView):
     @extend_schema(
         operation_id="google_logout",
@@ -198,13 +199,15 @@ def post(self, request: Request):
 
     def _handle_logout(self, request: Request):
         request.session.flush()
-        
-        response = Response({
-            "statusCode": status.HTTP_200_OK,
-            "message": AppMessages.GOOGLE_LOGOUT_SUCCESS,
-            "data": {"success": True},
-        })
-    
+
+        response = Response(
+            {
+                "statusCode": status.HTTP_200_OK,
+                "message": AppMessages.GOOGLE_LOGOUT_SUCCESS,
+                "data": {"success": True},
+            }
+        )
+
         self._clear_auth_cookies(response)
         return response
 
@@ -226,7 +229,7 @@ def _clear_auth_cookies(self, response):
         response.delete_cookie("ext-refresh", **delete_config)
 
         session_delete_config = {
-            "path": getattr(settings, 'SESSION_COOKIE_PATH', '/'),
-            "domain": getattr(settings, 'SESSION_COOKIE_DOMAIN', None),
+            "path": getattr(settings, "SESSION_COOKIE_PATH", "/"),
+            "domain": getattr(settings, "SESSION_COOKIE_DOMAIN", None),
         }
         response.delete_cookie("sessionid", **session_delete_config)

todo_project/settings/base.py

@@ -121,11 +121,7 @@
 }
 
 # JWT Configuration - Use different settings for tests vs production
-TESTING = (
-    'test' in sys.argv or 
-    'pytest' in sys.modules or 
-    os.getenv('TESTING') == 'True'
-)
+TESTING = "test" in sys.argv or "pytest" in sys.modules or os.getenv("TESTING") == "True"
 
 if TESTING:
     # Test JWT configuration (HS256 - simpler for tests)