U
    2›d#C  ã                   @   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)ƒ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/ƒZ0G d&d'„ d'ej,ƒZ1d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   úR/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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 )ÚBaseUserCreationFormzc
    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 confirmationú4Enter 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BaseUserCreationForm.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BaseUserCreationForm.__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$BaseUserCreationForm.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 BaseUserCreationForm._post_cleanTc                    s@   t ƒ jdd�}| | jd ¡ |r<| ¡  t| dƒr<|  ¡  |S )NF)ÚcommitrT   Úsave_m2m)r    ÚsaveÚset_passwordrX   Úhasattrrc   )r&   rb   Úuserr+   r   r   rd   ‹   s    
zBaseUserCreationForm.save)T)r/   r0   r1   Ú__doc__Ú_rZ   r   Ú	CharFieldÚPasswordInputr   Ú"password_validators_help_text_htmlrT   rU   rO   r7   r\   r]   rd   r2   r   r   r+   r   rB   T   s*    ÿüü
rB   c                   @   s   e Zd Zdd„ ZdS )ÚUserCreationFormc                 C   sR   | j  d¡}|rJ| jjjj|d� ¡ rJ|  td| j	 
| jjdg¡iƒ¡ n|S dS )z*Reject usernames that differ only in case.r?   )Zusername__iexactN)rX   rY   rQ   rM   ÚobjectsÚfilterÚexistsZ_update_errorsr   r_   Zunique_error_message)r&   r?   r   r   r   Úclean_username–   s     ÿþ  ÿÿÿÿ
zUserCreationForm.clean_usernameN)r/   r0   r1   rq   r   r   r   r   rm   •   s   rm   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                    sZ   t ƒ j||Ž | j d¡}|r8|j d| jj› d�¡|_| j d¡}|rV|j 	d¡|_d S )Nr`   z../../z
/password/Úuser_permissionsÚcontent_type)
r    r7   rN   rY   rI   Úformatr_   ÚpkZquerysetZselect_related)r&   r9   r:   r`   rt   r+   r   r   r7   ¹   s    ÿÿzUserChangeForm.__init__)	r/   r0   r1   r3   ri   r`   rO   r7   r2   r   r   r+   r   rr   ª   s   ÿþ	rr   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&   r~   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   r~   r   Ú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.
        r|   rV   N)Ú	is_activer   rZ   )r&   rg   r   r   r   r†   ü   s
    þz(AuthenticationForm.confirm_login_allowedc                 C   s   | j S r-   )r   ©r&   r   r   r   Úget_user  s    zAuthenticationForm.get_userc                 C   s   t | jd dd| jjid�S )Nr{   r?   )rW   Úparams)r   rZ   r‚   r„   r‰   r   r   r   r…     s
    
ýz*AuthenticationForm.get_invalid_login_error)N)r/   r0   r1   rh   r<   r   Z	TextInputr?   rj   ri   rk   r`   rZ   r7   r‡   r†   rŠ   r…   r2   r   r   r+   r   rx   Ç   s"   ýÿûrx   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Emailr}   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>E  s   ýz.PasswordResetForm.get_users.<locals>.<genexpr>)r€   Úget_email_field_nameZ_default_managerro   )r&   r�   Zactive_usersr   rœ   r   Ú	get_users7  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Úuidrg   ÚtokenÚprotocol)r•   N)rX   r
   r'   r£   r€   rŸ   r    r™   r   r   rw   Z
make_tokenr˜   )r&   Zdomain_overrider’   r“   Z	use_httpsZtoken_generatorr”   r~   r•   Zextra_email_contextr�   Zcurrent_siter¤   r£   r�   rg   Z
user_emailr)   r   r   r   rd   L  s6    


ùø
úzPasswordResetForm.save)N)r/   r0   r1   r   Z
EmailFieldri   Z
EmailInputr�   r˜   r    r	   rd   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 )ÚSetPasswordFormzZ
    A form that lets a user set their password without entering the old
    password
    rC   rD   zNew passwordr@   rF   rG   FrK   zNew password confirmationrz   c                    s   || _ tƒ j||Ž d S r-   ©rg   r    r7   ©r&   rg   r9   r:   r+   r   r   r7   �  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^   rg   r[   r   r   r   Úclean_new_password2“  s    þz#SetPasswordForm.clean_new_password2Tc                 C   s*   | j d }| j |¡ |r$| j ¡  | jS )Nr«   ©rX   rg   re   rd   ©r&   rb   r`   r   r   r   rd   ž  s
    

zSetPasswordForm.save)T)r/   r0   r1   rh   ri   rZ   r   rj   rk   r   rl   r«   r¬   r7   r®   rd   r2   r   r   r+   r   r¨   z  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 passwordFry   T©r@   rP   rG   rz   Ú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   rg   Zcheck_passwordr   rZ   )r&   r´   r   r   r   Úclean_old_password¼  s    
þz%PasswordChangeForm.clean_old_passwordN)r/   r0   r1   rh   r¨   rZ   ri   r   rj   rk   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   rd   ð  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   rh   ri   rZ   Zrequired_css_classr   rj   rk   r   rl   rT   rU   r7   r\   rd   Úpropertyr·   r2   r   r   r+   r   r¶   É  s0    ÿÿúü
r¶   )2r   Ú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   ri   r€   r   ZWidgetr   ÚFieldr3   rj   r<   Z	ModelFormrB   rm   rr   ZFormrx   rŒ   r¨   r±   r¶   r   r   r   r   Ú<module>   s6   	AQb,#