o di @sddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZd d ZGd d d eZGd ddZdS)N)ValidationError) UploadedFile) BaseValidator)HttpResponseBadRequest)status)json)FileFieldConstantscCsd}tt||S)Nzzapi/v1/(vehicle/insurance|master/upload/(amenities|connectors|locations|network)/csv|station/upload/csv|vehicle/insurance))boolresearch)stringpatternrK/var/www/html/evchargy.com/django-station-ev/station_ev/base/middlewares.pycheck_path_is_allowed src@s2eZdZdZddejdZdefddZ dS) FileValidatora< Validates the file size and extension of an uploaded file. Args: message (str, optional): The error message to display if the file is invalid. Defaults to a message indicating the allowed file formats and the maximum file size. zGInvalid file type or exceeds maximum file size. (Allowed file formats: z, z, maximum file size: 15 MB)valuecCsB|jtjkr t|j|jdd}|tjvrt|jdS)z Validates the given `value` (an UploadedFile instance). Args: value (UploadedFile): The file to validate. Raises: ValidationError: If the file is invalid. .N) sizer FILE_SIZErmessagenamesplitlower IMAGE_FORMATS)selfrextrrr__call__!s   zFileValidator.__call__N) __name__ __module__ __qualname____doc__joinrrrrrrrrrrs rc@s eZdZdZddZddZdS)FileValidationMiddlewarez Middleware that validates uploaded files. Args: get_response (callable): The callable to be called after processing the request. cCs ||_dS)z Initializes the middleware with the given `get_response` callable. Args: get_response (callable): The callable to be called after processing the request. N) get_response)rr%rrr__init__<s z!FileValidationMiddleware.__init__c Cst|jsJ|jdvrJ|jD]:\}}t|trIz ttj |Wqt yH}zt j d|j dd}tt|ddWYd}~Sd}~wwq||S)z Processes the incoming request, validating uploaded files if necessary. Args: request (HttpRequest): The HTTP request to process. Returns: HttpResponse: The response to return to the client. )POSTPUTPATCH BAD_REQUESTr)coderrzapplication/json) content_typeN)rpathmethodFILESitems isinstancerrrrrrHTTP_400_BAD_REQUESTmessagesrrdumpsr%)rrequest field_namereresponserrrrEs$    z!FileValidationMiddleware.__call__N)rr r!r"r&rrrrrr$4s r$)r django.core.exceptionsrdjango.core.files.uploadedfilerZdjango.core.validatorsr django.httprrest_frameworkrZrest_framework.utilsrbase.constantsrrrr$rrrrs       "