3 services are updated

api_services/api_modules/redis/redis_handlers.py

@@ -0,0 +1,53 @@
+from typing import Any, Union
+
+from api_validations.token.validations import TokenDictType, EmployeeTokenObject, OccupantTokenObject, CompanyToken, OccupantToken, UserType
+from api_controllers.redis.database import RedisActions
+from schemas import Users
+
+
+class RedisHandlers:
+
+    AUTH_TOKEN: str = "AUTH_TOKEN"
+
+    @classmethod
+    def process_redis_object(cls, redis_object: dict[str, Any]) -> TokenDictType:
+        """Process Redis object and return appropriate token object."""
+        if not redis_object.get("selected_company"):
+            redis_object["selected_company"] = None
+        if not redis_object.get("selected_occupant"):
+            redis_object["selected_occupant"] = None
+        if redis_object.get("user_type") == UserType.employee.value:
+            return EmployeeTokenObject(**redis_object)
+        elif redis_object.get("user_type") == UserType.occupant.value:
+            return OccupantTokenObject(**redis_object)
+        raise ValueError("Invalid user type")
+
+    @classmethod
+    def get_object_from_redis(cls, access_token: str) -> TokenDictType:
+        redis_response = RedisActions.get_json(list_keys=[RedisHandlers.AUTH_TOKEN, access_token, "*"])
+        if not redis_response.status:
+            raise ValueError("EYS_0001")
+        if redis_object := redis_response.first:
+            return cls.process_redis_object(redis_object)
+        raise ValueError("EYS_0002")
+
+    @classmethod
+    def set_object_to_redis(cls, user: Users, token, header_info):
+        result_delete = RedisActions.delete(list_keys=[RedisHandlers.AUTH_TOKEN, "*", str(user.uu_id)])
+        generated_access_token = PasswordModule.generate_access_token()
+        keys = [RedisHandlers.AUTH_TOKEN, generated_access_token, str(user.uu_id)]
+        RedisActions.set_json(list_keys=keys, value={**token, **header_info}, expires={"hours": 1, "minutes": 30})
+        return generated_access_token
+
+    @classmethod
+    def update_token_at_redis(cls, token: str, add_payload: Union[CompanyToken, OccupantToken]):
+        if already_token_data := RedisActions.get_json(list_keys=[RedisHandlers.AUTH_TOKEN, token, "*"]).first:
+            already_token = cls.process_redis_object(already_token_data)
+            if already_token.is_employee and isinstance(add_payload, CompanyToken):
+                already_token.selected_company = add_payload
+            elif already_token.is_occupant and isinstance(add_payload, OccupantToken):
+                already_token.selected_occupant = add_payload
+            list_keys = [RedisHandlers.AUTH_TOKEN, token, str(already_token.user_uu_id)]
+            result = RedisActions.set_json(list_keys=list_keys, value=already_token.model_dump(), expires={"hours": 1, "minutes": 30})
+            return result.first
+        raise ValueError("Something went wrong")