o
    [h67                     @   s:  d dl mZmZ d dlmZ d dlmZmZ d dlmZ	 d dlm
Z d dlmZ d dlmZmZ d dlmZmZmZmZ d d	lmZ d d
lmZ d dlmZmZ d dlmZmZ d dlm Z  d dl!m"Z" d dl#m$Z$ d dl%m&Z&m'Z' d dl(m)Z* d dl+m,Z, d dl-m.Z. d dl/m0Z0 d dl1m2Z2 d dl3m4Z4 e Z5G dd dZ6e$eddG dd de6e4Z7G dd de6e2Z8d3d d!Z9defd"d#Z:G d$d% d%Z;e$eddG d&d' d'e;e4Z<d(Z=e$eddG d)d* d*e;e2Z>e$eddG d+d, d,e;e4Z?e$eddG d-d. d.e;e2Z@G d/d0 d0e;e4ZAG d1d2 d2e;e2ZBdS )4    )urlsplit
urlunsplit)settings)REDIRECT_FIELD_NAMEget_user_model)login)logout)update_session_auth_hash)login_not_requiredlogin_required)AuthenticationFormPasswordChangeFormPasswordResetFormSetPasswordForm)default_token_generator)get_current_site)ImproperlyConfiguredValidationError)HttpResponseRedirect	QueryDict)resolve_url)reverse_lazy)method_decorator)url_has_allowed_host_and_schemeurlsafe_base64_decode)gettext_lazy)never_cache)csrf_protect)sensitive_post_parameters)TemplateView)FormViewc                   @   s:   e Zd ZdZeZe Zdd Zdd Z	dd Z
dd	 ZdS )
RedirectURLMixinNc                 C   s   |   p|  S N)get_redirect_urlget_default_redirect_urlself r'   /home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/django/contrib/auth/views.pyget_success_url(   s   z RedirectURLMixin.get_success_urlc                 C   sB   | j j| j| j j| j}t||  | j  d}|r|S dS )z6Return the user-originating redirect URL if it's safe.)urlZallowed_hostsZrequire_https )requestPOSTgetredirect_field_nameGETr   get_success_url_allowed_hosts	is_secure)r&   redirect_toZurl_is_safer'   r'   r(   r#   +   s   z!RedirectURLMixin.get_redirect_urlc                 C   s   | j  h| jS r"   )r,   get_hostsuccess_url_allowed_hostsr%   r'   r'   r(   r1   7   s   z.RedirectURLMixin.get_success_url_allowed_hostsc                 C   s   | j rt| j S td) Return the default redirect URL.z+No URL to redirect to. Provide a next_page.)	next_pager   r   r%   r'   r'   r(   r$   :   s   
z)RedirectURLMixin.get_default_redirect_url)__name__
__module____qualname__r7   r   r/   setr5   r)   r#   r1   r$   r'   r'   r'   r(   r!   #   s    r!   dispatch)namec                       s~   e Zd ZdZeZdZdZdZdZ	e
e e
ee
e fddZdd Zd	d
 Z fddZdd Z fddZ  ZS )	LoginViewz=
    Display the login form and handle the login action.
    Nzregistration/login.htmlFc                    sN   | j r| jjjr|  }|| jjkrtdt|S t j	|g|R i |S )NzsRedirection loop for authenticated user detected. Check that your LOGIN_REDIRECT_URL doesn't point to a login page.)
redirect_authenticated_userr,   useris_authenticatedr)   path
ValueErrorr   superr<   r&   r,   argskwargsr3   	__class__r'   r(   r<   M   s   zLoginView.dispatchc                 C   s   | j rt| j S ttjS r6   )r7   r   r   ZLOGIN_REDIRECT_URLr%   r'   r'   r(   r$   [   s   

z"LoginView.get_default_redirect_urlc                 C   s   | j p| jS r"   )authentication_form
form_classr%   r'   r'   r(   get_form_classb   s   zLoginView.get_form_classc                       t   }| j|d< |S )Nr,   )rD   get_form_kwargsr,   r&   rG   rH   r'   r(   rO   e      

