o di2@sddlZddlmZddlmZddlmZddlm Z m Z ddl m Z ddl mZGdd d eZGd d d eZGd d d eZGdddeZdS)Nstatus) CreateAPIView)MultiPartParser)IsAuthenticated IsAdminUser)Response)csv_master_serializersc@,eZdZdZefZejZe e fZ ddZ dS)AmenitiesCSVUploadViewa This is the AmenitiesCSVUploadView. It handles the CSV file upload for amenities data. Attributes: parser_classes (tuple): The parsers used for request data (MultiPartParser for handling file uploads). serializer_class (Serializer): The serializer class for the uploaded data. permission_classes (tuple): Permissions required to access this view (IsAuthenticated for authenticated users). Methods: post(request, *args, **kwargs): Handles the processing of uploaded CSV data. c s|jd}|stdditjdS|jds tdditjdSt|}t |j ddg}t fd d |Drfd d | D}t j|d d}t||r^|tdditjdSt|jtjdStddd|ditjdS)a* Handles the processing of uploaded CSV data and creation of amenities. Args: request (HttpRequest): The HTTP request containing the uploaded CSV file. Returns: Response: A response indicating the success or failure of the CSV data upload. filemessageFile not providedr.csvInvalid file formatnametype_idc3|]}|vVqdSN.0colcolumnsr]/var/www/html/evchargy.com/django-station-ev/station_ev/apps/master/views/csv_master_views.py >z.AmenitiesCSVUploadView.post..cSg|]\}}t|qSrdict)r_rowrrr ?z/AmenitiesCSVUploadView.post..TdatamanyzAmenities uploaded successfullyFile must have ,  columns)FILESgetrrHTTP_400_BAD_REQUESTrendswithpdread_csvlistralliterrowsr AmenitiesUploadCSVSerializersprintis_validsave HTTP_200_OKerrorsjoin selfrequestargskwargsr dfrequired_columns json_data serializerrrrposts:      zAmenitiesCSVUploadView.postN) __name__ __module__ __qualname____doc__rparser_classesr r4serializer_classrrpermission_classesrDrrrrr s  r c@r )ConnectorCSVUploadViewa This is the ConnectorCSVUploadView. It handles the CSV file upload for connectors data. Attributes: parser_classes (tuple): The parsers used for request data (MultiPartParser for handling file uploads). serializer_class (Serializer): The serializer class for the uploaded data. permission_classes (tuple): Permissions required to access this view (IsAuthenticated for authenticated users). Methods: post(request, *args, **kwargs): Handles the processing of uploaded CSV data. c s|jd}|stdditjdS|jds tdditjdSt|}t |j gd}t fdd |Drbd d | D}t j|d d }|rZ|tdditjdSt|jtjdStddd|ditjdS)a Handles the processing of uploaded CSV data. Args: request (HttpRequest): The HTTP request containing the uploaded CSV file. *args: Variable-length argument list. **kwargs: Variable-length keyword argument list. Returns: Response: The response message indicating whether the connectors were uploaded successfully. r r rrrr)rr short_namec3rrrrrrrrrz.ConnectorCSVUploadView.post..cSrrrrr!vrrrr#r$z/ConnectorCSVUploadView.post..Tr%z Connectors uploaded successfullyr(r)r*)r+r,rrr-rr.r/r0r1rr2r3r ConnectorCSVUploadSerializersr6r7r8r9r:r;rrrrDks8      zConnectorCSVUploadView.postN rErFrGrHrrIr rPrJrrrKrDrrrrrLWs  rLc@r )LocationCSVUploadViewa This is the LocationCSVUploadView class responsible for handling CSV file uploads containing location data. It allows users to upload location data from a CSV file, which is then processed and saved to the database. Attributes: parser_classes (tuple): The parsers used for request data, including MultiPartParser for handling file uploads. serializer_class (Serializer): The serializer class for deserializing the uploaded data. permission_classes (tuple): Specifies the permissions required to access this view, set to IsAuthenticated for authenticated users. Methods: post(request, *args, **kwargs): Handles the processing of uploaded CSV data, validating, and saving it to the database. c s|jd}|stdditjdS|jds tdditjdSt|}t |j ddg}t fd d |Drbd d | D}t j|d d}|rZ|tdditjdSt|jtjdStddd|ditjdS)ai Handles the processing of uploaded CSV data. Args: request (HttpRequest): The HTTP request containing the uploaded CSV file. *args: Variable-length argument list. **kwargs: Variable-length keyword argument list. Returns: Response: A response message indicating whether the connectors were uploaded successfully or any errors encountered. This method processes the uploaded CSV file containing location data. It validates the data, checks for required columns, and saves the location data to the database. r r rrrrrrc3rrrrrrrrrz-LocationCSVUploadView.post..cSrrrrNrrrr#r$z.LocationCSVUploadView.post..Tr%zLocations uploaded successfullyr(r)r*)r+r,rrr-rr.r/r0r1rr2r3r ZLocationsCSVUploadSerializersr6r7r8r9r:r;rrrrDs8      zLocationCSVUploadView.postNrQrrrrrRs  rRc@r )NetworkOperatorCSVUploadViewal This is the NetworkOperatorCSVUploadView class responsible for handling CSV file uploads containing network operator data. It allows users to upload network operator data from a CSV file, which is then processed and saved to the database. Attributes: parser_classes (tuple): The parsers used for request data, including MultiPartParser for handling file uploads. serializer_class (Serializer): The serializer class for deserializing the uploaded data. permission_classes (tuple): Specifies the permissions required to access this view, set to IsAuthenticated for authenticated users. Methods: post(request, *args, **kwargs): Handles the processing of uploaded CSV data, validating, and saving it to the database. Meta: permissions (tuple): Permissions required to access this view (IsAuthenticated for authenticated users). serializer_class (NetworkOperatorUploadCSVSerializers): The serializer used for input/output data. parser_classes (tuple): Parsers used for request data, including MultiPartParser for handling file uploads. c s|jd}|stdditjdS|jds tdditjdSt|}t |j ddg}t fd d |Drbd d | D}|j |d d}|rZ|tdditjdSt|jtjdStddd|ditjdS)a Handles the processing of uploaded CSV data. Args: request (HttpRequest): The HTTP request containing the uploaded CSV file. *args: Variable-length argument list. **kwargs: Variable-length keyword argument list. Returns: Response: The response message indicating whether the network operators were uploaded successfully. This method processes a CSV file containing network operator data, validates the data, and saves it to the database. It checks if the required file format and columns are present in the CSV file and handles the uploading process. r r rrrrrrc3rrrrrrrr:rz4NetworkOperatorCSVUploadView.post..cSrrrrNrrrr#;r$z5NetworkOperatorCSVUploadView.post..Tr%z&Network Operator uploaded successfullyr(r)r*)r+r,rrr-rr.r/r0r1rr2r3get_serializerr6r7r8r9r:r;rrrrDs4      z!NetworkOperatorCSVUploadView.postN) rErFrGrHrrIr Z#NetworkOperatorUploadCSVSerializersrJrrrKrDrrrrrSs  rS)pandasr/rest_frameworkrZrest_framework.genericsrZrest_framework.parsersrZrest_framework.permissionsrrrest_framework.responserZapps.master.serializersr r rLrRrSrrrrs     LNT