o di-:@s ddlmZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZGdddeZGdddeZGdddeZGdddeZ GdddeZ!GdddeZ"dS))ObjectDoesNotExiststatus)action)RetrieveAPIView)AllowAnyIsAuthenticated IsAdminUser)Response)models)StationCheckInListFilter)StationRatingListFilterStationRatingMediaListFilter)"station_checkIn_rating_serializers)AdminOrPostDeleteOnlyPermission)BaseModelViewSet)settingsc@s:eZdZdZejZejj Z ddgZ ddZ ddZdS) CreateStationRatingModelViewSeta This viewset allows authenticated users to create station rating records. Attributes: permission_classes (tuple): Permissions required to access this view (IsAuthenticated for authenticated users). serializer_class (StationRatingSerializer): The serializer used for input and output data. queryset (QuerySet): The base queryset for station rating records. http_method_names (list): The allowed HTTP methods, in this case, only POST and GET are allowed. Methods: get_queryset(self): Retrieves a queryset of station rating records for the requesting user. perform_create(serializer): Associates a newly created station rating record with the requesting user. postgetcCs|jj|jjdS)z Retrieves a queryset of station rating records for the requesting user. Returns: QuerySet: A filtered queryset containing station rating records associated with the user. user)querysetfilterrequestr)selfrj/var/www/html/evchargy.com/django-station-ev/station_ev/apps/station/views/station_checkIn_rating_views.py get_queryset)sz,CreateStationRatingModelViewSet.get_querysetc Ost|jdkrtdditjdS|j|jd}|jdd|j|j j dg}t |jD]\}}| |j d |d |j d |d d q,tj|dd |id}|jddtjjj |jdd} |j| tjd|j| d |id}t|jtjdS)aQ Create a new StationRating instance and associated StationRatingAssets instances. Args: request (Request): The HTTP request object. *args: Variable-length argument list. **kwargs: Arbitrary keyword arguments. Returns: Response: The HTTP response object. messageYou can upload up-to 5 imagesr)dataT)raise_exceptionrasset[] asset_type[asset asset_typerr"manycontextid)pkrating_station is_verified)r,)lenFILESr rHTTP_400_BAD_REQUESTget_serializerr"is_validsaverr enumerateappendrrRatingMediaSerializerr StationRatingobjectsrALLOW_AUTO_VERIFY_MEDIAHTTP_201_CREATED) rrargskwargs serializerassetsindexfileassets_serializerstation_ratingrrrcreate2s2    z&CreateStationRatingModelViewSet.createN)__name__ __module__ __qualname____doc__rStationRatingSerializerserializer_classr r;r<allrhttp_method_namesrrGrrrrrs  rc@sFeZdZdZeefZejZ e j j ZeZgdZddZddZdS)CreateRatingMediaModelViewSeta  This viewset allows authenticated users to upload media assets for a specific station rating. Attributes: permission_classes (tuple): Permissions required to access this view (IsAuthenticated for authenticated users). serializer_class (RatingMediaSerializer): The serializer used for handling media asset uploads. parser_classes (tuple): Parsers used for request data (MultiPartParser for file uploads). queryset (QuerySet): The base queryset for station rating media records. http_method_names (list): The allowed HTTP methods, in this case, only POST is allowed. Methods: create(request, *args, **kwargs): Handles the uploading of station media assets and associates them with a specific station rating. )rrpatchdeletec OsR|jd}ztjjj|d}|j|jjkrtdditj dWSWnt y2tdditj dYSwg}t |j D]\}}| |j d|d|jd |dd q:|sbtdd itj dS|jjstjjj|j|d } | t|d krtdditj dS|j|dd|id} | r| j|tjdt| jtjdSt| jtj dS)a Handles the uploading of station media assets and associates them with a specific station rating. 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 rating, and handles the creation of station media records. It verifies the provided check-in ID, asset data, and the associated user. rating_id)r-r z?You do not have permission to upload other users' rating media.rzInvalid rating IDr$r%r&r'zAssets are required)rating_station__userr0rr!Trr*r/)r@r r;r<rrrr rr4rr8r3r9r" is_superuserStationRatingMediarcountr2r5r6r7rr=r>errors) rrr?r@rSZ checkin_objrBrCrD station_mediarArrrrGxsh    z$CreateRatingMediaModelViewSet.createcOst|jd}ztjjj||j|jdd}Wntjjy+tdditj dYSw| |tdditj dS)a\ Remove a media from the user's 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 removal operation. rSr.)r0rTr-r zMedia not foundrzMedia deleted successful) r@rr rVr<r DoesNotExistr rHTTP_404_NOT_FOUNDperform_destroyHTTP_204_NO_CONTENT)rrr?r@rSZrating_station_mediarrrdestroys$     z%CreateRatingMediaModelViewSet.destroyN)rHrIrJrKrrpermission_classesrr:rMr rVr<rNrrfilterset_classrOrGr^rrrrrPbs  NrPc@s:eZdZdZefZejZe j j Z eZdgZddZdS)StationRatingModelViewSetas This viewset allows public access to retrieve station rating records associated with a specific station. Attributes: permission_classes (tuple): Permissions required to access this view (AllowAny for public access). serializer_class (StationRatingSerializer): The serializer used for station rating records. queryset (QuerySet): The base queryset for station rating records. http_method_names (list): The allowed HTTP methods, in this case, only GET is allowed. Methods: get_queryset(): Retrieves station rating records for a specific station based on the station ID. rcCs|jd}|jj|dS)az Retrieves station rating records for a specific station based on the station ID. Returns: QuerySet: A filtered queryset containing station rating records associated with the specified station. This method filters the base queryset to retrieve station rating records associated with the station specified by the station ID. station_id) station__id)r@rr)rrbrrrrs z&StationRatingModelViewSet.get_querysetN)rHrIrJrKrr_rrLrMr r;r<rNrr r`rOrrrrrras   rac@s(eZdZdZefZejZe j j Z dS) StationCheckInCalculationAPIViewz A view for retrieving Station Check-In calculations. This view allows retrieval of StationMaster data with specific permissions. N)rHrIrJrKrr_rZ#StationCheckInCalculationSerializerrMr StationMasterr<rNrrrrrrd s rdc@s6eZdZdZejZejj Z e Z ddgZddZdS)StationCheckInModelViewSetz A viewset for handling Station Check-In operations. This viewset supports creating StationCheckIn objects via HTTP POST. rrcCs|j|jjddS)a5 Perform the creation of a StationCheckIn object. This method is called when a new object is created. It associates the object with the current user. Args: serializer (Serializer): The serializer for creating the object. Returns: None rN)r7rr)rrArrrperform_create$s z)StationCheckInModelViewSet.perform_createN)rHrIrJrKrZStationCheckInSerializerrMr StationCheckInr<rNrr r`rOrgrrrrrfs  rfc@sfeZdZdZeefZejZ e j j ZeZgdZeddgddddZeddgd dd d Zd S) #UnverifiedStationRatingMediaViewSetab This viewset provides endpoints to manage unverified station rating media. Attributes: permission_classes (tuple): Specifies the permissions required to access this view. serializer_class: The serializer class used for handling station rating media data. queryset (QuerySet): The queryset for accessing unverified station rating media records. filterset_class: The filterset class used for filtering station rating media records. http_method_names (list): Specifies the supported HTTP methods for this view, including 'get', 'delete', and 'patch'. )rrRrQFrR)detailmethodsurl_pathc Os||jdg}ztjjj|d\}}td|ditj dWSt y=}ztdt |itj dWYd}~Sd}~ww)ag Delete selected unverified rating 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 rating media deletion operation. media_idsid__inr z media deleted successfullyrerrorN) r"rr rVr<rrRr rr] ExceptionstrHTTP_500_INTERNAL_SERVER_ERROR)rrr?r@rm deleted_count_errrdelete_selected_mediaF"  z9UnverifiedStationRatingMediaViewSet.delete_selected_mediarQverifiedc Os||jdg}ztjjj|djdd}td|ditj dWSt y=}ztdt |itj dWYd }~Sd }~ww) ak Verify selected unverified rating 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 rating media verification operation. rmrnT)r1r z media verified successfullyrrpN) r"rr rVr<rupdater r HTTP_200_OKrqrrrs)rrr?r@rm updated_countrvrrrverify_selected_mediabrxz9UnverifiedStationRatingMediaViewSet.verify_selected_mediaN)rHrIrJrKrr r_rr:rMr rVr<rNrrr`rOrrwr}rrrrri3s   riN)#django.core.exceptionsrrest_frameworkrrest_framework.decoratorsrZrest_framework.genericsrZrest_framework.permissionsrrr rest_framework.responser Z apps.stationr Z,apps.station.filters.station_checkin_filtersr $apps.station.filters.station_filtersr rZapps.station.serializersrbase.permissionsr base.viewsr station_evrrrPrardrfrirrrrs(          L "