o di@sddlmZddlmZddlmZddlmZddlm Z ddl m Z m Z ddl mZddlmZGd d d eZGd d d eZd S))ObjectDoesNotExiststatus)MultiPartParser)Response)user_vehicle_filters) UserVehicleUserVehicleAssets)user_vehicle_serializers)BaseModelViewSetc@sNeZdZdZejZej Z e j Z gdZddZddZddZd d Zd S) UserVehicleModelViewSeta This view manages user vehicle records, allowing user profile retrieval and vehicle management. Attributes: permission_classes (tuple): Permissions required to access this view (IsAuthenticated for authenticated users). parser_classes (tuple): Parsers used for request data (MultiPartParser for file uploads). serializer_class (user_vehicle_serializers.UserVehicleSerializer): The serializer class for user vehicle data. queryset (QuerySet): The set of user vehicle records that this viewset operates on. http_method_names (list): The allowed HTTP methods (POST, PUT, GET, DELETE are allowed). Methods: list(request): Retrieves a list of user vehicles (Not allowed, returns an error response). retrieve(request): Retrieves a specific user's vehicle details with permission checks. destroy(request): Deletes a specific user's vehicle with permission checks. )postputgetdeletecCs|jj|jjdS)a Filter user vehicle records based on the requesting user. Returns: QuerySet: User vehicle records belonging to the requesting user. This method filters the queryset to retrieve user vehicle records specific to the requesting user. user)querysetfilterrequestr)selfr`/var/www/html/evchargy.com/django-station-ev/station_ev/apps/account/views/user_vehicle_views.py get_queryset"s z$UserVehicleModelViewSet.get_querysetcCs|j|jjddS)a% Perform user vehicle record creation, associating it with the requesting user. Args: serializer (UserVehicleSerializer): The serializer for creating a new user vehicle. This method associates the created user vehicle record with the requesting user. rN)saverr)r serializerrrrperform_create-s z&UserVehicleModelViewSet.perform_createcOsB|}|j|jkrtdditjdS|j|d|id}t|jS)a Retrieve a specific user's vehicle details with permission checks. Args: request (HttpRequest): The HTTP request for retrieving a user's vehicle. Returns: Response: The user's vehicle details or an error response if permission is denied. This method retrieves the details of a specific user's vehicle, with permission checks to ensure that the requesting user has access to the requested vehicle. errorz4You do not have permission to see other user vehiclerr)context) get_objectrrrHTTP_403_FORBIDDENserializer_classdata)rrargskwargsinstancerrrrretrieve9s  z UserVehicleModelViewSet.retrievecOs@|}|j|jkrtddiddS||tdditjdS)a Delete a specific user's vehicle with permission checks. Args: request (HttpRequest): The HTTP request for deleting a user's vehicle. Returns: Response: A success message or an error response if permission is denied. This method deletes a specific user's vehicle, with permission checks to ensure that the requesting user has the authority to delete the vehicle. messagez#You can't delete other user vehiclerzVehicle deleted successfully)rrrperform_destroyrHTTP_204_NO_CONTENTrrr#r$r%rrrdestroyQs  zUserVehicleModelViewSet.destroyN)__name__ __module__ __qualname____doc__r ZUserVehicleSerializerr!robjectsallrrZUserVehicleListFilterfilterset_classhttp_method_namesrrr&r,rrrrr s   r c@s>eZdZdZejZefZe j Z ddgZ ddZddZdS) UserVehicleAssetsModelViewSeta_ A viewset for handling CRUD operations on UserVehicleAssets instances. This viewset supports the creation and deletion of UserVehicleAssets instances. It includes validation to ensure that users can only upload and delete their own vehicle media. Attributes: serializer_class: The serializer class for the UserVehicleAssets model. parser_classes (tuple): A tuple of parser classes, including MultiPartParser. queryset: The queryset containing all UserVehicleAssets instances. http_method_names (list): List of HTTP methods allowed on this viewset. r rc Os|jd}ztjj|d}|j|jjkrtdditjdWSWnt y2tdditj dYSwg}t |j D]\}}| d|j d|d iq:|jt|d krbtdd itjdS|smtdd itjdS|j|d d} | r| j|dtdditjdSt| jtjdS)a. Create UserVehicleAssets instances. Handles HTTP POST method to create UserVehicleAssets instances. Validates permissions to ensure users can only upload their own vehicle media. Returns: Response: The HTTP response indicating success or failure. user_vehicle_id)idr'z?You do not have permission to upload other users' rating media.rzUser vehicle dose not existassetzasset[]zYou can upload up-to 5 imageszAssets are requiredT)r"many) user_vehiclez$Vehicle media uploaded successfully.)r$rrr1rrrrHTTP_400_BAD_REQUESTrHTTP_404_NOT_FOUND enumerateFILESappenduser_vehicle_assetscountlenget_serializeris_validrHTTP_201_CREATEDerrors) rrr#r$r6r<ZassetsindexfilerrrrcreatesV    z$UserVehicleAssetsModelViewSet.createcOsB|}|j|jjkrtddiddS||tdditjdS)a0 Delete UserVehicleAssets instances. Handles HTTP DELETE method to delete UserVehicleAssets instances. Validates permissions to ensure users can only delete their own vehicle media. Returns: Response: The HTTP response indicating success or failure. r'z)You can't delete other user vehicle mediar(rz"Vehicle media deleted successfully)rrr<rr)rr*r+rrrr,s  z%UserVehicleAssetsModelViewSet.destroyN)r-r.r/r0r ZUserVehicleAssetSerializerr!rparser_classesr r1r2rr4rKr,rrrrr5os  >r5N)django.core.exceptionsrrest_frameworkrZrest_framework.parsersrrest_framework.responserZapps.account.filtersrZapps.account.modelsrr apps.account.serializersr base.viewsr r r5rrrrs      c