U
    Ha=                     @   sx  d dl Z d dlmZ d dlmZ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 d d	lmZ d d
lmZ d dlmZ d dlmZ d dlmZ d dlmZmZ e Z dd Z!G dd dej"Z#G dd dej$Z%G dd dej&Z'G dd dej(Z)G dd dej(Z*G dd dej+Z,G dd dej+Z-G dd  d ej+Z.G d!d" d"e.Z/G d#d$ d$ej+Z0dS )%    N)forms)authenticateget_user_modelpassword_validation)UNUSABLE_PASSWORD_PREFIXidentify_hasher)User)default_token_generator)get_current_site)ValidationError)EmailMultiAlternatives)loader)force_bytes)urlsafe_base64_encode)capfirst)gettextgettext_lazyc                 C   s    t d|  t d| kS )z
    Perform case-insensitive comparison of two identifiers, using the
    recommended algorithm from Unicode Technical Report 36, section
    2.11.2(B)(2).
    NFKC)unicodedata	normalizecasefold)s1s2 r   =/tmp/pip-unpacked-wheel-3jxiddxt/django/contrib/auth/forms.py_unicode_ci_compare   s    r   c                       s$   e Zd ZdZdZ fddZ  ZS )ReadOnlyPasswordHashWidgetz)auth/widgets/read_only_password_hash.htmlTc           	         s   t  |||}g }|r"|tr6|dtdi n`zt|}W n& tk
rh   |dtdi Y n.X ||	 D ]\}}|t||d qx||d< |S )NlabelzNo password set.z5Invalid password format or unknown hashing algorithm.)r   valuesummary)
superget_context
startswithr   appendr   r   
ValueErrorZsafe_summaryitems)	selfnamer   attrscontextr   ZhasherkeyZvalue_	__class__r   r   r!   %   s    z&ReadOnlyPasswordHashWidget.get_context)__name__
__module____qualname__Ztemplate_nameZ	read_onlyr!   __classcell__r   r   r+   r   r   !   s   r   c                       s    e Zd ZeZ fddZ  ZS )ReadOnlyPasswordHashFieldc                    s*   | dd | dd t j|| d S )NrequiredFdisabledT)
setdefaultr    __init__r&   argskwargsr+   r   r   r5   9   s    z"ReadOnlyPasswordHashField.__init__)r-   r.   r/   r   widgetr5   r0   r   r   r+   r   r1   6   s   r1   c                       s(   e Zd Z fddZ fddZ  ZS )UsernameFieldc                    s   t dt |S )Nr   )r   r   r    	to_python)r&   r   r+   r   r   r;   @   s    zUsernameField.to_pythonc                    s   t  |dddS )Nnoneusername)Zautocapitalizeautocomplete)r    widget_attrs)r&   r9   r+   r   r   r?   C   s    
zUsernameField.widget_attrs)r-   r.   r/   r;   r?   r0   r   r   r+   r   r:   ?   s   r:   c                       s   e Zd ZdZdediZejeddejddide	
 d	Zejed
ejddiddeddZG dd dZ fddZdd Z fddZd fdd	Z  ZS )UserCreationFormzc
    A form that creates a user, with no privileges, from the given username and
    password.
    password_mismatch'   The two password fields didn’t match.PasswordFr>   new-passwordr(   )r   stripr9   	help_textzPassword confirmation4Enter the same password as before, for verification.r   r9   rF   rG   c                   @   s   e Zd ZeZdZdeiZdS )zUserCreationForm.Meta)r=   r=   Nr-   r.   r/   r   modelfieldsr:   Zfield_classesr   r   r   r   Meta`   s   rM   c                    s:   t  j|| | jjj| jkr6d| j| jjj jjd< d S )NT	autofocus)r    r5   _metarK   USERNAME_FIELDrL   r9   r(   r6   r+   r   r   r5   e   s    zUserCreationForm.__init__c                 C   s>   | j d}| j d}|r:|r:||kr:t| jd dd|S N	password1	password2rA   code)cleaned_datagetr   error_messagesr&   rR   rS   r   r   r   clean_password2j   s    z UserCreationForm.clean_password2c              
      s`   t    | jd}|r\zt|| j W n. tk
rZ } z| d| W 5 d }~X Y nX d S )NrS   )	r    _post_cleanrV   rW   r   validate_passwordinstancer   Z	add_error)r&   passworderrorr+   r   r   r[   t   s    
zUserCreationForm._post_cleanTc                    s.   t  jdd}|| jd  |r*|  |S )NF)commitrR   )r    saveset_passwordrV   )r&   r`   userr+   r   r   ra      s
    zUserCreationForm.save)T)r-   r.   r/   __doc___rX   r   	CharFieldPasswordInputr   "password_validators_help_text_htmlrR   rS   rM   r5   rZ   r[   ra   r0   r   r   r+   r   r@   K   s*    
