o di?@sddlmZddlmZddlmZddlmZddlm Z m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZmZdd lmZdd lmZddlmZddlmZGdddeZ GdddeZ!GdddeZ"GdddeZ#GdddeZ$dS))settings)ObjectDoesNotExiststatus) CreateAPIView)AllowAnyIsAuthenticated)Response)APIView) TokenError) RefreshToken)ProfileListFilter)UserUserCode)account_serializers)BaseModelViewSet) sending_email)send_fast2sms_otpc@*eZdZdZefZejZdgZ ddZ dS)LoginOrRegistrationViewa This is the LoginOrRegistrationView. It handles user login or registration through a POST request. Attributes: permission_classes (tuple): Permissions required to access this view (AllowAny for public access). serializer_class (UserLoginOrRegistrationSerializer): The serializer used for input/output data. http_method_names (list): The allowed HTTP methods, in this case, only POST is allowed. Methods: post(request): Handles user login or registration logic. postc Os`tj|jd}|jdd|jd}|jd}|jd}|r@|r@z tjj||d}Wnty?tjj|||d}Ynw|r^z tjj|d }Wnty]tjj |||d }Ynwt jj |d } z|rst d | j tj|gd |r~|r~t|| j dWnty} ztdt| itjdWYd} ~ Sd} ~ wwt|}ddi|j} t| tjdS)z Handles user login or registration logic. Args: request (HttpRequest): The HTTP request containing user data. Returns: Response: The serialized user data or error response. dataTraise_exception mobile_numbermobile_country_codeemailrr)rrrr)rrruserzYour EVChargy Login OTPsubjectotp_code from_emailrecipient_listrotp custom_errorrNmessagezOTP sent successfully)r!UserLoginOrRegistrationSerializerris_validgetrobjectsrcreate create_userrcreate_or_update_codercoder FROM_EMAILr Exceptionr strrHTTP_400_BAD_REQUEST HTTP_200_OK) selfrequestargskwargs serializerrrrr! user_codeeresponser@[/var/www/html/evchargy.com/django-station-ev/station_ev/apps/account/views/account_views.pyr's`        " zLoginOrRegistrationView.postN __name__ __module__ __qualname____doc__rpermission_classesrr+serializer_classhttp_method_namesrr@r@r@rArs  rc@s*eZdZdZefZdgZejZ ddZ dS) VerifyLoginOrRegistrationOtpViewae This view handles OTP verification for user login or registration. Attributes: permission_classes (tuple): Permissions required to access this view (AllowAny for public access). http_method_names (list): The allowed HTTP methods (only POST is allowed). Methods: post(request): Handles user OTP verification logic. rc Os\tj|jd}|jdd|jd}|jd}|jd}|jd}|rF|rFz tjj||d} WntyEtd d it j d YSw|rdz tjj|d } Wntyctd d it j d YSwzt jj| |d} | r{td dit j d WSWntytd dit j d YSwt | } | j| jdu| j| j| jt| jt| d} t| S)a Handles OTP verification logic for user login or registration. Args: request (HttpRequest): The HTTP request containing user data. Returns: Response: The serialized user data or an error response. rTrrrrr(rr*zMUser Not Found: The mobile number you provided does not exist in our records.rrzEUser Not Found: The email you provided does not exist in our records.)r!r2z&OTP has been expired please try again!z*Invalid OTP! Please enter the correct OTP.N)idZ is_new_userrrrtokenrefresh)rVerifyOtpSerializerrr,r-rr.rr rr6r is_expiredr for_userrKnamerrrr5 access_token) r8r9r:r;r<rrrr(r!r=rM response_datar@r@rArysh               z%VerifyLoginOrRegistrationOtpView.postN) rCrDrErFrrGrIrrNrHrr@r@r@rArJis   rJc@r) ResendOtpViewa This view handles user login or registration and OTP resending for authentication. Attributes: permission_classes (tuple): Permissions required to access this view (AllowAny for public access). serializer_class (UserLoginOrRegistrationSerializer): The serializer used for input/output data. http_method_names (list): The allowed HTTP methods, in this case, only POST is allowed. Methods: post(request): Handles user login or registration and OTP resending logic. This view provides functionality for user login or registration, as well as resending OTP codes for authentication purposes. rc OsZtj|jd}|jdd|jd}|jd}|jd}|r@|r@z tjj||d}Wnty?tdd it j d YSw|r^z tjj|d }Wnty]tdd it j d YSwt jj |d } z|rst d| jtj|gd|r~|r~t|| jdWnty} ztdt| it j d WYd} ~ Sd} ~ wwt|}|j} d| d<t| S)a Handles user login or registration and OTP resending logic. Args: request (HttpRequest): The HTTP request containing user data. Returns: Response: The serialized user data or error response. This method processes user login or registration and resending OTP logic based on the provided mobile number and mobile country code. rTrrrrrr*z(The mobile number you entered is invalidrrz The email you entered is invalidr z&Resend mail for Login/Registration OTPr"r'r)Nz+OTP Sent Successfully to your email address)rr+rr,r-rr.rr rr6rr1rr2rr3rr4r5) r8r9r:r;r<rrrr!r=r>rSr@r@rArsd        " zResendOtpView.postNrBr@r@r@rArTs  rTc@sLeZdZdZejZej Z e Z ddgZ gdZddZddZd d Zd S) ProfileModelViewSeta This view handles OTP verification for user login or registration and user profile retrieval. Attributes: permission_classes (tuple): Permissions required to access this view (IsAuthenticated for authenticated users). http_method_names (list): The allowed HTTP methods (PUT and GET are allowed). Methods: list(request): Retrieves a list of user profiles (Not allowed, returns an error response). retrieve(request): Retrieves a specific user's profile details, with permission checks. rQr)patchr-deletecOsn|jjr0||}||}|dur!|j|dd}||jS|j|dd}t|jt j dStgt j dS)a Retrieves a list of user data based on the request parameters. Args: request (HttpRequest): The HTTP request object. *args: Variable-length argument list. **kwargs: Arbitrary keyword arguments. Returns: Response: A response containing a list of serialized user data. This method retrieves a list of users records based on the provided request parameters. The list of stations is paginated for efficient retrieval. NT)manyr) r! is_superuserfilter_queryset get_querysetpaginate_querysetget_serializerget_paginated_responserr rr7)r8r9r:r;querysetpager<r@r@rAlist2s  zProfileModelViewSet.listcOsH|}||jkr|jjstdditjdS|j|d|id}t|jS)a Retrieve a specific user's profile details, with permission checks. Args: request (HttpRequest): The HTTP request for retrieving a user's profile. Returns: Response: The user's profile details or an error response if permission is denied. This method retrieves the profile details of a specific user, with permission checks to ensure that the requesting user has access to the requested profile. errorz9You do not have permission to access this user's profile.rr9)context) get_objectr!rYr rHTTP_403_FORBIDDENrHr)r8r9r:r;instancer<r@r@rAretrieveNs zProfileModelViewSet.retrievecOsH|}||jkr|jjstdditjdS||tdditjdS)a Handles the removal of account. Args: request (HttpRequest): The HTTP request object containing information about the account to delete. *args: Variable-length argument list. **kwargs: Arbitrary keyword arguments. Returns: Response: A response indicating the success or failure of the account deletion operation. r*z#You can't delete other user accountrzAccount deleted successfully)rdr!rYr rr6perform_destroyHTTP_204_NO_CONTENT)r8r9r:r;rfr@r@rAdestroygs  zProfileModelViewSet.destroyN)rCrDrErFrZProfileSerializerrHrr.allr_r filterset_class search_fieldsrIrargrjr@r@r@rArUs  rUc@seZdZdZefZddZdS)UserLogoutViewa This is the UserLogoutView class responsible for handling user logout through a POST request. It allows users to log out and blacklist the provided refresh token, ensuring the token can no longer be used. Attributes: permission_classes (tuple): Permissions required to access this view (IsAuthenticated for authenticated users). Methods: post(request): Handles user logout and token blacklisting logic. c Csz|jd}|rt|}|ttjdWSty)tdditjdYSt yR}zddl }t | tdt | d tjdWYd}~Sd}~ww) a' Handles user logout and token blacklisting logic. Args: request (HttpRequest): The HTTP request containing user data, including the refresh token. Returns: Response: A response indicating the success or failure of the logout operation. rMrr*z Invalid token)rrrNz&There was some issue while logging out)r* traceback)rr-r blacklistr rr7r r6r4roprint format_excr5)r8r9Z refresh_tokenrLr>ror@r@rArs.     zUserLogoutView.postN)rCrDrErFrrGrr@r@r@rArns rnN)% django.confrdjango.core.exceptionsrrest_frameworkrZrest_framework.genericsrZrest_framework.permissionsrrrest_framework.responser rest_framework.viewsr Z#rest_framework_simplejwt.exceptionsr Zrest_framework_simplejwt.tokensr Z$apps.account.filters.profile_filtersr Zapps.account.modelsrrZapps.account.serializersrZ base.viewsrZaccount.utils.send_emailrZaccount.utils.send_otprrrJrTrUrnr@r@r@rAs(            UZ\c