a Eb'@sddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z GdddZ Gd d d e Z Gd d d e ZdS) N)apps NOT_PROVIDED)timezone)MigrationLoaderc@sReZdZdZdddZddZddZd d Zd d Zd dZ 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|_||_dSN)defaultssetspecified_appsdry_run)selfr r r r_/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/django/db/migrations/questioner.py__init__s  zMigrationQuestioner.__init__cCs||jvrdSzt|}Wn ty<|jddYS0t|j\}}|durd|jddSzt |}Wn t y|jddYS0t |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 5z2MigrationQuestioner.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_addition7sz)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)rr0old_namenew_namefield_instancerrrr5AszMigrationQuestioner.ask_renamecCs|jddS)Was this model really renamed?ask_rename_modelFr6)rold_model_statenew_model_staterrrr;Esz$MigrationQuestioner.ask_rename_modelcCs|jddS)z)Should these migrations really be merged? ask_mergeFr6rr*rrrr>IszMigrationQuestioner.ask_mergecCsdS)(Adding an auto_now_add field to a model.Nrr.rrrask_auto_now_add_additionMsz-MigrationQuestioner.ask_auto_now_add_addition)NNN) __name__ __module__ __qualname____doc__rrr1r3r5r;r>rArrrrr s rc@sXeZdZdddZddZdddZd d Zd d Zd dZddZ ddZ ddZ dS)InteractiveMigrationQuestionerNcCsJtd|}|s|dur|S|r0|ddvr:td}q|ddkS)Nz%s rZynzPlease answer yes or no: y)inputlower)rquestiondefaultresultrrr_boolean_inputTs    z-InteractiveMigrationQuestioner._boolean_inputcCst|t|D]\}}td|d|fqtd}z t|}WntyTYn"0d|krnt|krvnn|Std}q6dS)Nz %s) %srzSelect an option: rzPlease select a valid option: )print enumeraterHint ValueErrorr')rrJchoicesichoicerLvaluerrr _choice_input\s  z,InteractiveMigrationQuestioner._choice_inputc Cstd|rtd|dtdtd|r>> z>>> z;Please enter some code, or 'exit' (without quotes) to exit.exitr)datetimerzInvalid input: %sN) rNformatrHsysrXevalrYr SyntaxError NameError)rrKpromptcodeerrr _ask_defaultks.    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 rVr[rXrbrr/r0rTrrrr1s z4InteractiveMigrationQuestioner.ask_not_null_additioncCsP|jsL|d|d|dgd}|dkr0tS|dkrDtdn|SdS)r2z-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:)rczIgnore for now. Existing rows that contain NULL values will have to be handled manually, for example with a RunPython or RunSQL operation.rdrerfN)r rVrr[rXrbrhrrrr3s z6InteractiveMigrationQuestioner.ask_not_null_alterationcCs"d}|||||||jjfdS)r4z(Was %s.%s renamed to %s.%s (a %s)? [y/N]F)rM __class__rB)rr0r7r8r9msgrrrr5sz)InteractiveMigrationQuestioner.ask_renamecCs d}|||j|j|jfdS)r:z(Was the model %s.%s renamed to %s? [y/N]F)rMr*name)rr<r=rjrrrr;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)rMr?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 rowsrdrerfz timezone.now)rKNrgrhrrrrAs   z8InteractiveMigrationQuestioner.ask_auto_now_add_addition)N)rW) rBrCrDrMrVrbr1r3r5r;r>rArrrrrFSs  % rFc@s$eZdZddZddZddZdS)!NonInteractiveMigrationQuestionercCstddSNrfr[rXr.rrrr1sz7NonInteractiveMigrationQuestioner.ask_not_null_additioncCstSr rr.rrrr3sz9NonInteractiveMigrationQuestioner.ask_not_null_alterationcCstddSrmrnr.rrrrAsz;NonInteractiveMigrationQuestioner.ask_auto_now_add_additionN)rBrCrDr1r3rArrrrrlsrl)rYrr"r[Z django.appsrZdjango.db.modelsrZ django.utilsrloaderrrrFrlrrrrs    F&