o #jP- @sdZddlmZddlmZmZmZmZmZddl m Z m Z ddl m Z mZddlmZddlmZdd ee d efd d Zdd ee d efddZd d ee d efddZ    d!deee deee deee d efddZd"ded efddZdS)#z Flask route decorators for automatic request validation using Pydantic Provides Laravel-style validation decorators for Flask routes wraps)TypeOptionalCallableAnyDict)requestjsonify) BaseModelValidationError)ValidationException) ApiResponseThe given data was invalid.schemamessagecdtdtffdd }|S)a Decorator to validate JSON request body using Pydantic schema Args: schema: Pydantic BaseModel class for validation message: Custom error message for validation failures Usage: @validate_json(ActiveUsersRequest) def get_active_users(validated_data): # validated_data is now an ActiveUsersRequest instance pass freturnctfdd}|S)Nc sz$tjdd}|durtddgidd i|}|d|i|WStyD}zt|}t|j|j WYd}~Sd}~wty]}zt|j|j WYd}~Sd}~wt yz}ztj dt |dd WYd}~Sd}~ww) NTsilentjsonz$Request must contain valid JSON datazInvalid JSON datavalidated_jsonzValidation failed:  status_code) r get_jsonr from_custom_errorsr from_pydantic_errorrvalidation_errorerrorsr Exceptionerrorstr)argskwargs json_datavalidated_dataevalidation_exceptionrrrr7/var/www/html/oms-prod-mcp/app/decorators/validation.pydecorated_functions(  $z.decorator..decorated_functionrrr.rrrr- decoratorsz validate_json..decoratorrrrr2rr0r- validate_jsons"r5Invalid query parameters.cr)ay Decorator to validate query parameters using Pydantic schema Args: schema: Pydantic BaseModel class for validation message: Custom error message for validation failures Usage: @validate_query(PaginationRequest) def get_items(validated_query): # validated_query contains validated query parameters pass rrcr)Nc s"z:tj}|D]\}}|rt|||<q |dvr'|dk||<q di|}|g|Ri|WStyZ}zt |}t |j |j WYd}~Sd}~wtys}zt |j |j WYd}~Sd}~wty}zt jdt|ddWYd}~Sd}~ww)Ntruefalser8zQuery validation failed: rrr)r r&to_dictitemsisdigitintlowerr r r rr!r"rr#r$r%)r&r' query_datakeyvaluer)r*r+r,rr-r.Ps*   $z=validate_query..decorator..decorated_functionrr/r0r1r-r2Osz!validate_query..decoratorr3r4rr0r-validate_queryAs rBInvalid form data.cr)ar Decorator to validate form data using Pydantic schema Args: schema: Pydantic BaseModel class for validation message: Custom error message for validation failures Usage: @validate_form(UserRegistrationRequest) def register_user(validated_form): # validated_form contains validated form data pass rrcr)Nc sz)tj}tjrtjD]\}}|||<qdi|}|g|Ri|WStyI}zt|}t |j |j WYd}~Sd}~wtyb}zt |j |j WYd}~Sd}~wt y}ztj dt|ddWYd}~Sd}~ww)NzForm validation failed: rrr)r formr:filesr;r r r rr!r"rr#r$r%)r&r' form_datar@filer)r*r+r,rr-r.s$   $z.decorator..decorated_functionrr/r0r1r-r2sz validate_form..decoratorr3r4rr0r- validate_formrsrHNValidation failed. json_schema query_schema form_schemacs dtdtffdd }|S)a Decorator to validate multiple data sources (JSON, query, form) using different schemas Args: json_schema: Pydantic schema for JSON body validation query_schema: Pydantic schema for query parameters validation form_schema: Pydantic schema for form data validation message: Custom error message for validation failures Usage: @validate_multiple( json_schema=ContentStatsRequest, query_schema=PaginationRequest ) def get_content_stats(validated_json=None, validated_query=None, validated_form=None): # Each validated parameter contains the respective validated data pass rrcs tfdd}|S)Nc szxd}d}d}rtjdd}|durd i|}rLtj}|D]\}}|r4t|||<q%|dvrB|dk||<q%|rLd i|}rmtj} tj rdtj D]\}} | | |<q[| rmd i| }||||d|WSt y} zt | } t | j| jWYd} ~ Sd} ~ wt y} zt | j| jWYd} ~ Sd} ~ wty} zt jdt| ddWYd} ~ Sd} ~ ww) NTrr7r8)rvalidated_queryvalidated_formz Multi-source validation failed: rrr)r rr&r:r;r<r=r>rDrEr r r rr!r"rr#r$r%) r&r'rrMrNr(r?r@rArFrGr*r+)rrLrJrrKrr-r.sX      $z@validate_multiple..decorator..decorated_functionrr/rLrJrrKr1r-r2s6z$validate_multiple..decoratorr3)rJrKrLrr2rrOr-validate_multiples9rP Business rule validation failed.validator_funccsdtdtffdd }|S)a Decorator for custom business logic validation Args: validator_func: Function that takes request data and returns True if valid, raises ValidationException if invalid message: Custom error message for validation failures Usage: def validate_user_permissions(data): if not user_has_permission(data.user_id, 'read_reports'): raise ValidationException.from_custom_errors( {"user_id": ["User does not have permission to access reports"]}, "Permission denied" ) return True @validate_business_rules(validate_user_permissions) def get_reports(validated_data): pass rrcstfdd}|S)Nc sz,d}|rt|ddr|d}ntjddp tjp tj}||i|WStyF}zt|j |j WYd}~Sd}~wt yc}ztj dt |ddWYd}~Sd}~ww)Nr__dict__Trz!Business rule validation failed: rr)hasattrr rrDr:r&r rr!r"rr#r$r%)r&r'r)r*)rrRrr-r. s  $zFvalidate_business_rules..decorator..decorated_functionrr/rRr1r-r2 sz*validate_business_rules..decoratorr3)rRrr2rrUr-validate_business_rulessrV)r)r6)rC)NNNrI)rQ)__doc__ functoolsrtypingrrrrrflaskr r pydanticr r #app.exceptions.validation_exceptionr app.utils.response_helperrr%r5rBrHrPrVrrrr-s0   31/    T