o
    lf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   c/var/www/ticemtrilhas/avaliacao_env/lib/python3.10/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 w t|j\}}|du r2| jddS zt	
|}W n tyI   | jdd Y S w t|ddr[ttj|j}nt|drst|jdkridS tt|jd }td	d
 |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   s     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_configmigrations_import_path_r   	filenamesr   r   r   r      s,   

zMigrationQuestioner.ask_initialc                 C      dS )#Adding a NOT NULL field to a model.Nr   r   
field_name
model_namer   r   r   ask_not_null_addition9      z)MigrationQuestioner.ask_not_null_additionc                 C   r0   )"Changing a NULL field to NOT NULL.Nr   r2   r   r   r   ask_not_null_alteration>   r6   z+MigrationQuestioner.ask_not_null_alterationc                 C      | j ddS )Was this field really renamed?
ask_renameFr   r   )r   r4   old_namenew_namefield_instancer   r   r   r;   C      zMigrationQuestioner.ask_renamec                 C   r9   )Was this model really renamed?ask_rename_modelFr<   )r   old_model_statenew_model_stater   r   r   rB   G   r@   z$MigrationQuestioner.ask_rename_modelc                 C   r9   )z)Should these migrations really be merged?	ask_mergeFr<   r   r+   r   r   r   rE   K   r@   zMigrationQuestioner.ask_mergec                 C   r0   )(Adding an auto_now_add field to a model.Nr   r2   r   r   r   ask_auto_now_add_additionO   r6   z-MigrationQuestioner.ask_auto_now_add_additionc                 C   r0   ).Adding a unique field with a callable default.Nr   r2   r   r   r   $ask_unique_callable_default_additionT   r6   z8MigrationQuestioner.ask_unique_callable_default_addition)NNN)__name__
__module____qualname____doc__r   r   r5   r8   r;   rB   rE   rH   rJ   r   r   r   r   r	      s    
r	   c                       st   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   rT   	__class__r   r   r   [   s   z'InteractiveMigrationQuestioner.__init__c                 C   sz   | j j| ddd t }|s|d ur|S |r |d  dvr5| j jddd t }|r |d  dvs |d  dkS )N  endingr   ynzPlease answer yes or no: y)rT   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y7   Y nw d|  k rEt|krF|S  | j jddd t }q()	Nz %s) %sr   zSelect an option: rX   rY   Tr   zPlease select a valid option: )rT   r]   	enumerater^   int
ValueErrorr(   )r   r`   choicesichoicerb   valuer   r   r   _choice_inputm   s"   z,InteractiveMigrationQuestioner._choice_inputrX   c              
   C   s   | j d |r| j d| d | j d | j d 	 |r'd|}nd}| j j|d	d
 t }|s:|r:|}|sC| j d n1|dkrMtd n'z
t|i ttdW S  t	t
fys } z| j d|  W Y d}~nd}~ww q)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 promptTz[default: {}] >>> z>>> rX   rY   z;Please enter some code, or 'exit' (without quotes) to exit.exitr   )datetimer   zInvalid input: %sN)rT   r]   formatr^   rR   rl   evalrm   r   SyntaxError	NameError)r   ra   promptcodeer   r   r   _ask_default~   s<   
z+InteractiveMigrationQuestioner._ask_defaultc                 C   sF   | j s!| d| d| dddg}|dkrtd dS |  S dS )	r1   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   rk   rR   rl   ru   r   r3   r4   ri   r   r   r   r5      s   
z4InteractiveMigrationQuestioner.ask_not_null_additionc                 C   sR   | j s'| d| d| dg d}|dkrtS |dkr#td dS |  S dS )r7   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:)rv   zIgnore for now. Existing rows that contain NULL values will have to be handled manually, for example with a RunPython or RunSQL operation.rw   rx   ry   N)r   rk   r   rR   rl   ru   r{   r   r   r   r8      s   
z6InteractiveMigrationQuestioner.ask_not_null_alterationc                 C   s"   d}|  ||||||jjf dS )r:   z(Was %s.%s renamed to %s.%s (a %s)? [y/N]F)rc   rV   rK   )r   r4   r=   r>   r?   msgr   r   r   r;      s   z)InteractiveMigrationQuestioner.ask_renamec                 C   s    d}|  ||j|j|jf dS )rA   z(Was the model %s.%s renamed to %s? [y/N]F)rc   r+   name)r   rC   rD   r|   r   r   r   rB      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)rc   rF   r   r   r   rE      s   z(InteractiveMigrationQuestioner.ask_mergec                 C   sJ   | j s#| d| d| dddg}|dkrtd d
S | jdd	S d
S )rG   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 rowsrw   rx   ry   ztimezone.now)ra   Nrz   r{   r   r   r   rH      s   
z8InteractiveMigrationQuestioner.ask_auto_now_add_additionc                 C   sH   | j s"t }| d| d| dd| ddg}|dkr"td d	S )
rI   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.rx   ry   N)r   r   rk   rR   rl   )r   r3   r4   versionri   r   r   r   rJ     s   
zCInteractiveMigrationQuestioner.ask_unique_callable_default_addition)NNNNr
   )rX   )rK   rL   rM   r   rc   rk   ru   r5   r8   r;   rB   rE   rH   rJ   __classcell__r   r   rU   r   rO   Z   s    


(	rO   c                       sH   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 rP   )	verbositylogrQ   r   )r   r   r   r   r   r   rU   r   r   r   $  s   
z*NonInteractiveMigrationQuestioner.__init__c              	   C   s0   | j dkr| d| d| d| d d S d S )Nr   Field '' on model 'z' not migrated: r~   )r   r   )r   r3   r4   reasonr   r   r   log_lack_of_migration4  s   
z7NonInteractiveMigrationQuestioner.log_lack_of_migrationc                 C      |  ||d td d S )NzIit is impossible to add a non-nullable field without specifying a defaultry   r   rR   rl   r2   r   r   r   r5   ;     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   r2   r   r   r   r8   E  s   z9NonInteractiveMigrationQuestioner.ask_not_null_alterationc                 C   r   )NzUit is impossible to add a field with 'auto_now_add=True' without specifying a defaultry   r   r2   r   r   r   rH   M  r   z;NonInteractiveMigrationQuestioner.ask_auto_now_add_addition)NNNr   N)	rK   rL   rM   r   r   r5   r8   rH   r   r   r   rU   r   r   #  s    
r   )rm   r   r#   rR   django.appsr   django.core.management.baser   django.db.modelsr   django.utilsr   django.utils.versionr   loaderr   r	   rO   r   r   r   r   r   <module>   s    K J