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rttj|j}n0t|drt|jdkrd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 rt|krn 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rt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   zIgnore 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:
zContinue 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