a
    XDvg4  ã                   @   sš   d dl 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 ddlmZ G d	d
„ d
ƒZG dd„ deƒZG dd„ deƒZdS )é    N)Úapps)ÚOutputWrapper)ÚNOT_PROVIDED)Útimezone)Úget_docs_versioné   )ÚMigrationLoaderc                   @   sZ   e Zd ZdZddd„Zdd„ Zdd„ Zd	d
„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ ZdS )ÚMigrationQuestionerzË
    Give the autodetector responses to questions it might have.
    This base class has a built-in noninteractive mode, but the
    interactive subclass is what the command-line arguments will use.
    Nc                 C   s    |pi | _ |ptƒ | _|| _d S )N)ÚdefaultsÚsetÚspecified_appsÚdry_run)Úselfr
   r   r   © r   úX/var/www/sistema_ama/venv/lib/python3.9/site-packages/django/db/migrations/questioner.pyÚ__init__   s    
zMigrationQuestioner.__init__c                 C   sþ   || j v rdS zt |¡}W n  ty<   | j dd¡ Y S 0 t |j¡\}}|du rd| j dd¡S zt	 
|¡}W n  ty’   | j dd¡ Y S 0 t|ddƒr¶t tj |j¡¡}n0t|dƒræt|jƒdkrÒdS t t|jƒd ¡}td	d
„ |D ƒƒ S dS )z2Should we create an initial migration for the app?TÚask_initialFNÚ__file__Ú__path__r   r   c                 s   s    | ]}|d kr|  d¡V  qdS )z__init__.pyz.pyN)Úendswith)Ú.0Úxr   r   r   Ú	<genexpr>7   ó    z2MigrationQuestioner.ask_initial.<locals>.<genexpr>)r   r   Úget_app_configÚLookupErrorr
   Úgetr   Úmigrations_moduleÚlabelÚ	importlibÚimport_moduleÚImportErrorÚgetattrÚosÚlistdirÚpathÚdirnamer   ÚhasattrÚlenr   ÚlistÚany)r   Ú	app_labelÚ
app_configZmigrations_import_pathÚ_r   Ú	filenamesr   r   r   r      s(    

zMigrationQuestioner.ask_initialc                 C   s   dS )ú#Adding a NOT NULL field to a model.Nr   ©r   Ú
field_nameÚ
model_namer   r   r   Úask_not_null_addition9   s    z)MigrationQuestioner.ask_not_null_additionc                 C   s   dS )ú"Changing a NULL field to NOT NULL.Nr   r0   r   r   r   Úask_not_null_alteration>   s    z+MigrationQuestioner.ask_not_null_alterationc                 C   s   | j  dd¡S )úWas this field really renamed?Ú
ask_renameF©r
   r   )r   r2   Úold_nameÚnew_nameÚfield_instancer   r   r   r7   C   s    zMigrationQuestioner.ask_renamec                 C   s   | j  dd¡S )úWas this model really renamed?Úask_rename_modelFr8   )r   Úold_model_stateÚnew_model_stater   r   r   r=   G   s    z$MigrationQuestioner.ask_rename_modelc                 C   s   | j  dd¡S )z)Should these migrations really be merged?Ú	ask_mergeFr8   ©r   r+   r   r   r   r@   K   s    zMigrationQuestioner.ask_mergec                 C   s   dS )ú(Adding an auto_now_add field to a model.Nr   r0   r   r   r   Úask_auto_now_add_additionO   s    z-MigrationQuestioner.ask_auto_now_add_additionc                 C   s   dS )ú.Adding a unique field with a callable default.Nr   r0   r   r   r   Ú$ask_unique_callable_default_additionT   s    z8MigrationQuestioner.ask_unique_callable_default_addition)NNN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r3   r5   r7   r=   r@   rC   rE   r   r   r   r   r	      s   
