fixes

src/service/auth/auth.py

@@ -9,14 +9,14 @@ from src.errors.http_errors.errors import Errors
 from src.models.configs_read.env import env_settings
 from src.models.pydantic_models.model import RefreshTokensCreate, UserOut
 
-from .jwt import Hashes, Jwt
+from .jwt import HashService, JwtService
 
 
-class CurrentUser:
+class CurrentUserService:
     
     def __init__(self) -> None:
-        self.jwt_service=Jwt()
-        self.hash=Hashes()
+        self.jwt_service=JwtService()
+        self.hash=HashService()
         self.crud_db_actions=UsersCrudActions()
         self.jwt_db_actions=JwtCrudActions()
         self.error=Errors()
@@ -147,9 +147,12 @@ class CurrentUser:
             expires_at=datetime.now(UTC)+timedelta(days=env_settings.REFRESH_TOKEN_EXPIRE_DAYS),
         )
         
-        self.jwt_db_actions.create_and_update_token(RefreshTokensCreate.model_dump(new_token_record), old_jti, new_jti)
+        success = self.jwt_db_actions.create_and_update_token(RefreshTokensCreate.model_dump(new_token_record), old_jti, new_jti)
         
-        return (new_access_token,new_refresh_token)
+        if not success:
+            raise self.error.not_found_error(detail="Token not found")
+
+        return (new_access_token, new_refresh_token)
     
     
     
@@ -163,15 +166,14 @@ class CurrentUser:
         
         try:
             jti=UUID(jti)
-        except (ValueError, TypeError) as e:
+        except (ValueError, TypeError, AttributeError) as e:
             raise self.error.credentials_error(detail="Jwt token is incorrect") from e
         
-        current_token = self.jwt_db_actions.get_token_by_id(jti)
-        if current_token is None:
-            raise self.error.not_found_error(detail="Refresh Token Not Found")
-            
         '''logout by assigning revoked flag'''
-        return self.jwt_db_actions.logout(jti)
+        if self.jwt_db_actions.logout(jti):
+            return True
+        else:
+            raise self.error.not_found_error(detail="Refresh Token Not Found")
     
     
     
@@ -186,4 +188,4 @@ class CurrentUser:
         
         return (access_token, refresh_token)
     
-auth=CurrentUser()
+auth=CurrentUserService()