o [h5@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|_|p t|_||_dSN)defaultssetspecified_appsdry_run)selfr r rr/home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/django/db/migrations/questioner.py__init__s   zMigrationQuestioner.__init__cCs||jvrdSzt|}Wnty|jddYSwt|j\}}|dur2|jddSzt |}Wnt yI|jddYSwt |ddr[t t j|j}nt|drst|jdkridSt t|jd}td d |D S) z2Should we create an initial migration for the app?T ask_initialFN__file____path__rrcss"|] }|dkr|dVqdS)z __init__.pyz.pyN)endswith).0xrrr 7s z2MigrationQuestioner.ask_initial..)r rZget_app_config LookupErrorr getrmigrations_modulelabel importlib import_module ImportErrorgetattroslistdirpathdirnamerhasattrlenrlistany)r app_labelZ app_configZmigrations_import_path_r filenamesrrrrs,     zMigrationQuestioner.ask_initialcCdS)#Adding a NOT NULL field to a model.Nrr field_name model_namerrrask_not_null_addition9z)MigrationQuestioner.ask_not_null_additioncCr-)"Changing a NULL field to NOT NULL.Nrr/rrrask_not_null_alteration>r3z+MigrationQuestioner.ask_not_null_alterationcC|jddS)Was this field really renamed? ask_renameFr r)rr1old_namenew_namefield_instancerrrr8CzMigrationQuestioner.ask_renamecCr6)Was this model really renamed?ask_rename_modelFr9)rold_model_statenew_model_staterrrr?Gr=z$MigrationQuestioner.ask_rename_modelcCr6)z)Should these migrations really be merged? ask_mergeFr9rr*rrrrBKr=zMigrationQuestioner.ask_mergecCr-)(Adding an auto_now_add field to a model.Nrr/rrrask_auto_now_add_additionOr3z-MigrationQuestioner.ask_auto_now_add_additioncCr-).Adding a unique field with a callable default.Nrr/rrr$ask_unique_callable_default_additionTr3z8MigrationQuestioner.ask_unique_callable_default_addition)NNN) __name__ __module__ __qualname____doc__rrr2r5r8r?rBrErGrrrrr s  r csteZdZ dfdd ZdddZddZdd d Zd d Zd dZddZ ddZ ddZ ddZ ddZ ZS)InteractiveMigrationQuestionerNcs&tj|||d|pttj|_dSN)r r r)superrrsysstdout prompt_output)rr r rrQ __class__rrr[sz'InteractiveMigrationQuestioner.__init__cCsz|jj|dddt}|s|dur|S|r |ddvr5|jjdddt}|r |ddvs |ddkS)N endingrZynzPlease answer yes or no: y)rQwriteinputlower)rquestiondefaultresultrrr_boolean_inputcs z-InteractiveMigrationQuestioner._boolean_inputcCs|j|t|D]\}}|jd|d|fq |jjddd z t}t|}Wnty7Yn"tyJ|jdtdYnwd|krXt |krY|S|jjd ddq%) Nz %s) %srzSelect an option: rUrVT Cancelled.rzPlease select a valid option: ) rQrY enumeraterZint ValueErrorKeyboardInterruptrOexitr')rr\choicesichoicer^valuerrr _choice_inputms&    z,InteractiveMigrationQuestioner._choice_inputrUc Cs|jd|r|jd|d|jd|jd |r'd|}nd}|jj|d d zt}WntyJ|jd td Ynw|sQ|rQ|}|sZ|jd n4|dkrdtd n*z t|itt dWSt y}z|j|j j d|WYd}~nd}~wwq)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.zAccept 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.zType 'exit' to exit this promptTz[default: {}] >>> z>>> rUrVr`rz;Please enter some code, or 'exit' (without quotes) to exit.re)datetimerz: N) rQrYformatrZrdrOreevalrkr ExceptionrSrH)rr]promptcodeerrr _ask_defaultsF        &z+InteractiveMigrationQuestioner._ask_defaultcCsF|js!|d|d|dddg}|dkrtddS|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.NrrjrOrerrrr0r1rhrrrr2s z4InteractiveMigrationQuestioner.ask_not_null_additioncCsR|js'|d|d|dgd}|dkrtS|dkr#tddS|SdS)r4z-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:)rszIgnore for now. Existing rows that contain NULL values will have to be handled manually, for example with a RunPython or RunSQL operation.rtrurvN)rrjrrOrerrrxrrrr5s z6InteractiveMigrationQuestioner.ask_not_null_alterationcCs"d}|||||||jjfdS)r7z(Was %s.%s renamed to %s.%s (a %s)? [y/N]F)r_rSrH)rr1r:r;r<msgrrrr8sz)InteractiveMigrationQuestioner.ask_renamecCs d}|||j|j|jfdS)r>z(Was the model %s.%s renamed to %s? [y/N]F)r_r*name)rr@rAryrrrr?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)r_rCrrrrBsz(InteractiveMigrationQuestioner.ask_mergecCsJ|js#|d|d|dddg}|dkrtdd S|jdd Sd S) rDz#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 rowsrtrurvz timezone.now)r]NrwrxrrrrEs  z8InteractiveMigrationQuestioner.ask_auto_now_add_additioncCsH|js"t}|d|d|dd|ddg}|dkr"tdd S) rFz!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.rurvN)rrrjrOre)rr0r1versionrhrrrrGs zCInteractiveMigrationQuestioner.ask_unique_callable_default_addition)NNNNr )rU)rHrIrJrr_rjrrr2r5r8r?rBrErG __classcell__rrrRrrLZs  , rLcsHeZdZ     d fdd ZddZddZd d Zd d ZZS)!NonInteractiveMigrationQuestionerNrcs"||_||_tj|||ddSrM) verbositylogrNr)rr r rrrrRrrr*s z*NonInteractiveMigrationQuestioner.__init__c Cs0|jdkr|d|d|d|ddSdS)NrField ' ' on model 'z' not migrated: r{)rr)rr0r1reasonrrrlog_lack_of_migration:s z7NonInteractiveMigrationQuestioner.log_lack_of_migrationcC|||dtddS)NzIit is impossible to add a non-nullable field without specifying a defaultrvrrOrer/rrrr2A z7NonInteractiveMigrationQuestioner.ask_not_null_additioncCs|d|d|dtS)Nrrz8' given a default of NOT PROVIDED and must be corrected.)rrr/rrrr5Ksz9NonInteractiveMigrationQuestioner.ask_not_null_alterationcCr)NzUit is impossible to add a field with 'auto_now_add=True' without specifying a defaultrvrr/rrrrESrz;NonInteractiveMigrationQuestioner.ask_auto_now_add_addition)NNNrN) rHrIrJrrr2r5rEr}rrrRrr~)s r~)rkrr"rOZ django.appsrZdjango.core.management.baserZdjango.db.modelsrZ django.utilsrZdjango.utils.versionrloaderrr rLr~rrrrs      KP