o did@sddlZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z ddlmZmZmZddlmZdd lmZdd lmZdd lmZdd lmZmZdd lmZddl m!Z!m"Z"ddl#m$Z$ddl%m&Z&m'Z'ddl(m)Z)ddl*m+Z+Gddde Z,Gddde)Z-Gddde Z.Gddde)Z/Gddde)Z0Gddde)Z1Gdd d e)Z2Gd!d"d"eZ3Gd#d$d$e)Z4dS)%N)geosstatus)action) CreateAPIViewRetrieveAPIView)MultiPartParser)IsAuthenticated IsAdminUserAllowAny)Response)APIView)models)upload_station_csv_data)StationListFilterStationMediaListFilter)station_serializers)station_servicesstation_list_services)CustomPagination)AdminOrPostOnlyPermissionIsAdminToAllowCRUD)BaseModelViewSet)settingsc@s*eZdZdZefZefZej Z ddZ dS)StationCSVUploadViewa This is the StationCSVUploadView. It handles the CSV file upload for station data. Attributes: parser_classes (tuple): The parsers used for request data (MultiPartParser for handling file uploads). 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 sJ|jd}|stdditjdS|jds tdditjdSt|}t |j gd}t fdd |Dr|d  t j|d <|d  t j|d <|d  t j|d <|jd ddg}|D] \}} t| } | d| dd| d<| dp|d| d<|| qdt||jjtdditjdStddd|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 stations were uploaded successfully. filemessagezFile not providedrz.csvzInvalid file format)scarp_idnameaddress descriptiontitle_descriptionmeta_descriptionphone is_activeis_24_hours_openis_parking_freecost_descriptionlatitude longitude amenities connectorsphotosc3s|]}|vVqdSN).0colcolumnsr.\/var/www/html/evchargy.com/django-station-ev/station_ev/apps/station/views/stations_views.py `sz,StationCSVUploadView.post..r+r*r,T)valueZinplacer(r))r(r) coordinatesr'Nparking_cost_descriptionzStations uploaded successfullyzFile must have z, z columns)FILESgetr rHTTP_400_BAD_REQUESTrendswithpdZread_csvlistr2allapplyast literal_evalZfillnaZiterrowsdictappendruserid HTTP_200_OKjoin) selfrequestargskwargsrdfZrequired_columnsZ json_data_rowrow_datar.r1r3post/sF        zStationCSVUploadView.postN) __name__ __module__ __qualname____doc__rparser_classesr permission_classesrZStationCSVUploadSerializersserializer_classrQr.r.r.r3rs   rc@seZdZdZefZejZe j j Z eZddgZgdZddZddZd d Zed d gd egddddZed d gdegddddZdS)StationModelViewSeta This is the StationModelViewSet class responsible for managing station data. It provides endpoints for creating, retrieving, updating, and deleting station records. Attributes: permission_classes (tuple): Specifies the permissions required to access this view. It's set to IsAuthenticated, allowing only authenticated users to interact with station records. serializer_class (station_serializers.StationMasterSerializer): The serializer class used to serialize and deserialize station data. filter_backends (list): A list of filter backends to use for filtering station data. queryset (QuerySet): The queryset of station records that this viewset operates on. filterset_class (StationListFilter): The filterset class used to filter station data. search_fields (list): A list of fields to search for when filtering. http_method_names (list): A list of the HTTP methods that this viewset supports. Methods: get(request, *args, **kwargs): Retrieves station data based on the request parameters. delete(request, *args, **kwargs): Deletes a specific station record. list(request, *args, **kwargs): Retrieves a list of station data based on the request parameters. partial_update(request, *args, **kwargs): Handles partial updates (not allowed, returns an error). destroy(request, *args, **kwargs): Handles the destruction of station records (not allowed, returns an error). delete(request, pk=None): Deletes a specific station record based on the provided primary key. station_update(request, pk=None): Updates a specific station record based on the provided primary key. rrr:deletepatchc Cs|jjdd}|jjdvr|jjdd}|jjd}|jjd}|r(|s*|Sz%tt| d\}}tt| d\}}t j ||||f}|j|d WSt yZ|YSw) a/ Get the queryset of station records based on geographic coordinates. This method filters station records based on a bounding box defined by top-right and bottom-left lat/lng coordinates. Returns: QuerySet: The filtered queryset of station records. F is_rejected)DELETET is_verifiedtop_right_latlngbottom_left_latlng,)Zcoordinates__coveredby)querysetannotate_is_rejected_stationfilterrJmethodGETr:mapfloatsplitrPolygon from_bbox ValueError) rIrerbrcZ top_right_latZ top_right_lngZbottom_left_latZbottom_left_lngZ bounding_boxr.r.r3 get_querysets&    z StationModelViewSet.get_querysetcOs gd}tj||d}t|jS)as Retrieve a paginated list of station data based on request parameters. Args: request (HttpRequest): The HTTP request object. *args: Variable-length argument list. **kwargs: Arbitrary keyword arguments. Returns: Response: A paginated response containing a list of serialized station data. ) rFrr rr7r+ open_time close_timer%bookmarkratingassetsslugfieldsrget_paginated_responser data)rIrJrKrLrx serializerr.r.r3r>s  zStationModelViewSet.listcOs>|}tj||jdd}|jdd|t|jtjdS)Z Partially update a station record based on the provided data. Args: request (HttpRequest): The HTTP request object. *args: Variable-length argument list. **kwargs: Arbitrary keyword arguments. Returns: Response: The serialized data of the updated station record. Tr{partialraise_exceptionr) get_objectr#StationCreateUpdateMasterSerializerr{is_validsaver rrGrIrJrKrLstationr|r.r.r3partial_updates  z"StationModelViewSet.partial_updateFr:admin)detailmethodsurl_pathrWNcC gd}tj||d}t|jS)a5 Retrieve a paginated list of station data for admin purposes. Args: request (HttpRequest): The HTTP request object. pk: current object id, default null Returns: Response: A paginated response containing a list of serialized station data. )rFrrZall_network_operatorsrtrurvrwryrIrJpkrxr|r.r.r3get_admin_station_lists z*StationModelViewSet.get_admin_station_listzmap-pinscCr)a/ Retrieve a paginated list of station data for map pins. Args: request (HttpRequest): The HTTP request object. pk: current object id, default null Returns: Response: A paginated response containing a list of serialized station data. )rFrrr7rvrwryrr.r.r3 get_map_pinss z StationModelViewSet.get_map_pinsr-)rRrSrTrUrrWrStationMasterSerializerrX station_model StationMasterobjectsr?rerfilterset_class search_fieldshttp_method_namesrpr>rrr rr rr.r.r.r3rYs2 # rYc@s,eZdZdZefZejZe j j Z dZdS)StationRetrieveAPIViewz A view for retrieving details of a specific StationMaster object. This view allows retrieval of StationMaster objects by their slug field. It does not require authentication and can be accessed by any user (AllowAny permission). rvN)rRrSrTrUr rWrrrXrrrr?re lookup_fieldr.r.r.r3r4s  rc@s.eZdZdZefZejZe j j Z dgZdS)StationCreateUpdateModelViewSeta This is the StationCreateUpdateModelViewSet class responsible for handling CSV file uploads for station data. It provides endpoints for uploading and updating station data from CSV files. Attributes: permission_classes (tuple): Specifies the permissions required to access this view. It's set to IsAuthenticated, allowing only authenticated users to use this view. rQN)rRrSrTrUr rWrrrXrrrr?rerr.r.r.r3rBs    rc@sBeZdZdZeefZejZ e j j ZddgZddZddZdS) StationVerificationModelViewSeta+ This view handles the verification and update of station data from CSV file uploads. This view is responsible for managing the uploading and updating of station data from CSV files, specifically for the purpose of station verification. Attributes: permission_classes (tuple): Specifies the permissions required to access this view. It's set to (IsAuthenticated, IsAdminUser), allowing only authenticated and admin users to use this view. serializer_class (StationVerificationSerializer): The serializer used to validate and process station data. queryset (QuerySet): The initial queryset of station records that this viewset operates on. http_method_names (list): A list of the HTTP methods that this viewset supports, including 'GET' and 'PATCH'. r:r\cCs|jjddjddS)a9 Get the queryset of unverified station records. This method filters station records based on the 'is_verified' field, returning only those that are not yet verified (is_verified=False). Returns: QuerySet: The filtered queryset of unverified station records. Fr`r])rergrf)rIr.r.r3rpgs z,StationVerificationModelViewSet.get_querysetcOsV|}|j||jdd}|jdd|tjjj|dj ddt |jt j dS)r}Tr~rrr`r) rget_serializerr{rrr StationMediarrgupdater rrGrr.r.r3rxs  z.StationVerificationModelViewSet.partial_updateN)rRrSrTrUr r rWrZStationVerificationSerializerrXrrrr?rerrprr.r.r.r3rSs  rc@sFeZdZdZeefZejZ e j j ZeZgdZddZddZdS)StationMediaViewSetak This is the StationMediaViewSet class responsible for managing station media assets, such as images or files. It provides endpoints for uploading and deleting media assets associated with station records. Attributes: permission_classes (tuple): Specifies the permissions required to access this view. It's set to IsAuthenticated, allowing only authenticated users to use this view. parser_classes (tuple): The parsers used for request data, which includes MultiPartParser for handling file uploads. serializer_class (UploadStationMediaSerializer): The serializer class used for deserializing and handling media asset data. queryset (QuerySet): The queryset for accessing station media records. http_method_names (list): Specifies the supported HTTP methods for this view, including 'post' and 'delete'. Methods: create(request, *args, **kwargs): Handles the uploading of station media assets and associates them with a specific station. delete(request, *args, **kwargs): Handles the removal of station media assets. )r:rQr\r[cCs|jd}|jj|dS)a* Get the queryset of objects filtered by the associated station. This method retrieves the value of 'station_id' from the URL kwargs and filters the queryset based on the associated station. Returns: QuerySet: The filtered queryset of objects. station_idr)rLr:rerg)rIrr.r.r3rps z StationMediaViewSet.get_querysetc Os|jd}t|}|stdditjdSg}t|jD]\}}| |jd|d|j d|ddq|sEtdd itjdS|j j sft jjj|j |d } | t|d krftdd itjdS|j|d d|id} | r| j||j tjdt| j tjdSt| jtjdS)a Handles the uploading of station media assets and associates them with a specific station. Args: request (HttpRequest): The HTTP request object containing media assets and associated data. *args: Variable-length argument list. **kwargs: Arbitrary keyword arguments. Returns: Response: A response indicating the success or failure of the media asset upload operation. This method processes the uploaded media assets, associates them with a specified station, and handles the creation of station media records. It verifies the provided station ID, asset data, and the associated user. rrzInvalid station IDrzasset[]z asset_type[)asset asset_typezAssets are required)rErzYou can upload up-to 5 imagesTrJ)r{manycontext)rrEra)rLr:rZget_station_by_idr rr; enumerater9rDr{rE is_superuserrrrrgcountlenrrrrALLOW_AUTO_VERIFY_MEDIAHTTP_201_CREATEDerrors) rIrJrKrLrrruindexr station_mediar|r.r.r3createsZ   zStationMediaViewSet.createN)rRrSrTrUr rrWrZUploadStationMediaSerializerrXrrrr?rerrrrprr.r.r.r3rs  rc@s2eZdZdZeefZejZ e j j ZgdZdS)StationConnectorViewSeta A viewset for handling StationConnectors. This viewset allows CRUD operations (Create, Read, Update, Delete) for StationConnectors. It requires authentication and admin user permissions for access. Attributes: permission_classes (tuple): Permissions required to access this view. serializer_class: Serializer class for handling StationConnectors. queryset: The queryset containing all StationConnectors. http_method_names (list): The allowed HTTP methods for this viewset. )rQr[r\N)rRrSrTrUr r rWrZStationConnectorSerializerrXrStationConnectorsrr?rerr.r.r.r3rs    rc@s.eZdZdZefZeZej Z dgZ ddZ dS)StationPhotoAPIViewa API view to get a combined list of station media and rating media assets with pagination. Attributes: permission_classes (tuple): Permissions required to access this view (IsAuthenticated for authenticated users). pagination_class (CustomPagination): The custom pagination class for paginating the combined assets. serializer_class (CombinedAssetSerializer): The serializer used for serializing the combined assets. http_method_names (list): The allowed HTTP methods, in this case, only GET is allowed. Methods: get(request, *args, **kwargs): Retrieve the combined assets for a specific station based on the station ID. r:c Os|d}tjjj|ddddd}tjjj|ddddd}||}|D] }tj |d|d<q(t } | ||} |j | dd|id } | | jS) ak Retrieve the combined assets for a specific station based on the station ID. Args: request (HttpRequest): The HTTP request object. *args: Variable-length argument list. **kwargs: Arbitrary keyword arguments. Returns: Response: A paginated response containing the combined assets. rT)Z station__idrarrrF)Zrating_station__station__idrarJ)rr)r:rrrrgvaluesStationRatingMediaunionr MEDIA_URLrpaginate_querysetrXrzr{) rIrJrKrLrZstation_media_assetsZstation_rating_media_assetsZcombined_assetsZ asset_info paginationZ result_page serializersr.r.r3r:%s,       zStationPhotoAPIView.getN) rRrSrTrUr rWrpagination_classrZCombinedStationAssetSerializerrXrr:r.r.r.r3rs  rc@sfeZdZdZeefZejZ e j j ZeZgdZeddgddddZeddgd dd d Zd S) UnverifiedStationMediaViewSetaF This viewset provides endpoints to manage unverified station media. Attributes: permission_classes (tuple): Specifies the permissions required to access this view. serializer_class: The serializer class used for handling station media data. queryset (QuerySet): The queryset for accessing unverified station media records. filterset_class: The filterset class used for filtering station media records. http_method_names (list): Specifies the supported HTTP methods for this view, including 'get', 'delete', and 'patch'. rZFr[)rrrc Os||jdg}ztjjj|d\}}td|ditj dWSt y=}ztdt |itj dWYd}~Sd}~ww)aY Delete selected unverified media. Args: request (HttpRequest): The HTTP request object. *args: Variable-length argument list. **kwargs: Arbitrary keyword arguments. Returns: Response: A response indicating the success or failure of the media deletion operation. media_idsZid__inrz media deleted successfullyrerrorN) r{r:rrrrgr[r rHTTP_204_NO_CONTENT ExceptionstrHTTP_500_INTERNAL_SERVER_ERROR)rIrJrKrLrZ deleted_countrNer.r.r3delete_selected_mediad"  z3UnverifiedStationMediaViewSet.delete_selected_mediar\Zverifiedc Os||jdg}ztjjj|djdd}td|ditj dWSt y=}ztdt |itj dWYd }~Sd }~ww) a] Verify selected unverified media. Args: request (HttpRequest): The HTTP request object. *args: Variable-length argument list. **kwargs: Arbitrary keyword arguments. Returns: Response: A response indicating the success or failure of the media verification operation. rrTr`rz media verified successfullyrrN) r{r:rrrrgrr rrGrrr)rIrJrKrLr updated_countrr.r.r3verify_selected_mediarz3UnverifiedStationMediaViewSet.verify_selected_mediaN)rRrSrTrUr r rWrZStationMediaSerializerrXrrrr?rerrrrrrr.r.r.r3rQs   r)5rAZpandasr=django.contrib.gisrrest_frameworkrrest_framework.decoratorsrZrest_framework.genericsrrZrest_framework.parsersrZrest_framework.permissionsr r r rest_framework.responser rest_framework.viewsr Z apps.stationrrZ6apps.station.background_tasks.upload_station_csv_tasksrZ$apps.station.filters.station_filtersrrZapps.station.serializersrZapps.station.servicesrrZbase.paginatorrbase.permissionsrr base.viewsr station_evrrrYrrrrrrrr.r.r.r3s:            g1=n?