_V @sddlmZmZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z ddlmZddlmZGd d d eZGd d d eZGd ddeZdS))print_functionunicode_literalsN)apps) NOT_PROVIDED) datetime_safesixtimezone)input)MigrationLoaderc@sseZdZdZdddddZddZddZd d Zd d Zd dZ ddZ dS)MigrationQuestionerz Gives 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.|p i|_|pt|_||_dS)N)defaultssetspecified_appsdry_run)selfr rrrA/tmp/pip-build-ghmbqnp_/Django/django/db/migrations/questioner.py__init__szMigrationQuestioner.__init__cCsM||jkrdSytj|}Wn%tk rM|jjddSYnXtj|j}|dkr|jjddSyt j |}Wn%t k r|jjddSYnXt |drt jt jj|j}nGt |dr1t|jdkrdSt jt|jd}ntd d |D SdS) z2Should we create an initial migration for the app?T ask_initialFN__file____path__r rcss*|] }|dkr|jdVqdS)z __init__.pyz.pyN)endswith).0xrrr 7sz2MigrationQuestioner.ask_initial..)rrZget_app_config LookupErrorr getr migrations_modulelabel importlib import_module ImportErrorhasattroslistdirpathdirnamerlenrlistany)r app_labelZ app_configZmigrations_import_pathr filenamesrrrrs(   !zMigrationQuestioner.ask_initialcCsdS)z"Adding a NOT NULL field to a modelNr)r field_name model_namerrrask_not_null_addition9sz)MigrationQuestioner.ask_not_null_additioncCsdS)z!Changing a NULL field to NOT NULLNr)rr-r.rrrask_not_null_alteration>sz+MigrationQuestioner.ask_not_null_alterationcCs|jjddS)zWas this field really renamed? ask_renameF)r r)rr.old_namenew_namefield_instancerrrr1CszMigrationQuestioner.ask_renamecCs|jjddS)zWas this model really renamed?ask_rename_modelF)r r)rold_model_statenew_model_staterrrr5Gsz$MigrationQuestioner.ask_rename_modelcCs|jjddS)z-Do you really want to merge these migrations? ask_mergeF)r r)rr+rrrr8KszMigrationQuestioner.ask_merge) __name__ __module__ __qualname____doc__rrr/r0r1r5r8rrrrr s      r c@sseZdZdddZddZddZdd Zd d Zd d ZddZ ddZ dS)InteractiveMigrationQuestionerNcCsvtd|}| r'|dk r'|Sx8t|dksR|djdkratd}q*W|djdkS)Nz%s r rZynzPlease answer yes or no: y)r r(lower)rquestiondefaultresultrrr_boolean_inputRs +z-InteractiveMigrationQuestioner._boolean_inputc Cst|x2t|D]$\}}td|d|fqWtd}x[y6t|}d|kozt|knr|SWntk rYnXtd}qNWdS)Nz %s) %sr zSelect an option: rzPlease select a valid option: )print enumerater intr( ValueError)rr@choicesichoicerBvaluerrr _choice_inputZs   " z,InteractiveMigrationQuestioner._choice_inputcCstdtdxtjr/td}ntdjtjj}|s]tdq|dkrytjdqy"t |iit d6t d6SWqt t fk r}ztd |WYdd}~XqXqWdS) Nz3Please enter the default value now, as valid Pythonz_The datetime and django.utils.timezone modules are available, so you can do e.g. timezone.now()z>>> z;Please enter some code, or 'exit' (with no quotes) to exit.exitr datetimerzInvalid input: %s)rDrPY3r decodesysstdinencodingrMevalrr SyntaxError NameError)rcodeerrr _ask_defaulths     "z+InteractiveMigrationQuestioner._ask_defaultcCsX|jsT|jd||fddg}|dkrGtjdqT|jSndS)z"Adding a NOT NULL field to a modelzYou are trying to add a non-nullable field '%s' to %s without a default; we can't do that (the database needs something to populate existing rows). Please select a fix:z@Provide a one-off default now (will be set on all existing rows)z+Quit, and let me add a default in models.pyN)rrLrQrMrY)rr-r.rJrrrr/}s     z4InteractiveMigrationQuestioner.ask_not_null_additioncCsk|jsg|jd||fdddg}|dkr>tS|dkrZtjdqg|jSndS)z!Changing a NULL field to NOT NULLzYou are trying to change the nullable field '%s' on %s to non-nullable without a default; we can't do that (the database needs something to populate existing rows). Please select a fix:z@Provide a one-off default now (will be set on all existing rows)zIgnore for now, and let me handle existing rows with NULL myself (e.g. because you added a RunPython or RunSQL operation to handle NULL values in a previous data migration)z+Quit, and let me add a default in models.pyrZr[N)rrLrrQrMrY)rr-r.rJrrrr0s      z6InteractiveMigrationQuestioner.ask_not_null_alterationcCs/d}|j||||||jjfdS)zWas this field really renamed?z+Did you rename %s.%s to %s.%s (a %s)? [y/N]F)rC __class__r9)rr.r2r3r4msgrrrr1sz)InteractiveMigrationQuestioner.ask_renamecCs,d}|j||j|j|jfdS)zWas this model really renamed?z+Did you rename the %s.%s model to %s? [y/N]F)rCr+name)rr6r7r]rrrr5sz/InteractiveMigrationQuestioner.ask_rename_modelcCs|jddddS)NzH Merging will only work if the operations printed above do not conflict z8with each other (working on different fields or models) z4Do you want to merge these migration branches? [y/N]F)rC)rr+rrrr8sz(InteractiveMigrationQuestioner.ask_merge) r9r:r;rCrLrYr/r0r1r5r8rrrrr=Ps       r=c@s(eZdZddZddZdS)!NonInteractiveMigrationQuestionercCstjddS)Nr[)rQrM)rr-r.rrrr/sz7NonInteractiveMigrationQuestioner.ask_not_null_additioncCstS)N)r)rr-r.rrrr0sz9NonInteractiveMigrationQuestioner.ask_not_null_alterationN)r9r:r;r/r0rrrrr_s  r_) __future__rrr r$rQZ django.appsrZdjango.db.models.fieldsrZ django.utilsrrrZdjango.utils.six.movesr loaderr objectr r=r_rrrrs   Al