U
    5øiI.  ã                   @   sœ   d Z ddlZddlZddl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 ddlmZ dd	lmZ G d
d„ deƒZdZG dd„ deƒZdS )z*
Management utility to create superusers.
é    N)Úget_user_model)Úget_default_username)Úvalidate_password)Ú
exceptions)ÚBaseCommandÚCommandError)ÚDEFAULT_DB_ALIAS)Úcapfirstc                   @   s   e Zd ZdS )ÚNotRunningInTTYExceptionN)Ú__name__Ú
__module__Ú__qualname__© r   r   ú[/tmp/pip-unpacked-wheel-gcyiijx6/django/contrib/auth/management/commands/createsuperuser.pyr
      s   r
   Úpasswordc                       s`   e Zd ZdZdZdZ‡ fdd„Zdd„ Z‡ fdd	„Zd
d„ Z	ddd„Z
ddd„Zdd„ Z‡  ZS )ÚCommandzUsed to create a superuser.T)Ústdinc                    s.   t ƒ j||Ž tƒ | _| jj | jj¡| _d S )N)ÚsuperÚ__init__r   Ú	UserModelÚ_metaÚ	get_fieldÚUSERNAME_FIELDÚusername_field)ÚselfÚargsÚkwargs©Ú	__class__r   r   r      s    zCommand.__init__c                 C   s¼   |j d| jj dd� |j ddddd| jj d	� |j d
tdd� | jjD ]l}| jj |¡}|jr |jj	r„|jj	jj
s„td| ƒ‚q¶|j d| dd| fd� qJ|j d| d| d� qJd S )Nz--%sz&Specifies the login for the superuser.)Úhelpz	--noinputz
--no-inputÚstore_falseÚinteractivezòTells Django to NOT prompt the user for input of any kind. You must use --%s with --noinput, along with an option for any other required field. Superusers created with --noinput will not be able to log in until they're given a valid password.)ÚactionÚdestr   z
--databasez4Specifies the database to use. Default is "default".)Údefaultr   z\Required field '%s' specifies a many-to-many relation through model, which is not supported.Úappendz?Specifies the %s for the superuser. Can be used multiple times.)r"   r   z#Specifies the %s for the superuser.)Úadd_argumentr   r   r   ÚREQUIRED_FIELDSr   r   Úmany_to_manyÚremote_fieldZthroughZauto_createdr   )r   ÚparserÚ
field_nameÚfieldr   r   r   Úadd_arguments"   sN    
þ   üý
ýþÿ ÿÿþþzCommand.add_argumentsc                    s   |  dtj¡| _tƒ j||ŽS )Nr   )ÚgetÚsysr   r   Úexecute)r   r   Úoptionsr   r   r   r0   M   s    zCommand.executec              
   O   sú  || j j }|d }i }| jj}z| j j t¡ W n tjk
rH   Y n
X d |t< �z |d �r"i }t	| j
dƒr~| j
 ¡ s~t‚t|d�}|r°|  |||¡}	|	rÈ| j |	¡ d }n|dkrÈtdt|ƒ ƒ‚|d k�r|  | j|¡}
|  | j|
|¡}|rÈ|  |||¡}	|	rÈ| j |	¡ d }qÈqÈ||| j j< | jj�r>| jj |¡n||| j j< | j jD ]ê}| j j |¡}|| ||< || d k	�r’| || d ¡||< || d k�r
|  |¡}
|  ||
¡}|||< |j�r’|�r’| ¡ �sîd ||< | j d¡ �q’dd	„ | d
¡D ƒ||< �q’|j�s|| ||< |j�rR|j || ¡||< �qRt|k�rB|t d k�rBt ¡ }t d¡}||k�r‚| j d¡ �q>| ¡ dk�r | j d¡ �q>zt|| j f |Žƒ W nZ tjk
�r } z8| j d  |j!¡¡ t"dƒ}| #¡ dk�r W Y ¢�q>W 5 d }~X Y nX ||t< �q>�n t|k�rFdt$j%k�rFt$j%d |t< |d k�rht$j% &d| j j '¡  ¡}|d k�r„td| j j ƒ‚n|  |||¡}	|	�r t|	ƒ‚||| j j< | j jD ]Œ}d| '¡  }|| �pØt$j% &|¡}|�sìtd| ƒ‚| j j |¡}| |d ¡||< |j�r´t(|| t)ƒ�r´dd	„ ||  d
¡D ƒ||< �q´| j j* +|¡j,f |Ž |d dk�rr| j- d¡ W n€ t.k
�r    | j d¡ t/ 0d¡ Y nV tjk
�rÔ } ztd  |j!¡ƒ‚W 5 d }~X Y n" tk
�rô   | j- d¡ Y nX d S )NÚdatabaser!   Úisatty)r2   Ú ú%s cannot be blank.z"Error: This field cannot be blank.c                 S   s   g | ]}|  ¡ ‘qS r   ©Ústrip©Ú.0Úpkr   r   r   Ú
<listcomp>Š   s     z"Command.handle.<locals>.<listcomp>ú,zPassword (again): z#Error: Your passwords didn't match.z&Error: Blank passwords aren't allowed.Ú
z:Bypass password validation and create user anyway? [y/N]: ÚyZDJANGO_SUPERUSER_PASSWORDZDJANGO_SUPERUSER_z!You must use --%s with --noinput.c                 S   s   g | ]}|  ¡ ‘qS r   r6   r8   r   r   r   r;   ¿   s    Ú	verbosityé   zSuperuser created successfully.z
Operation cancelled.ú; z‡Superuser creation skipped due to not running in a TTY. You can run `manage.py createsuperuser` in your project to create one manually.)1r   r   r   Úverbose_namer   r   ÚPASSWORD_FIELDr   ZFieldDoesNotExistÚhasattrr   r3   r
   r   Ú_validate_usernameÚstderrÚwriter   r	   Ú_get_input_messageÚget_input_datar)   Úmodelr'   Úcleanr(   r7   ÚsplitZmany_to_oneÚgetpassr   ÚValidationErrorÚjoinÚmessagesÚinputÚlowerÚosÚenvironr.   ÚupperÚ
isinstanceÚstrÚ_default_managerÚ
db_managerZcreate_superuserÚstdoutÚKeyboardInterruptr/   Úexit)r   r   r1   Úusernamer2   Ú	user_dataÚverbose_field_nameZfake_user_dataZdefault_usernameÚ	error_msgÚmessager+   r,   Zinput_valuer   Z	password2ÚerrÚresponseÚenv_varÚvalueÚer   r   r   ÚhandleQ   sÒ    


ÿþ






ÿ ÿzCommand.handleNc              
   C   sn   t |ƒ}|r|dkr|}z| |d¡}W n@ tjk
rh } z | j dd |j¡ ¡ d}W 5 d}~X Y nX |S )zm
        Override this method if you want to customize data inputs or
        validation exceptions.
        r4   Nz	Error: %srA   )rQ   rK   r   rN   rF   rG   rO   rP   )r   r,   ra   r$   Ú	raw_valueÚvalrf   r   r   r   rI   Ò   s    zCommand.get_input_datac                 C   sL   dt |jƒ|rd| nd|jrDd|jjjj|jr8| ¡ n|jjf ndf S )Nz%s%s%s: z (leave blank to use '%s')r4   z (%s.%s))	r	   rB   r)   rJ   r   Zobject_namer(   Zm2m_target_field_namer+   )r   r,   r$   r   r   r   rH   â   s    ý
þúzCommand._get_input_messagec              
   C   sž   | j jrBz| jj |¡ |¡ W n | jjk
r8   Y n
X d| S |sRdt|ƒ S z| j  |d¡ W n4 t	j
k
r˜ } zd |j¡ W Y ¢S d}~X Y nX dS )z=Validate username. If invalid, return a string error message.z Error: That %s is already taken.r5   NrA   )r   Úuniquer   rX   rY   Zget_by_natural_keyZDoesNotExistr	   rK   r   rN   rO   rP   )r   r]   r_   r2   rf   r   r   r   rE   ì   s    zCommand._validate_username)N)N)r   r   r   r   Zrequires_migrations_checksZstealth_optionsr   r-   r0   rg   rI   rH   rE   Ú__classcell__r   r   r   r   r      s   + 


r   )Ú__doc__rM   rS   r/   Zdjango.contrib.authr   Zdjango.contrib.auth.managementr   Z'django.contrib.auth.password_validationr   Zdjango.corer   Zdjango.core.management.baser   r   Z	django.dbr   Zdjango.utils.textr	   Ú	Exceptionr
   rC   r   r   r   r   r   Ú<module>   s   