zLoginView.get_form_kwargsc                 C   s   t | j|  t|  S )z)Security check complete. Log the user in.)
auth_loginr,   get_userr   r)   r&   formr'   r'   r(   
form_validj   s   zLoginView.form_validc                    sH   t  jdi |}t| j}|| j|  d|d|ji| jpi  |S )Nsite	site_namer'   )	rD   get_context_datar   r,   updater/   r#   r=   extra_contextr&   rG   contextZcurrent_siterH   r'   r(   rY   o   s   

zLoginView.get_context_data)r8   r9   r:   __doc__r   rL   rK   template_namer?   r[   r   r   r   r   r<   r$   rM   rO   rV   rY   __classcell__r'   r'   rH   r(   r>   A   s     r>   c                       s`   e Zd ZdZddgZdZdZeeee	 fddZ
 fdd	Zd
d Z fddZ  ZS )
LogoutViewzH
    Log out the user and display the 'You are logged out' message.
    postoptionszregistration/logged_out.htmlNc                    s   t  j|g|R i |S r"   rD   r<   )r&   r,   rF   rG   rH   r'   r(   r<      s   zLogoutView.dispatchc                    s>   t | |  }|| krt|S t j|g|R i |S )zLogout may be done via POST.)auth_logoutr)   Zget_full_pathr   rD   r.   rE   rH   r'   r(   rb      s
   zLogoutView.postc                 C   s(   | j rt| j S tjrttjS | jjS rJ   )r7   r   r   ZLOGOUT_REDIRECT_URLr,   rB   r%   r'   r'   r(   r$      s
   

z#LogoutView.get_default_redirect_urlc                    sD   t  jdi |}t| j}|||jtdd d| jpi  |S )Nz
Logged out)rW   rX   titlesubtitler'   )rD   rY   r   r,   rZ   r=   _r[   r\   rH   r'   r(   rY      s   
	zLogoutView.get_context_data)r8   r9   r:   r^   Zhttp_method_namesr_   r[   r   r   r   r<   rb   r$   rY   r`   r'   r'   rH   r(   ra   }   s    		ra   Nc                 C   s   t |ptj}tj|d| S )zR
    Log out the user if they are logged in. Then redirect to the login page.
    )r7   )r   r   	LOGIN_URLra   Zas_view)r,   	login_urlr'   r'   r(   logout_then_login   s   rk   c                 C   sR   t |ptj}tt|}|r#t|d dd}| ||< |jdd|d< tt|S )zM
    Redirect the user to the login page, passing the given 'next' page.
       T)Zmutable/)safe)	r   r   ri   listr   r   	urlencoder   r   )nextrj   r/   Zresolved_urlZlogin_url_partsZquerystringr'   r'   r(   redirect_to_login   s   rr   c                       s    e Zd ZdZ fddZ  ZS )PasswordContextMixinNc                    s2   t  jdi |}|| jd d| jpi  |S )N)rf   rg   r'   )rD   rY   rZ   rf   r[   r&   rG   r]   rH   r'   r(   rY      s
   z%PasswordContextMixin.get_context_data)r8   r9   r:   r[   rY   r`   r'   r'   rH   r(   rs      s    rs   c                       s`   e Zd ZdZdZeZdZdZdZ	e
dZdZedZeZee fddZ fd	d
Z  ZS )PasswordResetViewz&registration/password_reset_email.htmlNz'registration/password_reset_subject.txtZpassword_reset_donez%registration/password_reset_form.htmlzPassword resetc                       t  j|i |S r"   rd   r&   rF   rG   rH   r'   r(   r<         zPasswordResetView.dispatchc              	      sF   | j  | j| j| j| j| j | j| jd}|jdi | t	 
|S )N)Z	use_httpstoken_generator
from_emailemail_template_namesubject_template_namer,   html_email_template_nameextra_email_contextr'   )r,   r2   ry   rz   r{   r|   r}   r~   saverD   rV   )r&   rU   optsrH   r'   r(   rV      s   
zPasswordResetView.form_valid)r8   r9   r:   r{   r~   r   rL   rz   r}   r|   r   success_urlr_   rh   rf   r   ry   r   r   r<   rV   r`   r'   r'   rH   r(   ru      s    ru   Z_password_reset_tokenc                   @   s   e Zd ZdZedZdS )PasswordResetDoneViewz%registration/password_reset_done.htmlzPassword reset sentN)r8   r9   r:   r_   rh   rf   r'   r'   r'   r(   r      s    r   c                       s   e Zd ZeZdZdZdZedZ	dZ
edZeZee ee fddZd	d
 Z fddZ fddZ fddZ  ZS )PasswordResetConfirmViewFNzset-passwordZpassword_reset_completez(registration/password_reset_confirm.htmlzEnter new passwordc                    s   d|vsd|vrt dd| _| |d | _| jd ur\|d }|| jkrA| jjt}| j	
| j|r@d| _t j|i |S n| j	
| j|r\|| jjt< | jj|| j}t|S | |  S )Nuidb64tokenz:The URL path must contain 'uidb64' and 'token' parameters.FT)r   	validlinkrS   r@   reset_url_tokenr,   sessionr.   INTERNAL_RESET_SESSION_TOKENry   Zcheck_tokenrD   r<   rB   replacer   Zrender_to_responserY   )r&   rF   rG   r   Zsession_tokenZredirect_urlrH   r'   r(   r<   
  s*   

z!PasswordResetConfirmView.dispatchc                 C   sT   zt | }tjj|}tjj|d}W |S  tt	t
tjtfy)   d }Y |S w )N)pk)r   decode	UserModel_metar   Z	to_pythonZ_default_managerr.   	TypeErrorrC   OverflowErrorZDoesNotExistr   )r&   r   uidr   r@   r'   r'   r(   rS   ,  s   	
z!PasswordResetConfirmView.get_userc                    rN   Nr@   )rD   rO   r@   rP   rH   r'   r(   rO   <  rQ   z(PasswordResetConfirmView.get_form_kwargsc                    s4   |  }| jjt= | jrt| j|| j t |S r"   )	r   r,   r   r   post_reset_loginrR   post_reset_login_backendrD   rV   )r&   rU   r@   rH   r'   r(   rV   A  s
   
z#PasswordResetConfirmView.form_validc                    s>   t  jdi |}| jrd|d< |S |d tddd |S )NTr   zPassword reset unsuccessfulF)rU   rf   r   r'   )rD   rY   r   rZ   rh   rt   rH   r'   r(   rY   H  s   	z)PasswordResetConfirmView.get_context_data)r8   r9   r:   r   rL   r   r   r   r   r   r_   rh   rf   r   ry   r   r   r   r<   rS   rO   rV   rY   r`   r'   r'   rH   r(   r      s      r   c                       s(   e Zd ZdZedZ fddZ  ZS )PasswordResetCompleteViewz)registration/password_reset_complete.htmlzPassword reset completec                    s$   t  jdi |}ttj|d< |S )Nrj   r'   )rD   rY   r   r   ri   rt   rH   r'   r(   rY   \  s   z*PasswordResetCompleteView.get_context_data)r8   r9   r:   r_   rh   rf   rY   r`   r'   r'   rH   r(   r   W  s    r   c                       sf   e Zd ZeZedZdZedZ	e
e e
ee
e fddZ fddZ fdd	Z  ZS )
PasswordChangeViewZpassword_change_donez&registration/password_change_form.htmlzPassword changec                    rv   r"   rd   rw   rH   r'   r(   r<   h  s   zPasswordChangeView.dispatchc                    s   t   }| jj|d< |S r   )rD   rO   r,   r@   rP   rH   r'   r(   rO   n  s   
z"PasswordChangeView.get_form_kwargsc                    s"   |   t| j|j t |S r"   )r   r	   r,   r@   rD   rV   rT   rH   r'   r(   rV   s  s   zPasswordChangeView.form_valid)r8   r9   r:   r   rL   r   r   r_   rh   rf   r   r   r   r   r<   rO   rV   r`   r'   r'   rH   r(   r   b  s    r   c                       s0   e Zd ZdZedZee fddZ  Z	S )PasswordChangeDoneViewz&registration/password_change_done.htmlzPassword change successfulc                    rv   r"   rd   rw   rH   r'   r(   r<     rx   zPasswordChangeDoneView.dispatch)
r8   r9   r:   r_   rh   rf   r   r   r<   r`   r'   r'   rH   r(   r   {  s
    r   r"   )Curllib.parser   r   Zdjango.confr   Zdjango.contrib.authr   r   r   rR   r   re   r	   Zdjango.contrib.auth.decoratorsr
   r   Zdjango.contrib.auth.formsr   r   r   r   Zdjango.contrib.auth.tokensr   Zdjango.contrib.sites.shortcutsr   Zdjango.core.exceptionsr   r   Zdjango.httpr   r   Zdjango.shortcutsr   Zdjango.urlsr   Zdjango.utils.decoratorsr   Zdjango.utils.httpr   r   Zdjango.utils.translationr   rh   Zdjango.views.decorators.cacher   Zdjango.views.decorators.csrfr   Zdjango.views.decorators.debugr   Zdjango.views.generic.baser   Zdjango.views.generic.editr    r   r!   r>   ra   rk   rr   rs   ru   r   r   r   r   r   r   r'   r'   r'   r(   <module>   sR    
;
/



W
