o
    ×ŸŽi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   úiC:\Users\psant\Downloads\Anu Raj Backend\school\venv\Lib\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   Zget_app_configÚLookupErrorr   Úgetr   Úmigrations_moduleÚlabelÚ	importlibÚimport_moduleÚImportErrorÚgetattrÚosÚlistdirÚpathÚdirnamer   ÚhasattrÚlenr   ÚlistÚany)r   Ú	app_labelZ
app_configZ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   r-   )ú"Changing a NULL field to NOT NULL.Nr   r/   r   r   r   Úask_not_null_alteration>   r3   z+MigrationQuestioner.ask_not_null_alterationc                 C   ó   | j  dd¡S )úWas this field really renamed?Ú
ask_renameFN©r   r   )r   r1   Úold_nameÚnew_nameÚfield_instancer   r   r   r8   C   ó   zMigrationQuestioner.ask_renamec                 C   r6   )úWas this model really renamed?Úask_rename_modelFNr9   )r   Úold_model_stateÚnew_model_stater   r   r   r?   G   r=   z$MigrationQuestioner.ask_rename_modelc                 C   r6   )z)Should these migrations really be merged?Ú	ask_mergeFNr9   ©r   r*   r   r   r   rB   K   r=   zMigrationQuestioner.ask_mergec                 C   r-   )ú(Adding an auto_now_add field to a model.Nr   r/   r   r   r   Úask_auto_now_add_additionO   r3   z-MigrationQuestioner.ask_auto_now_add_additionc                 C   r-   )ú.Adding a unique field with a callable default.Nr   r/   r   r   r   Ú$ask_unique_callable_default_additionT   r3   z8MigrationQuestioner.ask_unique_callable_default_addition)NNN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r2   r5   r8   r?   rB   rE   rG   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   rQ   ©Ú	__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   ZynzPlease answer yes or no: Úy)rQ   Ú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: rU   rV   Tr   zPlease select a valid option: )rQ   rY   Ú	enumeraterZ   ÚintÚ
ValueErrorr'   )r   r\   ÚchoicesÚiÚchoicer^   Úvaluer   r   r   Ú_choice_inputm   s"   ÿÿ÷z,InteractiveMigrationQuestioner._choice_inputrU   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>>> rU   rV   z;Please enter some code, or 'exit' (without quotes) to exit.Úexitr   )Údatetimer   zInvalid input: %sN)rQ   rY   ÚformatrZ   rO   rh   Úevalri   r   ÚSyntaxErrorÚ	NameError)r   r]   Ú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 )	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   rg   rO   rh   rq   ©r   r0   r1   re   r   r   r   r2   ¦   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 )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:)rr   zŠIgnore for now. Existing rows that contain NULL values will have to be handled manually, for example with a RunPython or RunSQL operation.rs   rt   ru   N)r   rg   r   rO   rh   rq   rw   r   r   r   r5   ½   s   ÿú
ÿz6InteractiveMigrationQuestioner.ask_not_null_alterationc                 C   s"   d}|   ||||||jjf d¡S )r7   z(Was %s.%s renamed to %s.%s (a %s)? [y/N]FN)r_   rS   rH   )r   r1   r:   r;   r<   Úmsgr   r   r   r8   Ù   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]FN)r_   r*   Úname)r   r@   rA   rx   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_   rC   r   r   r   rB   ñ   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 )rD   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 rowsrs   rt   ru   ztimezone.now)r]   Nrv   rw   r   r   r   rE   ù   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 )
rF   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.rt   ru   N)r   r   rg   rO   rh   )r   r0   r1   Úversionre   r   r   r   rG     s   ýúü
zCInteractiveMigrationQuestioner.ask_unique_callable_default_addition)NNNNr
   )rU   )rH   rI   rJ   r   r_   rg   rq   r2   r5   r8   r?   rB   rE   rG   Ú__classcell__r   r   rR   r   rL   Z   s    ÿ


(	rL   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 rM   )Ú	verbosityÚlogrN   r   )r   r   r   r   r~   r   rR   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: rz   )r~   r   )r   r0   r1   Ú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 defaultru   ©rƒ   rO   rh   r/   r   r   r   r2   ;  ó   ý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   r/   r   r   r   r5   E  s   ÿz9NonInteractiveMigrationQuestioner.ask_not_null_alterationc                 C   r„   )NzUit is impossible to add a field with 'auto_now_add=True' without specifying a defaultru   r…   r/   r   r   r   rE   M  r†   z;NonInteractiveMigrationQuestioner.ask_auto_now_add_addition)NNNr   N)	rH   rI   rJ   r   rƒ   r2   r5   rE   r|   r   r   rR   r   r}   #  s    ú
r}   )ri   r   r"   rO   Zdjango.appsr   Zdjango.core.management.baser   Zdjango.db.modelsr   Zdjango.utilsr   Zdjango.utils.versionr   Úloaderr   r	   rL   r}   r   r   r   r   Ú<module>   s    K J