o Fh@@sddlmZddlmZmZddlmZddlmZddl m Z m Z ddl m Z mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZddlmZddlmZddl Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'Z'ddl(Z(GdddeZ)GdddeZ*GdddeZ+GdddeZ,GdddeZ-ddZ.Gd d!d!eZ/Gd"d#d#eZ0Gd$d%d%eZ1Gd&d'd'eZ2Gd(d)d)eZ3Gd*d+d+eZ4Gd,d-d-eZ5dS).status)APIViewexception_handler)Response)Token) CustomUserOTP)CustomUserSerializer OTPSerializer)get_object_or_404)timezone) authenticate) timedelta) AccessToken)AuthenticationFailedNotAuthenticated)JWTAuthentication)IsAuthenticatedN) send_mail)reverse)settingsc@eZdZdZddZdS)AdminLoginAPIViewz? View to authenticate admin users and return JWT Token c Os|jd}|jd}z+t||d}|dur-|jr-t|}tddt|dtj dWStdd d tj dWSt yY}ztdd t|t d tjdWYd}~Sd}~ww) N mobile_numberpassword)rrrzLogin successful)rmessagetokenrrz+Invalid credentials or user is not an adminrrzAn unexpected error occured)rrexc traceback)datagetris_staffrfor_userrstrr HTTP_200_OKHTTP_401_UNAUTHORIZED Exceptionr! format_excHTTP_400_BAD_REQUEST) selfrequestargskwargsrruserrer2$/var/www/html/loop/accounts/views.pyposts.    zAdminLoginAPIView.postN__name__ __module__ __qualname____doc__r4r2r2r2r3r rc@r) UserLoginViewz; View for user login (also used for registration). cCsb|jd}tjj|d\}}tjj|dt dd}tjj ||dt dd|d t j d S) Nrrr0ii')r0otp_coderzOTP sent successfully!)rrotpr)r"r#r objects get_or_creater filterdeleterandomrandintcreaterrr')r,r-rr0createdr>r2r2r3r4<s   zUserLoginView.postNr5r2r2r2r3r;7r:r;c@r)OTPVerificationViewz$ View for OTP verification. c CszW|jd}|jd}ttdd}tjj|||dd }|rMt t |d}t jj |d\}}|j } |td d |j| |jd tjd WStd ddtjd WStys} ztd ddtjd WYd} ~ Sd} ~ ww)Nrr>)minutes)user__mobile_numberr>created_at__gtez -created_atr<r=rzOTP verified successfully!)rrrnew_user is_verifiedrrzInvalid or expired OTP.rAn unexpected error occurred.)r"r#rnowrr r@rBorder_byfirstr r rrA first_namerCrkeyrNrr'r)r+) r,r-rr>two_minutes_agor?r0rrGrMr1r2r2r3r4PsJ    zOTPVerificationView.postNr5r2r2r2r3rHKr:rHc@r)UserIsVerifiedViewz0 View to check if the user is verified. c Cs|jdd}z tjj|d}|j}|jrddd}nddd}t|tj dWStj y@dd d}t|tj dYSt y]}zdd d}t|tj dWYd}~Sd}~ww) NHTTP_AUTHORIZATIONrrTzUser is verified.rrzUser is not verifiedrInvalid or expired token.rO)METAr#splitrr@r0rNrrr' DoesNotExistr(r)r+)r,r- token_keyrr0 response_datar1r2r2r3r#s     zUserIsVerifiedView.getNr6r7r8r9r#r2r2r2r3rV|r:rVc@r)UserUpdateViewz* View to update user information. c Cs|jdd}z4tjj|d}|j}t||jdd}|r2| t dd|jdt j dWSt d d |j d t j dWStjyRt d d d t jdYStyo}zd dd }t |t jdWYd}~Sd}~ww)NrWrrXTr"partialz&User information updated successfully.rrr"rrz Invalid data.rrerrors Bad request: Token is not valid.rrO)rZr#r[rr@r0r r"is_validsaverrr'rer\r(r)r+)r,r-r]rr0 serializerr1r^r2r2r3patchs>   zUserUpdateView.patchN)r6r7r8r9rjr2r2r2r3r`r:r`cCs(t||}t|trddd|_|SdS)NrrYr)r isinstancerr")r contextresponser2r2r3custom_exception_handlers  rnc@$eZdZdZegZegZddZdS) UserListViewz5 View to list users with optional filtering. c Csxz|j}tjjd i|}t|dd}tdd|jdtjdWSty;}ztddd tj dWYd}~Sd}~ww) NT)manyrzUsers retrieved successfullyrcrrAn unexpected error occurredrr2) r"r r@rBr rrr'r)HTTP_500_INTERNAL_SERVER_ERROR)r,r- filter_paramsusersrir1r2r2r3r4s$ zUserListView.postN r6r7r8r9rauthentication_classesrpermission_classesr4r2r2r2r3rp  rpc@ro)CreateDriverViewz& View to create a new driver. c Csz(t|jd}|r|jddtdddtjdWStdd |jd tjdWSt yF}zdd d}t|tjdWYd}~Sd}~ww) N)r"T) is_driverrzDriver created successfullyrrr Invalid datardAn unexpected error occured.) r r"rgrhrrHTTP_201_CREATEDrer+r))r,r-rir1 respose_datar2r2r3r4s*   zCreateDriverView.postNrvr2r2r2r3rzryrzc@ro)DeleteDriverViewz( View to delete a driver by ID. c Csztjj|d}|tdddtjdWStjy*tdddtjdYSt yE}ztdddtj dWYd}~Sd}~ww) NidrzDriver deleted successfullyrrrDriver not foundrr) r r@r#rCrrr'r\HTTP_404_NOT_FOUNDr)rs)r,r- driver_iddriverr1r2r2r3rC s& zDeleteDriverView.deleteN) r6r7r8r9rrwrrxrCr2r2r2r3rryrc@ro)UpdateDriverViewz2 View to update driver information by ID. c Csz1tjj|d}t||jdd}|r%|tdd|jdtj dWStdd |j d tj dWStj yEtdd d tj dYStyb}zdd d }t|tj dWYd}~Sd}~ww)NrTrarz'Driver information updated successfullyrcrrr|rdrrr})r r@r#r r"rgrhrrr'rer+r\rr))r,r-rrrir1rr2r2r3rj(sB   zUpdateDriverView.patchN) r6r7r8r9rrwrrxrjr2r2r2r3r ryrc@ro)DriverDetailViewz4 View to retrieve driver information by ID. c Csztjj|d}t|}tdd|jdtjdWStjy,tdddtj dYSt yG}ztdd dtj dWYd}~Sd}~ww) Nrrz)Driver information retrieved successfullyrcrrrrrr) r r@r#r rr"rr'r\rr)rs)r,r-rrrir1r2r2r3r#Vs, zDriverDetailView.getN) r6r7r8r9rrwrrxr#r2r2r2r3rNryrc@r)SendVerificationEmailViewz8 View to send a verification email to the user. c Cs|jdd}zEtjj|d}|j}t}t||_ t |_ |j ddgd|td|gd}td d |tj|jgd d tdd |dtjdWStjyctdddtjdYSty}zddt|d}t|tjdWYd}~Sd}~ww)NrWrrXverification_tokentoken_created_at update_fieldsz verify-email)r.zVerify your accountz1Please click on the link to verify your account: F) fail_silentlyz%Verification email sent successfully.)rrzverification linkrrrfrrOrr exception)rZr#r[rr@r0uuiduuid4r&rrrPrrhbuild_absolute_urirrrDEFAULT_FROM_EMAILemailrrr'r\r(r)r+) r,r-r]rr0rverification_linkr1r^r2r2r3r4ssL    zSendVerificationEmailView.postNr5r2r2r2r3rnr:rc@r)VerifyEmailViewzH View to handle email verification with token expiration check. c CszLtjj|d}t|j}|tddkr!tdddtj dWS|rB|j sBd|_ d|_ d|_|j gd d tdd dtj dWStdd dtj dWStjy`tdd dtj dYSty~}ztddt|dtj dWYd}~Sd}~ww)N)rr)daysrzVerification token has expired.rrT)rNrrrzEmail verified successfully.zEmail is already verified.zInvalid verification token.rOr)r r@r#rrPrrrrr+rNrrhr'r\r)r&)r,r-rr0 token_ager1r2r2r3r#sL  zVerifyEmailView.getNr_r2r2r2r3rr:r)6rest_frameworkrrest_framework.viewsrrrest_framework.responserrest_framework.authtoken.modelsrmodelsr r serializersr r django.shortcutsr django.utilsrdjango.contrib.authrdatetimerrest_framework_simplejwt.tokensrrest_framework.exceptionsrr'rest_framework_simplejwt.authenticationrrest_framework.permissionsrrDdjango.core.mailr django.urlsr django.confrrr!rr;rHrVr`rnrprzrrrrrr2r2r2r3sB              1+ . .