validation requester updated

2024-12-10 16:45:53 +03:00
parent 6ba0e37ffd
commit 4308f1e7b1
20 changed files with 671 additions and 113 deletions
--- a/service_app/routers/validations/router.py
+++ b/service_app/routers/validations/router.py
@@ -0,0 +1,109 @@
+from fastapi import status
+from fastapi.routing import APIRouter
+from fastapi.requests import Request
+from fastapi.exceptions import HTTPException
+from api_validations.validations_request import (
+    EndpointValidation,
+)
+from pydantic import BaseModel
+
+
+validations_route = APIRouter(prefix="/validations", tags=["Validations"])
+validations_route.include_router(validations_route, include_in_schema=True)
+
+
+class EndpointValidationResponse(BaseModel):
+    language: str
+    headers: dict
+    validation: dict
+
+
+def retrieve_validation_from_class(selected_event, events):
+    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)
+    return function_class.__event_validation__.get(event_function_code, None)
+
+
+@validations_route.post(path="/endpoint", summary="Retrieve validation of endpoint")
+def user_list(request: Request, validation: EndpointValidation):
+    import api_events.events as events
+    from api_services.redis.functions import get_object_via_access_key
+    from databases import (
+        EndpointRestriction,
+        Events,
+    )
+
+    valid_token = get_object_via_access_key(request=request)
+    if not valid_token:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail=f"No valid token found in the request.",
+        )
+    endpoint_active = EndpointRestriction.filter_one(
+        EndpointRestriction.endpoint_name.ilike(f"%{str(validation.endpoint)}%"),
+        system=True,
+    ).data
+    if not endpoint_active:
+        raise HTTPException(
+            status_code=status.HTTP_401_UNAUTHORIZED,
+            detail=f"This endpoint {str(validation.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_401_UNAUTHORIZED,
+                detail="Selected company is not found in the token object.",
+            )
+        selected_event = Events.filter_one(
+            Events.endpoint_id == endpoint_active.id,
+            Events.id.in_(valid_token.selected_company.reachable_event_list_id),
+        ).data
+        if not selected_event:
+            raise HTTPException(
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                detail="This endpoint requires event validation. Please contact your responsible company to use this event.",
+            )
+        active_validation = retrieve_validation_from_class(selected_event, events)
+        if not active_validation:
+            raise HTTPException(
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                detail="This endpoint requires event validation. Please contact your responsible company to use this event.",
+            )
+        headers = getattr(
+            active_validation, str(valid_token.lang).lower(), active_validation.tr
+        )
+        return EndpointValidationResponse(
+            language=valid_token.lang,
+            headers=headers,
+            validation=active_validation.model_json_schema(),
+        )
+    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_one(
+            Events.endpoint_id == endpoint_active.id,
+            Events.id.in_(valid_token.selected_occupant.reachable_event_list_id),
+        ).data
+        if not selected_event:
+            raise HTTPException(
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                detail=f"This endpoint {str(validation.endpoint)} requires event validation. Please contact your responsible company to use this event.",
+            )
+        active_validation = retrieve_validation_from_class(selected_event, events)
+        if not active_validation:
+            raise HTTPException(
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                detail=f"This endpoint {str(validation.endpoint)} requires event validation. Please contact your responsible company to use this event.",
+            )
+        headers = getattr(
+            active_validation, str(valid_token.lang), active_validation.tr
+        )
+        return EndpointValidationResponse(
+            language=valid_token.lang,
+            headers=headers,
+            validation=active_validation.model_json_schema(),
+        )