r	   c                       sr   e Zd Zd‡ fdd„	Zddd„Zdd„ Zdd	d
„Zdd„ Zdd„ Zdd„ Z	dd„ Z
dd„ Zdd„ Zdd„ Z‡  ZS )ÚInteractiveMigrationQuestionerNc                    s&   t ƒ j|||d� |pttjƒ| _d S ©N)r
   r   r   )Úsuperr   r   ÚsysÚstdoutÚprompt_output)r   r
   r   r   rO   ©Ú	__class__r   r   r   [   s    ÿz'InteractiveMigrationQuestioner.__init__c                 C   sh   | j j|› d�dd� tƒ }|s,|d ur,|S |r@|d  ¡ dvrX| j jddd� tƒ }q,|d  ¡ dkS )Nú Ú ©Úendingr   ZynzPlease answer yes or no: Úy)rO   ÚwriteÚinputÚlower)r   ÚquestionÚdefaultÚresultr   r   r   Ú_boolean_inputc   s    z-InteractiveMigrationQuestioner._boolean_inputc                 C   sª   | j  |› ¡ t|ƒD ] \}}| j  d|d |f ¡ q| j jddd� tƒ }zt|ƒ}W n tyl   Y n"0 d|  k r†t|ƒkrŽn n|S | j jddd� tƒ }qNd S )Nz %s) %sr   zSelect an option: rS   rT   r   zPlease select a valid option: )rO   rW   Ú	enumeraterX   ÚintÚ
ValueErrorr(   )r   rZ   ÚchoicesÚiÚchoicer\   Úvaluer   r   r   Ú_choice_inputm   s    z,InteractiveMigrationQuestioner._choice_inputrS   c              
   C   sì   | j  d¡ |r$| j  d|› d�¡ | j  d¡ | j  d¡ |rLd |¡}nd}| j j|dd	� tƒ }|sr|rr|}|s„| j  d
¡ q<|dkr˜t d¡ q<zt|i ttdœƒW S  t	t
fyä } z| j  d| ¡ W Y d}~q<d}~0 0 q<dS )zÿ
        Prompt for a default value.

        The ``default`` argument allows providing a custom default value (as a
        string) which will be shown to the user and used as the return value
        if the user doesn't provide any other input.
        z/Please enter the default value as valid Python.zAccept the default 'z/' by pressing 'Enter' or provide another value.zxThe datetime and django.utils.timezone modules are available, so it is possible to provide e.g. timezone.now as a value.zType 'exit' to exit this promptz[default: {}] >>> z>>> rS   rT   z;Please enter some code, or 'exit' (without quotes) to exit.Úexitr   )Údatetimer   zInvalid input: %sN)rO   rW   ÚformatrX   rM   rf   Úevalrg   r   ÚSyntaxErrorÚ	NameError)r   r[   ÚpromptÚcodeÚer   r   r   Ú_ask_default~   s4    
ÿÿÿz+InteractiveMigrationQuestioner._ask_defaultc                 C   sD   | j s@|  d|› d|› d�ddg¡}|dkr8t d¡ n|  ¡ S dS )	r/   z.It is impossible to add a non-nullable field 'z' to z{ without specifying a default. This is because the database needs something to populate existing rows.
Please select a fix:úbProvide a one-off default now (will be set on all existing rows with a null value for this column)ú6Quit and manually define a default value in models.py.é   é   N©r   re   rM   rf   ro   ©r   r1   r2   rc   r   r   r   r3   ¦   s    ÿûúz4InteractiveMigrationQuestioner.ask_not_null_additionc                 C   sP   | j sL|  d|› d|› d�g d¢¡}|dkr0tS |dkrDt d¡ n|  ¡ S dS )r4   z-It is impossible to change a nullable field 'z' on zŠ to non-nullable without providing a default. This is because the database needs something to populate existing rows.
Please select a fix:)rp   zŠIgnore for now. Existing rows that contain NULL values will have to be handled manually, for example with a RunPython or RunSQL operation.rq   rr   rs   N)r   re   r   rM   rf   ro   ru   r   r   r   r5   ½   s    ÿúz6InteractiveMigrationQuestioner.ask_not_null_alterationc                 C   s"   d}|   ||||||jjf d¡S )r6   z(Was %s.%s renamed to %s.%s (a %s)? [y/N]F)r]   rQ   rF   )r   r2   r9   r:   r;   Úmsgr   r   r   r7   Ù   s    ûÿ÷z)InteractiveMigrationQuestioner.ask_renamec                 C   s    d}|   ||j|j|jf d¡S )r<   z(Was the model %s.%s renamed to %s? [y/N]F)r]   r+   Úname)r   r>   r?   rv   r   r   r   r=   è   s    ÿýz/InteractiveMigrationQuestioner.ask_rename_modelc                 C   s   |   dd¡S )Nz°
Merging will only work if the operations printed above do not conflict
with each other (working on different fields or models)
Should these migration branches be merged? [y/N]F)r]   rA   r   r   r   r@   ñ   s    üz(InteractiveMigrationQuestioner.ask_mergec                 C   sH   | j sD|  d|› d|› d�ddg¡}|dkr8t d¡ n| jdd	�S d
S )rB   z#It is impossible to add the field 'z' with 'auto_now_add=True' to zf without providing a default. This is because the database needs something to populate existing rows.
zDProvide a one-off default now which will be set on all existing rowsrq   rr   rs   ztimezone.now)r[   Nrt   ru   r   r   r   rC   ù   s    ÿýûz8InteractiveMigrationQuestioner.ask_auto_now_add_additionc                 C   sH   | j sDtƒ }|  d|› d|› d�d|› d�dg¡}|dkrDt d¡ d	S )
rD   z!Callable default on unique field Ú.zO will not generate unique values upon migrating.
Please choose how to proceed:
z›Continue making this migration as the first step in writing a manual migration to generate unique values described here: https://docs.djangoproject.com/en/z=/howto/writing-migrations/#migrations-that-add-unique-fields.z)Quit and edit field options in models.py.rr   rs   N)r   r   re   rM   rf   )r   r1   r2   Úversionrc   r   r   r   rE     s    ýúü
zCInteractiveMigrationQuestioner.ask_unique_callable_default_addition)NNNN)N)rS   )rF   rG   rH   r   r]   re   ro   r3   r5   r7   r=   r@   rC   rE   Ú__classcell__r   r   rP   r   rJ   Z   s    ÿ


