o gk@sddlmZddlmZddlmZddlmZmZm Z m Z m Z m Z m Z mZddlmZddlmZddlmZddlmZdd lmZdd lmZd d lmZmZmZmZdd lmZdd lm Z!dd l"m Z ddl#m$Z$m%Z%m&Z&ddl'm(Z(eZddl)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl*m2Z2ddlm3Z3ddl4m5Z5ddl6m7Z7ddl.m8Z8ddl9m:Z:ddl;mZ>ddl?Z?ddl@mAZAddlBmCZCddlDmEZEdd l9mFZFdd!lGmHZHdd"lImJZJe+d#ZKd$d%ZLd&d'ZMd(d)ZNGd*d+d+eZOGd,d-d-eZPGd.d/d/eZQGd0d1d1eZRGd2d3d3eZSGd4d5d5eZTGd6d7d7eZUGd8d9d9eZVGd:d;d;eZWGdd?d?ejXZZGd@dAdAeZ[GdBdCdCeZ\GdDdEdEeZ]dS)F)Responsestatus)APIView)UserRegistrationSerializerUserLoginSerializerUserChangePasswordSerializer SendPasswordResetEmailSerializerUserPasswordResetSerializerUserTypeSerializerUserDetailSerializersend_verification_email) UserRenderer) RefreshToken)TokenRefreshView)IsAuthenticated)generics)get_user_model)UserusertypeCodeConfirmationEmailVerification)ValidationError)success_responsefailed_responsegenerate_random_text)JWTAuthenticationN)logger_settings)validate_email)validate_required_params)record)validation_msg)AllowAny)settings)render_to_stringis_emailrender) strip_tags)EmailMultiAlternatives)urlsafe_base64_encode)default_token_generator) force_bytes)get_object_or_404) timedelta)timezonemsgcCst|}t|t|jdS)N)refreshaccess)rfor_userstr access_token)userr3r9;/var/www/html/authentication-server/authentication/views.pyget_tokens_for_user6s r;cC t|dS)Nz login.htmlr(requestr9r9r: login_page= r?cCr<)Nz register.htmlr(r=r9r9r: register_page@r@rAc@eZdZddZdS)CustomTokenRefreshViewc Osn|jd}zt|}|j}tdt|itjdWSty6}ztdt|itj dWYd}~Sd}~ww)Nr3r4rdetail) datagetrr7rr6r HTTP_200_OK ExceptionHTTP_400_BAD_REQUEST)selfr>argskwargs refresh_tokenr3new_access_tokener9r9r:postDs "zCustomTokenRefreshView.postN__name__ __module__ __qualname__rPr9r9r9r:rCC rCc@rB)UserRegistrationViewc Ostd}z;t|jd}|r7|}td}ttdd}tj j |||dt |j |t dtjdWSt|jtjdWStyZ} ztt| tjdWYd} ~ Sd} ~ ww)NrEminutesr8code expires_atzGRegistration successful. Please check your email for verification code.r2r])rrrEis_validsaver1nowr0robjectscreater emailrrHTTP_201_CREATEDrerrorsrIrHr6HTTP_500_INTERNAL_SERVER_ERROR) rJr>rKrL process_code serializerr8verification_coder^rOr9r9r:rPNs8  zUserRegistrationView.postNrQr9r9r9r:rVMrUrVc@rB)VerifyEmailViewcCs|jd}|jd}z:tjj|d}tjj||dtd}|s,t dt j dWSd|_ t|_ |d|_|td t jdWStjyWt d t jdYSw) Nrer]reF)r8r]is_usedexpires_at__gtz$Invalid or expired verification coder_Tz/Email verified successfully. You can now login.User not found)rErFrrcrfilterr1rbfirstrrrI is_activeemail_verified_atrarnrrG DoesNotExistHTTP_404_NOT_FOUND)rJr>rerkr8 verificationr9r9r:rPss@    zVerifyEmailView.postNrQr9r9r9r:rlrrUrlc@rB)ResendVerificationCodeViewcCs|jd}zAtjj|d}|jrtdtjdWStd}t t dd}t jj |djd d t jj|||d t|j|td tjdWStjyXtd tjdYSw)NrermzEmail is already verifiedr_rWrYrZ)r8T)rnr\z4New verification code sent. Please check your email.rp)rErFrrcrsrrrIrr1rbr0rrqupdaterdr rerrGrurv)rJr>rer8rkr^r9r9r:rPs6   zResendVerificationCodeView.postNrQr9r9r9r:rxrUrxc@eZdZdddZdS) UserLoginViewNc CsTtd}td|dtddzt|jd}|jddrtd|dtd |jd }|jd }|jd }d}|rbtjj |d  }|satd|dt dt t dt jdWSn#|rtjj |d }|std|dt dt t dt jdWS|r||std|dt d t t d t jdWS|jst dt jdWSt|} td|dtddtd| iddWStd|d|jt |jt jdWSty} ztd|dt| dt t| t jdWYd} ~ Sd} ~ wty)} ztd|dt| dt dt jdWYd} ~ Sd} ~ ww)NrWinfo: startedz - User Login StartedrXTraise_exception validatedusernamerepasswordrmwarn invalid_emailr_rinvalid_usernamez+Please verify your email before logging in. main_processz - User login successfultokenz Login Success)rEr2errorz - ValidationError occurredz - Unexpected error occurredzAn unexpected error occurred)rr!log_msgrrEr`rFrrcrqrrr"rrrIcheck_passwordrsHTTP_403_FORBIDDENr;rrgDjangoValidationErrorr6rHrh) rJr>formatrirjrrerr8rrOr9r9r:rPas      zUserLoginView.postNrQr9r9r9r:r{`r{c@s"eZdZegZegZdddZdS)UserChangePasswordViewNc Cstd}td|dtddz7t|jd|jid}|jdd r5td|dtd d td d WStd|d|jdt dt j dWSt ys}ztd|dt |dt dt |t jdWYd}~Sd}~ww)NrWr|r}r~z. - UserChangePasswordView POST request startedr8rEcontextTr completedz - Password changed successfullyzPassword Changed Successfullyr2rz+ - Validation errors during password changez1Validation errors occurred during password changer_rz3 - Unexpected error occurred during password changeAn unexpected error occurred: )rr!rrrEr8r`rrgrrrIrHr6rh)rJr>rrirjrOr9r9r:rPsH   zUserChangePasswordView.postr)rRrSrTrauthentication_classesrpermission_classesrPr9r9r9r:rsrc@s*eZdZegZegZdddZddZdS)SendPasswordResetEmailViewNc Cs0td}td|dtddz[t|jd}|jddrY|jd }tjj |d }t |}t t |j}d |d |} ||| td|dtd d|tddWStd|d|jdtdtjdWSty} ztd|dt| dtdt| tjdWYd} ~ Sd} ~ ww)NrWr|r}r~z2 - SendPasswordResetEmailView POST request startedrXTrrermz/http://127.0.0.1:8000/user-auth/reset-password//rz- - Password reset email sent successfully to z2Password Reset Link Sent. Please check your email.rrz8 - Validation errors during password reset email requestz>Validation errors occurred during password reset email requestr_rz. - Please try again with a valid email addressr)rr!rr rEr`validated_datarrcrFr- make_tokenr,r.pksend_password_reset_emailrrgrrrIrHr6rh) rJr>rrirjrer8ruid reset_linkrOr9r9r:rPsP      zSendPasswordResetEmailView.postcCsRd}tdd|i}t|}tjd}t||||g}||d|jdddS)NzYour Password Reset Linkzpassword_reset.htmlr EMAIL_USERz text/htmlF) fail_silently)r%r*osenvironrFr+attach_alternativesend)rJrersubject html_message plain_message from_email email_messager9r9r:rs  z4SendPasswordResetEmailView.send_password_reset_emailr) rRrSrTrrenderer_classesr#rrPrr9r9r9r:rs   +rc@seZdZegZdddZdS)UserPasswordResetViewNc Cstd}td|dtddzC|r|stdtjdWSt|j||dd }|jd d rAtd|dtd d t ddWStd|d|j dtdtjdWSt y}ztd|dt |dtdt |tj dWYd}~Sd}~ww)NrWr|r}r~z- - UserPasswordResetView POST request startedz#Missing token or uid in the requestr_)rrrTrrz - Password reset successfullyzPassword Reset Successfullyrrz2 - Validation errors during password reset requestz8Validation errors occurred during password reset requestrz: - Unexpected error occurred during password reset requestr)rr!rrrrIr rEr`rrgrHr6rh)rJr>rrrrirjrOr9r9r:rP sR   zUserPasswordResetView.postr)rRrSrTrrrPr9r9r9r:rsrc@rB) LogoutViewc Ostd}td|dtdd|jd}|s(td|dtd tjd Szt|}| td|dtd d td tj d WSt yp}ztd|dt |dt dt |tjd WYd}~Sd}~ww)NrWr|r}r~z" - LogoutView POST request startedrMrz: Refresh token is missingzRefresh token is required.r_rz - Logout successfulzLogout successful.rz* - Unexpected error occurred during logoutz#An internal server error occurred: )rr!rrErFrrrIr blacklistrGrHr6r)rJr>rKrLrirMrrOr9r9r:rP7s6 $zLogoutView.postNrQr9r9r9r:r6rUrc@s eZdZeZddZddZdS)UserTypeListViewcCs2tj}|jjdd}|dur|j|d}|S)Nsubtyper)rrcallr> query_paramsrFrq)rJquerysetrr9r9r: get_querysetUs  zUserTypeListView.get_querysetcOsntd}|}|j|dd}|r(td|dtddt|jtj dStd |d t d tj dS) NrWTmanyr|r}rz" - User Types fetched successfullyr_rz+: No user types found for the given subtypez%No user types found for given subtype) rrget_serializerexistsr!rrrErrGrrI)rJr>rKrLrirrjr9r9r:list\szUserTypeListView.listN)rRrSrTr serializer_classrrr9r9r9r:rRs rc@seZdZeZddZdS)UserTypeBySubtypeViewcCsTtd}tjj|d}|j|dd}td|dtdd|d t|jt j d S) NrWrTrr|r}rz% - f"User types fetched for subtype: "r_) rrrcrqrr!rrrErrG)rJr>rri user_typesrjr9r9r:rFmszUserTypeBySubtypeView.getN)rRrSrTr rrFr9r9r9r:rjs rc@seZdZddZddZdS)!CheckUniqueUsernameOrEmailAPIViewc Os$td}zc|j}d|vr@|d}||r0tjj|dr/td|dt dt j dWSntd|dt d t j dWSd |vrb|d }tjj|d rbtd|d t d t j dWSt ddWSt y}ztd|dt|t dt|t jdWYd}~Sd}~ww)NrWrermrz: Email is already registeredzEmail is already registered.r_z: Invalid email formatzInvalid email format.rrz: Username is already takenzUsername is already taken.z&Both email and username are available.rrz: Unexpected error occurred: r)rrErFr'rrcrqrr!rrrIrrHr6rh) rJr>rKrLrirErerrOr9r9r:rPws.    $z&CheckUniqueUsernameOrEmailAPIView.postcCst|Srr&)rJvaluer9r9r:r'sz*CheckUniqueUsernameOrEmailAPIView.is_emailN)rRrSrTrPr'r9r9r9r:rvs rc@rz) UserCheckViewNc Cstd}ztjj|d}td|d|dtd|jdtjdWStj yAtd |d|d td d d tj dYSw)NrW)idr|z: User with ID z foundT)ruser_idrrz not foundFrp)rmessage) rrrcrFr!rrrrGrurv)rJr>rrrir8r9r9r:rFszUserCheckView.getr)rRrSrTrFr9r9r9r:rrrc@seZdZegZddZdS)UserDetailViewcOsBtd}|jd}|stStt|d}t|}t|jtjdS)NrWrermr) rrErFrr/rr rrG)rJr>rKrLrirer8rjr9r9r:rPs  zUserDetailView.postN)rRrSrTrrrPr9r9r9r:rs r)^rest_framework.responserrest_frameworkrrest_framework.viewsrauthentication.serializersrrrr r r r r authentication.renderersrrest_framework_simplejwt.tokensrrest_framework_simplejwt.viewsrrest_framework.permissionsrrdjango.contrib.authrmodelsrrrrdjango.core.exceptionsrrutils.exception utils.helperrrr'rest_framework_simplejwt.authenticationrre utils.datardjango.core.validatorsrutils.validationr utils.loggerr!r"r# django.confr$django.template.loaderr%r'django.shortcutsr)django.utils.htmlr*django.core.mailr+rdjango.utils.httpr,django.contrib.auth.tokensr-django.utils.encodingr.r/datetimer0 django.utilsr1rr;r?rArCrVrlrxr{rrrr ListAPIViewrrrrrr9r9r9r:sp  (                            %'HD)90