first commit

2024-11-07 17:44:29 +03:00
commit 643d6d8f65
247 changed files with 420800 additions and 0 deletions
--- a/api_services/redis/auth_actions/token.py
+++ b/api_services/redis/auth_actions/token.py
@@ -0,0 +1,105 @@
+from fastapi import HTTPException, status
+from fastapi.requests import Request
+
+from database_sql_models import Events
+
+# url_that_not_requires_event_validation = [
+#     "/authentication/login",
+#     "/authentication/select",
+#     "/authentication/valid",
+#     "/authentication/refresh",
+#     "/authentication/change_password",
+#     "/authentication/create_password",
+#     "/authentication/disconnect",
+#     "/authentication/logout",
+#     "/authentication/refresher",
+#     "/authentication/forgot",
+#     "/authentication/avatar",
+# ]
+
+
+def parse_token_object_to_dict(request: Request):  # from requests import Request
+    from api_services.redis.functions import get_object_via_access_key
+    from databases import EndpointRestriction
+    import api_events.events as events
+
+    if valid_token := get_object_via_access_key(request=request):
+        endpoint_name = str(request.url).replace(str(request.base_url), "/")
+        endpoint_active = EndpointRestriction.filter_active(
+            EndpointRestriction.endpoint_name.ilike(f"%{endpoint_name}%")
+        ).data[0]
+        if not endpoint_active:
+            raise HTTPException(
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                detail="This endpoint is not active for this user, please contact your responsible company for further information.",
+            )
+
+        if valid_token.user_type == 1:
+            if not valid_token.selected_company:
+                raise HTTPException(
+                    status_code=status.HTTP_418_IM_A_TEAPOT,
+                    detail="Selected company is not found in the token object.",
+                )
+            selected_event = Events.filter_active(
+                Events.endpoint_id == endpoint_active.id,
+                Events.id.in_(valid_token.selected_company.reachable_event_list_id),
+            )
+            if not selected_event.data:
+                raise HTTPException(
+                    status_code=status.HTTP_401_UNAUTHORIZED,
+                    detail="This endpoint requires event validation. Please contact your responsible company to use this event.",
+                )
+            selected_event = selected_event.data[0]
+            event_function_class = getattr(selected_event, "function_class", None)
+            event_function_code = getattr(selected_event, "function_code", None)
+            function_class = getattr(events, event_function_class, None)
+            active_function = getattr(
+                function_class,
+                function_class.__event_keys__.get(event_function_code, None),
+                None,
+            )
+            if not active_function:
+                raise HTTPException(
+                    status_code=status.HTTP_401_UNAUTHORIZED,
+                    detail="This endpoint requires event validation. Please contact your responsible company to use this event.",
+                )
+            valid_token.available_event = active_function
+            return valid_token
+        elif valid_token.user_type == 2:
+            if not valid_token.selected_occupant:
+                raise HTTPException(
+                    status_code=status.HTTP_418_IM_A_TEAPOT,
+                    detail="Selected occupant is not found in the token object.",
+                )
+            selected_event = Events.filter_active(
+                Events.endpoint_id == endpoint_active.id,
+                Events.id.in_(valid_token.selected_occupant.reachable_event_list_id),
+            )
+            if not selected_event.data:
+                raise HTTPException(
+                    status_code=status.HTTP_401_UNAUTHORIZED,
+                    detail="This endpoint requires event validation. Please contact your responsible company to use this event.",
+                )
+            selected_event = selected_event.data[0]
+            event_function_class = getattr(selected_event, "function_class", None)
+            event_function_code = getattr(selected_event, "function_code", None)
+            function_class = getattr(events, event_function_class, None)
+            active_function = getattr(
+                function_class,
+                function_class.__event_keys__.get(event_function_code, None),
+                None,
+            )
+            if not active_function:
+                raise HTTPException(
+                    status_code=status.HTTP_401_UNAUTHORIZED,
+                    detail="This endpoint requires event validation. Please contact your responsible company to use this event.",
+                )
+            valid_token.available_event = active_function
+            return valid_token
+        valid_token.available_event = None
+        return valid_token
+    user_type = "Company" if valid_token.user_type == 1 else "Occupant"
+    raise HTTPException(
+        status_code=status.HTTP_401_UNAUTHORIZED,
+        detail=f"Token of this user is not valid. Please login and refresh {user_type} selection.",
+    )