(	rJ   c                       s>   e Zd Zd‡ fdd„	Zdd„ Zdd„ Zd	d
„ Zdd„ Z‡  ZS )Ú!NonInteractiveMigrationQuestionerNr   c                    s"   || _ || _tƒ j|||d� d S rK   )Ú	verbosityÚlogrL   r   )r   r
   r   r   r|   r}   rP   r   r   r   $  s    ýz*NonInteractiveMigrationQuestioner.__init__c              	   C   s,   | j dkr(|  d|› d|› d|› d�¡ d S )Nr   úField 'ú' on model 'z' not migrated: rx   )r|   r}   )r   r1   r2   Úreasonr   r   r   Úlog_lack_of_migration4  s    
ÿÿz7NonInteractiveMigrationQuestioner.log_lack_of_migrationc                 C   s   |   ||d¡ t d¡ d S )NzIit is impossible to add a non-nullable field without specifying a defaultrs   ©r�   rM   rf   r0   r   r   r   r3   ;  s    ýz7NonInteractiveMigrationQuestioner.ask_not_null_additionc                 C   s   |   d|› d|› d�¡ tS )Nr~   r   z8' given a default of NOT PROVIDED and must be corrected.)r}   r   r0   r   r   r   r5   E  s    ÿz9NonInteractiveMigrationQuestioner.ask_not_null_alterationc                 C   s   |   ||d¡ t d¡ d S )NzUit is impossible to add a field with 'auto_now_add=True' without specifying a defaultrs   r‚   r0   r   r   r   rC   M  s    ýz;NonInteractiveMigrationQuestioner.ask_auto_now_add_addition)NNNr   N)	rF   rG   rH   r   r�   r3   r5   rC   rz   r   r   rP   r   r{   #  s        ú
r{   )rg   r   r#   rM   Údjango.appsr   Údjango.core.management.baser   Údjango.db.modelsr   Údjango.utilsr   Údjango.utils.versionr   Úloaderr   r	   rJ   r{   r   r   r   r   Ú<module>   s   K J