U e4@sddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZGd d d ZGd d d eZGd ddeZdS)N)apps) OutputWrapper) NOT_PROVIDED)timezone)get_docs_version)MigrationLoaderc@sZeZdZdZdddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ dS)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. NcCs |pi|_|pt|_||_dS)N)defaultssetspecified_appsdry_run)selfr r r rC/tmp/pip-unpacked-wheel-lctamlir/django/db/migrations/questioner.py__init__s  zMigrationQuestioner.__init__cCs||jkrdSzt|}Wn"tk r>|jddYSXt|j\}}|dkrf|jddSzt |}Wn"t k r|jddYSXt |ddrt t j|j}n0t|drt|jdkrdSt t|jd}td d |D SdS) z2Should we create an initial migration for the app?T ask_initialFN__file____path__rrcss |]}|dkr|dVqdS)z __init__.pyz.pyN)endswith).0xrrr 7sz2MigrationQuestioner.ask_initial..)r rZget_app_config LookupErrorr getrmigrations_modulelabel importlib import_module ImportErrorgetattroslistdirpathdirnamerhasattrlenrlistany)r app_labelZ app_configZmigrations_import_path_r filenamesrrrrs(   zMigrationQuestioner.ask_initialcCsdS)#Adding a NOT NULL field to a model.Nrr field_name model_namerrrask_not_null_addition9sz)MigrationQuestioner.ask_not_null_additioncCsdS)"Changing a NULL field to NOT NULL.Nrr-rrrask_not_null_alteration>sz+MigrationQuestioner.ask_not_null_alterationcCs|jddS)Was this field really renamed? ask_renameFr r)rr/old_namenew_namefield_instancerrrr4CszMigrationQuestioner.ask_renamecCs|jddS)Was this model really renamed?ask_rename_modelFr5)rold_model_statenew_model_staterrrr:Gsz$MigrationQuestioner.ask_rename_modelcCs|jddS)z)Should these migrations really be merged? ask_mergeFr5rr)rrrr=KszMigrationQuestioner.ask_mergecCsdS)(Adding an auto_now_add field to a model.Nrr-rrrask_auto_now_add_additionOsz-MigrationQuestioner.ask_auto_now_add_additioncCsdS).Adding a unique field with a callable default.Nrr-rrr$ask_unique_callable_default_additionTsz8MigrationQuestioner.ask_unique_callable_default_addition)NNN) __name__ __module__ __qualname____doc__rrr0r2r4r:r=r@rBrrrrr s r csreZdZdfdd ZdddZddZdd d Zd d Zd dZddZ ddZ ddZ ddZ ddZ ZS)InteractiveMigrationQuestionerNcs&tj|||d|pttj|_dSN)r r r )superrrsysstdout prompt_output)rr r r rL __class__rrr[s z'InteractiveMigrationQuestioner.__init__cCsh|jj|dddt}|s,|dk r,|S|r@|ddkrX|jjdddt}q,|ddkS)N ZendingrZynzPlease answer yes or no: y)rLwriteinputlower)rquestiondefaultresultrrr_boolean_inputcs z-InteractiveMigrationQuestioner._boolean_inputcCs|j|t|D] \}}|jd|d|fq|jjdddt}z t|}Wntk rnYn"Xd|krt|krnn|S|jjdddt}qNdS)Nz %s) %srzSelect an option: rPrQrzPlease select a valid option: )rLrS enumeraterTint ValueErrorr&)rrVchoicesichoicerXvaluerrr _choice_inputms z,InteractiveMigrationQuestioner._choice_inputrPc Cs|jd|r$|jd|d|jd|jd|rLd|}nd}|jj|dd t}|sr|rr|}|s|jd q<|d krtd q>> z>>> rPrQz;Please enter some code, or 'exit' (without quotes) to exit.exitr)datetimerzInvalid input: %sN) rLrSformatrTrJrbevalrcr SyntaxError NameError)rrWpromptcodeerrr _ask_default~s4     z+InteractiveMigrationQuestioner._ask_defaultcCsD|js@|d|d|dddg}|dkr8tdn|SdS) 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.Nr rarJrbrkrr.r/r_rrrr0s z4InteractiveMigrationQuestioner.ask_not_null_additioncCsR|jsN|d|d|ddddg}|dkr2tS|dkrFtdn|Sd S) r1z-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:rlzIgnore for now. Existing rows that contain NULL values will have to be handled manually, for example with a RunPython or RunSQL operation.rmrnroN)r rarrJrbrkrqrrrr2s z6InteractiveMigrationQuestioner.ask_not_null_alterationcCs"d}|||||||jjfdS)r3z(Was %s.%s renamed to %s.%s (a %s)? [y/N]F)rYrNrC)rr/r6r7r8msgrrrr4sz)InteractiveMigrationQuestioner.ask_renamecCs d}|||j|j|jfdS)r9z(Was the model %s.%s renamed to %s? [y/N]F)rYr)name)rr;r<rrrrrr:sz/InteractiveMigrationQuestioner.ask_rename_modelcCs |ddS)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)rYr>rrrr=sz(InteractiveMigrationQuestioner.ask_mergecCsH|jsD|d|d|dddg}|dkr8tdn |jdd Sd S) r?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 rowsrmrnroz timezone.now)rWNrprqrrrr@s   z8InteractiveMigrationQuestioner.ask_auto_now_add_additioncCsH|jsDt}|d|d|dd|ddg}|dkrDtdd S) rAz!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.rnroN)r rrarJrb)rr.r/versionr_rrrrB s   zCInteractiveMigrationQuestioner.ask_unique_callable_default_addition)NNNN)N)rP)rCrDrErrYrarkr0r2r4r:r=r@rB __classcell__rrrMrrGZs  ( rGcs>eZdZd fdd ZddZddZd d Zd d ZZS)!NonInteractiveMigrationQuestionerNrcs"||_||_tj|||ddSrH) verbositylogrIr)rr r r rxryrMrrr$sz*NonInteractiveMigrationQuestioner.__init__c Cs,|jdkr(|d|d|d|ddS)NrField ' ' on model 'z' not migrated: rt)rxry)rr.r/reasonrrrlog_lack_of_migration4s z7NonInteractiveMigrationQuestioner.log_lack_of_migrationcCs|||dtddS)NzIit is impossible to add a non-nullable field without specifying a defaultror}rJrbr-rrrr0;s z7NonInteractiveMigrationQuestioner.ask_not_null_additioncCs|d|d|dtS)Nrzr{z8' given a default of NOT PROVIDED and must be corrected.)ryrr-rrrr2Esz9NonInteractiveMigrationQuestioner.ask_not_null_alterationcCs|||dtddS)NzUit is impossible to add a field with 'auto_now_add=True' without specifying a defaultror~r-rrrr@Ms z;NonInteractiveMigrationQuestioner.ask_auto_now_add_addition)NNNrN) rCrDrErr}r0r2r@rvrrrMrrw#s rw)rcrr!rJZ django.appsrZdjango.core.management.baserZdjango.db.modelsrZ django.utilsrZdjango.utils.versionrloaderrr rGrwrrrrs      KJ