o #j@sdZddlZddlZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZdefd d Zdefd d Zdefd dZdS)z{ Global Flask error handling middleware Provides centralized error handling for the Flask API with Laravel-style responses N)Flaskrequestjsonifyg) HTTPException)ValidationException) ApiResponse) get_loggerappc stdtdtffdd }dfdd}dfd d }d fd d }dfdd}dfdd}dfdd}dfdd}tdtffdd } tdtffdd } dS)zu Register global error handlers for the Flask application Args: app: Flask application instance error_handlerecs*d|jd|jt|j|jS)z.handle_validation_exceptionc"dt|tjdddS)zHandle 400 Bad Request errorsz Bad request: z.Bad request. Please check your request format.r status_coder strrerrorrrrrhandle_bad_requestz3register_error_handlers..handle_bad_requestcr)zHandle 401 Unauthorized errorszUnauthorized access attempt: z>Unauthorized. Please provide valid authentication credentials.r rrrrrrhandle_unauthorized$rz4register_error_handlers..handle_unauthorizedcr)zHandle 403 Forbidden errorszForbidden access attempt: z=Forbidden. You don't have permission to access this resource.r"rrrrrrhandle_forbidden*rz1register_error_handlers..handle_forbiddenc dtjtjdddS)zHandle 404 Not Found errorszResource not found: z%The requested resource was not found.r$r)r rurlrrrrrrhandle_not_found0z1register_error_handlers..handle_not_foundcs(dtjdtjtjdddS)z$Handle 405 Method Not Allowed errorszMethod not allowed:  z%Method not allowed for this endpoint.r)r)r rmethodr&rrrrrrhandle_method_not_allowed6sz:register_error_handlers..handle_method_not_allowedcr%)z#Handle 429 Too Many Requests errorszRate limit exceeded: z*Too many requests. Please try again later.r-r)r r remote_addrrrrrrrhandle_too_many_requests<r(z9register_error_handlers..handle_too_many_requestscs^dt|tjdr(tjdt|dtddStjdddS)z5Handle 500 Internal Server Error - Always return JSONzInternal server error: DEBUG Traceback: r0rz"An internal server error occurred.rr traceback format_excconfiggetrrr rrrhandle_internal_server_errorB z=register_error_handlers..handle_internal_server_errorcs0d|jd|jtj|jpd|jdS)z3Handle generic HTTP exceptions - Always return JSONzHTTP exception: z - zAn error occurred.r)r code descriptionrrrrrrhandle_http_exceptionQsz6register_error_handlers..handle_http_exceptioncs^dt|tjdr(tjdt|dtddStjdddS)z4Handle any unhandled exceptions - Always return JSONzUnhandled exception: r1zUnhandled error: r2r0rzAn unexpected error occurred.r3rr8rrhandle_generic_exceptionWr:z9register_error_handlers..handle_generic_exceptionN)r errorhandlerrr Exception) r rrr!r#r'r,r/r9r=r>rr8rregister_error_handlerss*rAcsFtdjfdd}jfdd}jfdd}dS) zz Register request lifecycle handlers for the Flask application Args: app: Flask application instance request_handlercsxjdr8dtjdtjtjrtjddnd}|r(d|tjr:dt tjdSdSdS) z;Run before each request for additional validation and setupr1z Request: r*T)silentNz JSON Body: zQuery Params: ) r6r7debugrr+r&is_jsonget_jsonargsdict) json_bodyr8rrbefore_request_validationps z.before_request_validationc sd|jd<d|jd<d|jd<jddr%d |jd <d |jd <d |jd<|jdkru|jsu|jdsuz&d|r?|jddnd|jd}t ||_ d|_ d|jW|St yt}z d|WYd}~|Sd}~ww|S)z'Ensure all responses are JSON formattednosniffzX-Content-Type-OptionsDENYzX-Frame-Optionsz 1; mode=blockzX-XSS-Protection ENABLE_CORST*zAccess-Control-Allow-OriginzGET, POST, PUT, DELETE, OPTIONSzAccess-Control-Allow-MethodszContent-Type, AuthorizationzAccess-Control-Allow-Headersrzapplication/jsonzAn error occurred)as_textz Unknown error)rrrz+Converted non-JSON error response to JSON: z$Failed to convert response to JSON: N)headersr6r7rrE content_type startswithget_datajsondumpsdatar r@r)response error_datar r8rrensure_json_response~s0        z7register_request_handlers..ensure_json_responsecs&d|jdtjdtj|S)z"Run after each request for cleanupz Response: z for r*)rDrrr+r&)rWrrrafter_request_cleanups"z8register_request_handlers..after_request_cleanupN)r before_request after_request)r rJrYrZrr8rregister_request_handlersgs  r]cCst|t|dS)zp Complete error handling setup for Flask application Args: app: Flask application instance N)rAr])r rrrsetup_error_handlings r^)__doc__r4rTflaskrrrrwerkzeug.exceptionsr#app.exceptions.validation_exceptionrapp.utils.response_helperrapp.utils.helpersr rAr]r^rrrrs    XA