U
    d1@                     @   s  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 d dl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-7vj2equ7/django/contrib/auth/forms.py_unicode_ci_compare   s    r   c                       s,   e Zd ZdZdZ fddZ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_contextc                 C   s   d S Nr   )r&   Zid_r   r   r   id_for_label;   s    z'ReadOnlyPasswordHashWidget.id_for_label)__name__
__module____qualname__Ztemplate_nameZ	read_onlyr!   r.   __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   r7   B   s    z"ReadOnlyPasswordHashField.__init__)r/   r0   r1   r   widgetr7   r2   r   r   r+   r   r3   ?   s   r3   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=   I   s    zUsernameField.to_pythonc                    s   t  |dddS )Nnoneusername)Zautocapitalizeautocomplete)r    widget_attrs)r&   r;   r+   r   r   rA   L   s    
zUsernameField.widget_attrs)r/   r0   r1   r=   rA   r2   r   r   r+   r   r<   H   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   stripr;   	help_textzPassword confirmation4Enter the same password as before, for verification.r   r;   rH   rI   c                   @   s   e Zd ZeZdZdeiZdS )zUserCreationForm.Meta)r?   r?   Nr/   r0   r1   r   modelfieldsr<   Zfield_classesr   r   r   r   Metaj   s   rO   c                    s:   t  j|| | jjj| jkr6d| j| jjj jjd< d S )NT	autofocus)r    r7   _metarM   USERNAME_FIELDrN   r;   r(   r8   r+   r   r   r7   o   s    zUserCreationForm.__init__c                 C   s>   | j d}| j d}|r:|r:||kr:t| jd dd|S N	password1	password2rC   code)cleaned_datagetr   error_messagesr&   rT   rU   r   r   r   clean_password2v   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 )NrU   )	r    _post_cleanrX   rY   r   validate_passwordinstancer   Z	add_error)r&   passworderrorr+   r   r   r]      s    
zUserCreationForm._post_cleanTc                    s.   t  jdd}|| jd  |r*|  |S )NF)commitrT   )r    saveset_passwordrX   )r&   rb   userr+   r   r   rc      s
    zUserCreationForm.save)T)r/   r0   r1   __doc___rZ   r   	CharFieldPasswordInputr   "password_validators_help_text_htmlrT   rU   rO   r7   r\   r]   rc   r2   r   r   r+   r   rB   T   s*    
rB   c                       s>   e Zd ZeededdZG dd dZ fddZ  ZS )UserChangeFormrE   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   rI   c                   @   s   e Zd ZeZdZdeiZdS )zUserChangeForm.Meta__all__r?   NrL   r   r   r   r   rO      s   rO   c                    sN   t  j|| | jd}|r,|jd|_| jd}|rJ|jd|_d S )Nr`   z../password/user_permissionscontent_type)r    r7   rN   rY   rI   formatZquerysetZselect_related)r&   r9   r:   r`   rm   r+   r   r   r7      s    zUserChangeForm.__init__)	r/   r0   r1   r3   rg   r`   rO   r7   r2   r   r   r+   r   rk      s   	rk   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.
    rP   TrG   )r;   rE   Fr@   current-passwordr   rH   r;   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    r7   	UserModelrQ   	get_fieldrR   username_field
max_lengthrN   r;   r(   r   r   verbose_name)r&   rv   r9   r:   Zusername_max_lengthr+   r   r   r7      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`   )rX   rY   r   rv   rw   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.
        rt   rV   N)	is_activer   rZ   )r&   re   r   r   r   r~      s
    z(AuthenticationForm.confirm_login_allowedc                 C   s   | j S r-   )rw   r&   r   r   r   get_user   s    zAuthenticationForm.get_userc                 C   s   t | jd dd| jjidS )Nrs   r?   )rW   params)r   rZ   rz   r|   r   r   r   r   r}      s
    
z*AuthenticationForm.get_invalid_login_error)N)r/   r0   r1   rf   r<   r   Z	TextInputr?   rh   rg   ri   r`   rZ   r7   r   r~   r   r}   r2   r   r   r+   r   rp      s"   rp   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Emailru   r@   emailrG   )r   r{   r;   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>)rx   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uidre   tokenprotocol)r   N)rX   r
   r'   r   rx   r   r   r   r   r   pkZ
make_tokenr   )r&   Zdomain_overrider   r   Z	use_httpsZtoken_generatorr   rv   r   Zextra_email_contextr   Zcurrent_siter   r   r   re   Z
user_emailr)   r   r   r   rc   3  s6    



zPasswordResetForm.save)N)r/   r0   r1   r   Z
EmailFieldrg   Z
EmailInputr   r   r   r	   rc   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
    rC   rD   zNew passwordr@   rF   rG   FrK   zNew password confirmationrr   c                    s   || _ t j|| d S r-   re   r    r7   r&   re   r9   r:   r+   r   r   r7   v  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_password2rC   rV   rX   rY   r   rZ   r   r^   re   r[   r   r   r   clean_new_password2z  s    z#SetPasswordForm.clean_new_password2Tc                 C   s*   | j d }| j| |r$| j  | jS )Nr   rX   re   rd   rc   r&   rb   r`   r   r   r   rc     s
    

zSetPasswordForm.save)T)r/   r0   r1   rf   rg   rZ   r   rh   ri   r   rj   r   r   r7   r   rc   r2   r   r   r+   r   r   a  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 passwordFrq   Tr@   rP   rG   rr   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   rV   )rX   re   Zcheck_passwordr   rZ   )r&   r   r   r   r   clean_old_password  s    
z%PasswordChangeForm.clean_old_passwordN)r/   r0   r1   rf   r   rZ   rg   r   rh   ri   r   Zfield_orderr   r   r   r   r   r     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.
    rC   rD   r4   rE   rF   Tr   rG   FrK   zPassword (again)r@   rJ   c                    s   || _ t j|| d S r-   r   r   r+   r   r   r7     s    z AdminPasswordChangeForm.__init__c                 C   sL   | j d}| j d}|r:|r:||kr:t| jd ddt|| j |S rS   r   r[   r   r   r   r\     s    z'AdminPasswordChangeForm.clean_password2c                 C   s*   | j d }| j| |r$| j  | jS )zSave the new password.rT   r   r   r   r   r   rc     s
    

zAdminPasswordChangeForm.savec                    s*   t  j}| jD ]}||krg   S qdgS )Nr`   )r    changed_datarN   )r&   datar'   r+   r   r   r     s
    

z$AdminPasswordChangeForm.changed_data)T)r/   r0   r1   rf   rg   rZ   Zrequired_css_classr   rh   ri   r   rj   rT   rU   r7   r\   rc   propertyr   r2   r   r   r+   r   r     s0    
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   rg   rx   r   ZWidgetr   ZFieldr3   rh   r<   Z	ModelFormrB   rk   ZFormrp   r   r   r   r   r   r   r   r   <module>   s4   	?Qb-#