mdb-engine 0.1.6__py3-none-any.whl → 0.4.12__py3-none-any.whl

mdb_engine/auth/session_manager.py

@@ -8,13 +8,16 @@ This module is part of MDB_ENGINE - MongoDB Engine.
 
 import logging
 from datetime import datetime, timedelta
-from typing import Any, Dict, List, Optional
+from typing import Any
 
 from bson.objectid import ObjectId
 
 try:
-    from pymongo.errors import (ConnectionFailure, OperationFailure,
-                                ServerSelectionTimeoutError)
+    from pymongo.errors import (
+        ConnectionFailure,
+        OperationFailure,
+        ServerSelectionTimeoutError,
+    )
 except ImportError:
     ConnectionFailure = Exception
     OperationFailure = Exception
@@ -93,10 +96,10 @@ class SessionManager:
         user_id: str,
         device_id: str,
         refresh_jti: str,
-        device_info: Optional[Dict[str, Any]] = None,
-        ip_address: Optional[str] = None,
-        session_fingerprint: Optional[str] = None,
-    ) -> Optional[Dict[str, Any]]:
+        device_info: dict[str, Any] | None = None,
+        ip_address: str | None = None,
+        session_fingerprint: str | None = None,
+    ) -> dict[str, Any] | None:
         """
         Create a new user session.
 
@@ -120,9 +123,7 @@ class SessionManager:
                 # Remove oldest inactive session
                 await self.cleanup_inactive_sessions(user_id)
                 # Check again
-                active_sessions = await self.get_user_sessions(
-                    user_id, active_only=True
-                )
+                active_sessions = await self.get_user_sessions(user_id, active_only=True)
                 if len(active_sessions) >= self.max_sessions:
                     # Force remove oldest session
                     if active_sessions:
@@ -168,13 +169,11 @@ class SessionManager:
             ValueError,
             TypeError,
         ) as e:
-            logger.error(
-                f"Error creating session for user {user_id}: {e}", exc_info=True
-            )
+            logger.error(f"Error creating session for user {user_id}: {e}", exc_info=True)
             return None
 
     async def update_session_activity(
-        self, refresh_jti: str, ip_address: Optional[str] = None
+        self, refresh_jti: str, ip_address: str | None = None
     ) -> bool:
         """
         Update session last_seen timestamp (activity tracking).
@@ -205,14 +204,10 @@ class SessionManager:
             ValueError,
             TypeError,
         ) as e:
-            logger.error(
-                f"Error updating session activity for {refresh_jti}: {e}", exc_info=True
-            )
+            logger.error(f"Error updating session activity for {refresh_jti}: {e}", exc_info=True)
             return False
 
-    async def get_session_by_refresh_token(
-        self, refresh_jti: str
-    ) -> Optional[Dict[str, Any]]:
+    async def get_session_by_refresh_token(self, refresh_jti: str) -> dict[str, Any] | None:
         """
         Get session by refresh token JWT ID.
 
@@ -223,9 +218,7 @@ class SessionManager:
             Session document or None if not found
         """
         try:
-            session = await self.collection.find_one(
-                {"refresh_jti": refresh_jti, "active": True}
-            )
+            session = await self.collection.find_one({"refresh_jti": refresh_jti, "active": True})
             return session
         except (
             OperationFailure,
@@ -241,7 +234,7 @@ class SessionManager:
             return None
 
     async def validate_session_fingerprint(
-        self, refresh_jti: str, current_fingerprint: str, strict: Optional[bool] = None
+        self, refresh_jti: str, current_fingerprint: str, strict: bool | None = None
     ) -> bool:
         """
         Validate session fingerprint matches stored fingerprint.
@@ -267,9 +260,7 @@ class SessionManager:
             stored_fingerprint = session.get("session_fingerprint")
 
             if not stored_fingerprint:
-                strict_mode = (
-                    strict if strict is not None else self.fingerprinting_strict
-                )
+                strict_mode = strict if strict is not None else self.fingerprinting_strict
                 return not strict_mode
 
             return stored_fingerprint == current_fingerprint
@@ -297,7 +288,7 @@ class SessionManager:
 
     async def get_user_sessions(
         self, user_id: str, active_only: bool = True
-    ) -> List[Dict[str, Any]]:
+    ) -> list[dict[str, Any]]:
         """
         Get all sessions for a user.
 
@@ -313,9 +304,7 @@ class SessionManager:
             if active_only:
                 query["active"] = True
 
-            sessions = (
-                await self.collection.find(query).sort("last_seen", -1).to_list(None)
-            )
+            sessions = await self.collection.find(query).sort("last_seen", -1).to_list(None)
             return sessions
         except (
             OperationFailure,
@@ -324,9 +313,7 @@ class SessionManager:
             ValueError,
             TypeError,
         ) as e:
-            logger.error(
-                f"Error getting sessions for user {user_id}: {e}", exc_info=True
-            )
+            logger.error(f"Error getting sessions for user {user_id}: {e}", exc_info=True)
             return []
 
     async def revoke_session(self, session_id: Any) -> bool:
@@ -367,9 +354,7 @@ class SessionManager:
             logger.error(f"Error revoking session {session_id}: {e}", exc_info=True)
             return False
 
-    async def revoke_user_sessions(
-        self, user_id: str, exclude_device_id: Optional[str] = None
-    ) -> int:
+    async def revoke_user_sessions(self, user_id: str, exclude_device_id: str | None = None) -> int:
         """
         Revoke all sessions for a user.
 
@@ -400,12 +385,10 @@ class SessionManager:
             ValueError,
             TypeError,
         ) as e:
-            logger.error(
-                f"Error revoking sessions for user {user_id}: {e}", exc_info=True
-            )
+            logger.error(f"Error revoking sessions for user {user_id}: {e}", exc_info=True)
             return 0
 
-    async def cleanup_inactive_sessions(self, user_id: Optional[str] = None) -> int:
+    async def cleanup_inactive_sessions(self, user_id: str | None = None) -> int:
         """
         Clean up inactive sessions (beyond inactivity timeout).