Exemplo n.º 1
0
 def request_validation(
     self,
     request: FlaskRequest,
     query: Optional[Type[BaseModel]],
     body: Optional[Request],
     headers: Optional[Type[BaseModel]],
     cookies: Optional[Type[BaseModel]],
 ):
     req_query = request.args or {}
     if request.content_type == "application/json":
         parsed_body = request.get_json() or {}
     else:
         parsed_body = request.get_data() or {}
     req_headers = request.headers or {}
     req_cookies = request.cookies or {}
     request.context = Context(
         query=query.parse_obj(req_query) if query else None,
         body=body.model.parse_obj(parsed_body)
         if body and body.model else None,
         headers=headers.parse_obj(req_headers) if headers else None,
         cookies=cookies.parse_obj(req_cookies) if cookies else None,
     )