r@   c                       s>   e Zd ZeededdZG dd dZ fddZ  ZS )UserChangeFormrC   u   Raw passwords are not stored, so there is no way to see this user’s password, but you can change the password using <a href="{}">this form</a>.)r   rG   c                   @   s   e Zd ZeZdZdeiZdS )zUserChangeForm.Meta__all__r=   NrJ   r   r   r   r   rM      s   rM   c                    sN   t  j|| | jd}|r,|jd|_| jd}|rJ|jd|_d S )Nr^   z../password/user_permissionscontent_type)r    r5   rL   rW   rG   formatZquerysetZselect_related)r&   r7   r8   r^   rk   r+   r   r   r5      s    zUserChangeForm.__init__)	r-   r.   r/   r1   re   r^   rM   r5   r0   r   r   r+   r   ri      s   	ri   c                       s   e Zd ZdZeejddiddZeje	ddej
dd	idd
Ze	de	ddZd fdd	Zdd Zdd Zdd Zdd Z  ZS )AuthenticationFormzs
    Base class for authenticating users. Extend this to get a form that accepts
    username/password logins.
    rN   TrE   )r9   rC   Fr>   current-passwordr   rF   r9   z^Please enter a correct %(username)s and password. Note that both fields may be case-sensitive.zThis account is inactive.)invalid_logininactiveNc                    s|   || _ d| _t j|| tjtj| _| jj	p4d}|| j
d _	|| j
d jjd< | j
d jdkrxt| jj| j
d _dS )z
        The 'request' parameter is set for custom auth use by subclasses.
        The form data comes in via the standard 'data' kwarg.
        N   r=   Z	maxlength)request
user_cacher    r5   	UserModelrO   	get_fieldrP   username_field
max_lengthrL   r9   r(   r   r   verbose_name)r&   rt   r7   r8   Zusername_max_lengthr+   r   r   r5      s    zAuthenticationForm.__init__c                 C   s\   | j d}| j d}|d k	rV|rVt| j||d| _| jd krJ|  n| | j | j S )Nr=   r^   )r=   r^   )rV   rW   r   rt   ru   get_invalid_login_errorconfirm_login_allowed)r&   r=   r^   r   r   r   clean   s    

zAuthenticationForm.cleanc                 C   s   |j st| jd dddS )a  
        Controls whether the given User may log in. This is a policy setting,
        independent of end-user authentication. This default behavior is to
        allow login by active users, and reject login by inactive users.

        If the given user cannot log in, this method should raise a
        ``ValidationError``.

        If the given user may log in, this method should return None.
        rr   rT   N)	is_activer   rX   )r&   rc   r   r   r   r|      s
    z(AuthenticationForm.confirm_login_allowedc                 C   s   | j S N)ru   r&   r   r   r   get_user   s    zAuthenticationForm.get_userc                 C   s   t | jd dd| jjidS )Nrq   r=   )rU   params)r   rX   rx   rz   r   r   r   r   r{      s
    
z*AuthenticationForm.get_invalid_login_error)N)r-   r.   r/   rd   r:   r   Z	TextInputr=   rf   re   rg   r^   rX   r5   r}   r|   r   r{   r0   r   r   r+   r   rn      s"   rn   c                	   @   sZ   e Zd ZejeddejddiddZddd	Zd
d Z	dddde
ddddf	ddZdS )PasswordResetFormZEmailrs   r>   emailrE   )r   ry   r9   Nc                 C   sb   t ||}d| }t ||}t||||g}	|dk	rVt ||}
|	|
d |	  dS )zO
        Send a django.core.mail.EmailMultiAlternatives to `to_email`.
         Nz	text/html)r   Zrender_to_stringjoin
splitlinesr   Zattach_alternativesend)r&   subject_template_nameemail_template_namer)   
from_emailZto_emailhtml_email_template_namesubjectbodyZemail_messageZ
html_emailr   r   r   	send_mail   s    zPasswordResetForm.send_mailc                    s6   t  t jjf d  ddi} fdd|D S )a  Given an email, return matching user(s) who should receive a reset.

        This allows subclasses to more easily customize the default policies
        that prevent inactive users and users with unusable passwords from
        resetting their password.
        z
%s__iexactr~   Tc                 3   s*   | ]"}|  rt t|r|V  qd S r   )Zhas_usable_passwordr   getattr).0ur   email_field_namer   r   	<genexpr>  s   z.PasswordResetForm.get_users.<locals>.<genexpr>)rv   get_email_field_nameZ_default_managerfilter)r&   r   Zactive_usersr   r   r   	get_users  s      zPasswordResetForm.get_usersz'registration/password_reset_subject.txtz&registration/password_reset_email.htmlFc
              	   C   s   | j d }
