o di>9@sddlZddlmZddlmZddlmZddlmZddl m Z m Z ddl m ZddlmZdd lmZmZdd lmZd d Zd dZddZddZddZddZd+ddZddZddZddZdd Z d!d"Z!d#d$Z"d%d&Z#d'd(Z$d)d*Z%dS),N)ObjectDoesNotExist) ContentFile)Q)User)ConnectorMasterNetworkOperatorMaster)models) StationMedia)AssetTypeConstantsFileFieldConstants)settingscC<t|dkrgSdd|D}tjjj|djddd}|S)z Retrieve station IDs associated with a list of connector IDs. Args: connector_ids (list): A list of connector IDs to query. Returns: list: A list of station IDs that have the specified connector IDs. rcSg|] }|rt|qSisdigitint).0 connector_idrra/var/www/html/evchargy.com/django-station-ev/station_ev/apps/station/services/station_services.py  z4get_station_ids_by_connector_ids..)Zconnector__id__in station__idTflatlen station_modelStationConnectorsobjectsfilter values_list)Z connector_idsexisting_station_idsrrr get_station_ids_by_connector_idss r#cCs,tjjt|dt|d@jddd}|S)aU Retrieve station IDs associated with connectors within a specified kilowatt range. Args: min_kilowatts (float): The minimum kilowatt value. max_kilowatts (float): The maximum kilowatt value. Returns: list: A list of station IDs that have connectors with kilowatt values within the specified range. )Zkilowatts__gte)Zkilowatts__lterTr)rrrr rr!)Z min_kilowattsZ max_kilowattsr"rrr&get_station_ids_by_connector_kilowatts$s  r$cCr )z Retrieve station IDs associated with a list of location IDs. Args: location_ids (list): A list of location IDs to query. Returns: list: A list of station IDs that have the specified location IDs. rcSrrr)r location_idrrrrCrz3get_station_ids_by_location_ids..)location__id__inrTr)rrStationLocationrr r!) location_idsr"rrrget_station_ids_by_location_ids6s r)cCr )a Retrieve station IDs associated with a list of network operator IDs. Args: network_operator_ids (list): A list of network operator IDs to query. Returns: list: A list of station IDs that have the specified network operator IDs. rcSrrr)rnetwork_operator_idrrrr]s z;get_station_ids_by_network_operator_ids..)Znetwork_operator__id__inrTrr)Znetwork_operator_idsr"rrr'get_station_ids_by_network_operator_idsMs r+cCr )z Retrieve station IDs associated with a list of amenity IDs. Args: amenity_ids (list): A list of amenity IDs to query. Returns: list: A list of station IDs that have the specified amenity IDs. rcSrrr)rZ amenity_idrrrrxrz2get_station_ids_by_amenity_ids..)amenities__id__inrTr)rrStationAmenitiesrr r!) amenity_idsr"rrrget_station_ids_by_amenity_idsks r/cC(z tjjj|dWStyYdSw)z Retrieve a station record by its ID. Args: station_id (int): The ID of the station to retrieve. Returns: StationMaster: The station record corresponding to the provided station ID, or None if it does not exist. idN)r StationMasterrgetr) station_idrrrget_station_by_ids   r6cCs`|r$tjjj||d\}}|s"|D] \}}t|||q||Stjjjdi|}|S)a Create or update a station record based on scrap ID and validated data. Args: scarp_id (str): The scrap ID to identify the station. validated_data (dict): Validated data containing station details. Returns: StationMaster: The created or updated station record. This function creates or updates a station record based on the scrap ID. If a station with the given scrap ID exists, it updates the existing record with the provided data. If not, it creates a new station record. )scarp_iddefaultsNr)rr3r get_or_createitemssetattrsavecreate)r7validated_datastationcreatedkeyvaluerrrhandle_stationss rCcCstjjj||d\}}|S)a Create or retrieve a station user association. Args: station (StationMaster): The station record to associate with the user. user (User): The user to associate with the station. Returns: StationUser: The created or retrieved station user association. This function creates or retrieves a station user association, linking a station with a user. )r?user)r StationUserrr9)r?rD station_userr@rrrhandle_stations_users rGc Cstjjj|djddd}dd|D}t|t|}tjjj||d|D]}z tjjj||dWq)tt fyAYq)wd S) a? Handle amenities for a station, updating or creating StationAmenities records. Args: station (StationMaster): The station for which amenities are being managed. new_locations (list of LocationMaster): The list of new locations to be associated with the station. Returns: None r?Z location__idTrcSg|]}|jqSrr1)rlocationrrrrz$handle_locations..)r?r&)r?rJN) rr'rr r!setdeleter9r ValueError)r?Z new_locationsZexisting_location_idsr(Zlocations_to_deleterJrrrhandle_locationss(  rOcCsntjjj|djddd}dd|D}t|t|}tjjj||d|D] }tjjj||dq)d S) a@ Handle amenities for a station, updating or creating StationAmenities records. Args: station (StationMaster): The station for which amenities are being managed. new_amenities (list of AmenitiesMaster): The list of new amenities to be associated with the station. Returns: None rHZ amenities__idTrcSrIrr1)ramenityrrrrrKz$handle_amenities..)r?r,)r? amenitiesN)rr-rr r!rLrMr9)r?Z new_amenitiesZexisting_amenity_idsr.Zamenity_to_deleterPrrrhandle_amenitiess  rRcCsJtjjj|d|D]}tjjj||d|d|ddq dS)a Handle connectors for a station, updating StationConnectors records. Args: station (StationMaster): The station for which connectors are being managed. new_connectors (list of dict): The list of new connectors data to be associated with the station. Each item in the list is a dictionary with keys 'connector', 'kilowatts', and 'make' for the connector data. Returns: None rH connector kilowattsnetwork_operatorr?rSrTrUN)rrrr rMr=r4)r?new_connectorsconnector_datarrrhandle_update_connectorssrYcCs>|D]}|dD]}tjjj||d|d|ddqqdS)a Handle connectors for a station, updating or creating StationConnectors records. Args: station (StationMaster): The station for which connectors are being managed. new_connectors (list of dict): The list of new connectors data to be associated with the station. Each item in the list is a dictionary with keys 'connector', 'kilowatts', and 'make' for the connector data. Returns: None outletsrSrTmakerVN)rrrr=)r?rWrXoutletrrrhandle_connectors#s r]c Cs~|d}|d}|d}|d}|d}|r=|r=d|i}|r'||d<|s+|r1|p.||d<tjj||d\}}|SdS)a8 Get or create a network operator based on the provided data. Args: network_operator_data (dict): Data for the network operator, including 'id', 'name', 'phone', 'url', and 'action_url'. Returns: NetworkOperatorMaster: The created or existing network operator instance. r2namephoneurl action_urltype_idr8N)r4rrr9) Znetwork_operator_dataZ network_idr^r_r`rar8rUr=rrrget_or_create_network_operator<s      rdcCs>|d}|d}|r|rtjj|||dd\}}|SdS)a Get or create a connector based on the provided data. Args: connector_data (dict): Data for the connector, including 'connector' and 'connector_name'. Returns: ConnectorMaster: The created or existing connector instance. rSconnector_name)r^ short_namerbN)r4rrr9)rXrrerSr=rrrget_or_create_connector\s  rgcCr0)z Get or create a connector based on the provided data. Args: suggest_id (int): id of SuggestStation tabel Returns: ConnectorMaster: To get existing SuggestStation instance. r1N)rSuggestStationrr4r)Z suggest_idrrrget_suggest_stationss   ric Cs|rozYtj}|t|kr|d|}|D]D}|d}|dd}|tjvrXt|}|j dkrX|j }t ||ddd} t j j|d}tj j|| tj|dd} | qWdStyn} zWYd} ~ dSd} ~ wwdS) Nr`.)r^)pkT)r?asset asset_typerD is_verified)r EXTRACT_STATION_IMAGE_COUNTrr4splitlowerr IMAGE_FORMATSrequests status_codecontentrrrr r=r IMAGEr< Exception) r?rDZstation_photosZextract_image_countZphotoZfirst_photo_url extensionresponserwZ content_file station_mediaerrrhandle_station_medias>      r~)NN)&rudjango.core.exceptionsrdjango.core.files.baserdjango.db.modelsrZapps.account.modelsrapps.master.modelsrr apps.stationrrZapps.station.modelsr base.constantsr r station_evr r#r$r)r+r/r6rCrGrOrRrYr]rdrgrir~rrrrs4        %!