|s$t|}|j}|j}n| }}t }| |
D ]X}t||}|||tt	|j
||||rpdndd|	p|i }| j||||||d q>dS )zf
        Generate a one-use only link for resetting password and send it to the
        user.
        r   httpshttp)r   domain	site_nameuidrc   tokenprotocol)r   N)rV   r
   r'   r   rv   r   r   r   r   r   pkZ
make_tokenr   )r&   Zdomain_overrider   r   Z	use_httpsZtoken_generatorr   rt   r   Zextra_email_contextr   Zcurrent_siter   r   r   rc   Z
user_emailr)   r   r   r   ra     s6    




    zPasswordResetForm.save)N)r-   r.   r/   r   Z
EmailFieldre   Z
EmailInputr   r   r   r	   ra   r   r   r   r   r      s$    
   r   c                       s   e Zd ZdZdediZejedejddidde	
 d	Zejed
dejddiddZ fddZdd ZdddZ  ZS )SetPasswordFormza
    A form that lets a user change set their password without entering the old
    password
    rA   rB   zNew passwordr>   rD   rE   FrI   zNew password confirmationrp   c                    s   || _ t j|| d S r   rc   r    r5   r&   rc   r7   r8   r+   r   r   r5   P  s    zSetPasswordForm.__init__c                 C   sL   | j d}| j d}|r:|r:||kr:t| jd ddt|| j |S )Nnew_password1new_password2rA   rT   rV   rW   r   rX   r   r\   rc   rY   r   r   r   clean_new_password2T  s    z#SetPasswordForm.clean_new_password2Tc                 C   s*   | j d }| j| |r$| j  | jS )Nr   rV   rc   rb   ra   r&   r`   r^   r   r   r   ra   `  s
    

zSetPasswordForm.save)T)r-   r.   r/   rd   re   rX   r   rf   rg   r   rh   r   r   r5   r   ra   r0   r   r   r+   r   r   <  s$    r   c                   @   sV   e Zd ZdZejdediZejeddej	dddd	d
Z
dddgZdd ZdS )PasswordChangeFormz[
    A form that lets a user change their password by entering their old
    password.
    password_incorrectzAYour old password was entered incorrectly. Please enter it again.zOld passwordFro   Tr>   rN   rE   rp   old_passwordr   r   c                 C   s,   | j d }| j|s(t| jd dd|S )zB
        Validate that the old_password field is correct.
        r   r   rT   )rV   rc   Zcheck_passwordr   rX   )r&   r   r   r   r   clean_old_passwordy  s    
z%PasswordChangeForm.clean_old_passwordN)r-   r.   r/   rd   r   rX   re   r   rf   rg   r   Zfield_orderr   r   r   r   r   r   h  s    
r   c                       s   e Zd ZdZdediZdZejedej	dddd	d
e
 dZejedej	ddid	d
eddZ fddZdd ZdddZe fddZ  ZS )AdminPasswordChangeFormzN
    A form used to change the password of a user in the admin interface.
    rA   rB   r2   rC   rD   Tr   rE   FrI   zPassword (again)r>   rH   c                    s   || _ t j|| d S r   r   r   r+   r   r   r5     s    z AdminPasswordChangeForm.__init__c                 C   sL   | j d}| j d}|r:|r:||kr:t| jd ddt|| j |S rQ   r   rY   r   r   r   rZ     s    z'AdminPasswordChangeForm.clean_password2c                 C   s*   | j d }| j| |r$| j  | jS )zSave the new password.rR   r   r   r   r   r   ra     s
    

zAdminPasswordChangeForm.savec                    s*   t  j}| jD ]}||krg   S qdgS )Nr^   )r    changed_datarL   )r&   datar'   r+   r   r   r     s
    

z$AdminPasswordChangeForm.changed_data)T)r-   r.   r/   rd   re   rX   Zrequired_css_classr   rf   rg   r   rh   rR   rS   r5   rZ   ra   propertyr   r0   r   r   r+   r   r     s,    
r   )1r   Zdjangor   Zdjango.contrib.authr   r   r   Zdjango.contrib.auth.hashersr   r   Zdjango.contrib.auth.modelsr   Zdjango.contrib.auth.tokensr	   Zdjango.contrib.sites.shortcutsr
   Zdjango.core.exceptionsr   Zdjango.core.mailr   Zdjango.templater   Zdjango.utils.encodingr   Zdjango.utils.httpr   Zdjango.utils.textr   Zdjango.utils.translationr   r   re   rv   r   ZWidgetr   ZFieldr1   rf   r:   Z	ModelFormr@   ri   ZFormrn   r   r   r   r   r   r   r   r   <module>   s2   		<NN,