o uvXj3@sddlZddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddlmZddlmZdgd gd gd ggd d gdZe dkrOeddgineddgie dkrdedgdGdddeZddZdddZddZdS) N)DictListTupleUnion)MODELS_MODULE_NAME) import_string)VERSION)CollisionResolvingRunner)SubclassesFindercachesettingsget_user_model transaction) AvgCaseCountFMaxMinPrefetchQSumWhentimezone)zdjango.core.cachez django.confzdjango.contrib.authz django.dbdjango.db.modelsz django.utils) zdjango.core.urlresolversreversez django.urls)r r)ExistsOuterRefSubqueryc@s eZdZdS)ObjectImportErrorN)__name__ __module__ __qualname__r&r&`/var/www/html/myproject/venv/lib/python3.10/site-packages/django_extensions/management/shells.pyr"*sr"cCs^tt|d}zz ||tdWWSty#|dYWSwty.|YSw)a Retrieve application name from models.py module path >>> get_app_name('testapp.models.foo') 'testapp' 'testapp' instead of 'some.testapp' for compatibility: >>> get_app_name('some.testapp.models.foo') 'testapp' >>> get_app_name('some.models.testapp.models.foo') 'testapp' >>> get_app_name('testapp.foo') 'testapp' >>> get_app_name('some.testapp.foo') 'testapp' .r)listreversedsplitindexr ValueError IndexError)mod_namerpartsr&r&r' get_app_name.s  r1Fc Csi}|D]b}z*t|tjr)t|}|||dd<|s't|d|Wqt|ttfr%t |dkr%t|dtjsO|sMt| d|dWqt|dttfrt dd|dDrt|dii|d}g}|dD]*}z t ||||<Wnt y|st| d ||d|fYqtw||qt|st|d |dd |fnt|dtjr|dd krt|dii|d}t|D] }t ||||<q|st|d |dnIt t|dii|dg|d}|||d<|st|d |d|dfn|s$t| d|d|dfn |s1t| d|Wqtygz|sGt| d|Wntyd|sbt| d|d|dfYnwYqw|S)a Import the items in import_directives and return a list of the imported items Each item in import_directives should be one of the following forms * a tuple like ('module.submodule', ('classname1', 'classname2')), which indicates a 'from module.submodule import classname1, classname2' * a tuple like ('module.submodule', 'classname1'), which indicates a 'from module.submodule import classname1' * a tuple like ('module.submodule', '*'), which indicates a 'from module.submodule import *' * a simple 'module.submodule' which indicates 'import module.submodule'. Returns a dict mapping the names to the imported items r(rz import %sz7Unable to import %r: module name must be of type stringrcss|] }t|tjVqdSN) isinstancesix string_types).0er&r&r' fszimport_items..z.Unable to import %r from %r: %r does not existfrom %s import %s, *zfrom %s import *z9Unable to import %r from %r: names must be of type stringz1Unable to import %r: names must be of type stringzUnable to import %rzUnable to import %r from %r)r4r5r6 __import__r+print SQL_COLTYPEr)tuplelenERRORallgetattrAttributeErrorappendjoindir ImportError TypeError) import_directivesstyle quiet_loadimported_objects directiveimported_objectimported_namesnamekr&r&r' import_itemsKsz  (      " rTcsddlmddlm}js|ddlm dg}t dg}||dv}d t d it d it d i}ifd d  fdd}fdd} fddfdd} d} z ddl m } d} Wn t yYnw|r st dt| d} t| D]\} }|| <qi| r|rt| D],\}}|dd}t|j}|vsd||fvrq|jg|j|q|s| D]4\}}|sqt|j}|vrq|D] }d||jfvrq|jr|jg|j|jqq| s)t d|r%dnd |t d!drU s=t d"tt d} t| D] \} }|| <qKt d#i}|r~ sht dt| d} t| D] \} }|| <qtS)$Nrapps)setup)r dont_loadSHELL_PLUS_DONT_LOADr<rMSHELL_PLUS_MODEL_ALIASESSHELL_PLUS_APP_PREFIXESSHELL_PLUS_PRE_IMPORTScsi}ttD]F\}}t|}|i}|}t|D].}d||fvr+q ||}|s=|r;d||f}n|}||g||d||fq q |S)a Collect dictionary from names to possible models. Model is represented as his full path. Name of model can be alias if SHELL_PLUS_MODEL_ALIASES or SHELL_PLUS_APP_PREFIXES is specified for this model. This dictionary is used by collision resolver. At this phase we can't import any models, because collision resolver can change results. :return: Dict[str, List[str]]. Key is name, value is list of full model's path's. %s.%sz%s_%s)sortedr5 iteritemsr1get setdefaultrF)models_to_importapp_modmodelsapp_name app_aliasesprefix model_namealias) app_prefixesrX load_models model_aliasesr&r'&get_dict_from_names_to_possible_modelss"     z>import_objects..get_dict_from_names_to_possible_modelscs:tdg}|rstdt|dSdS)NSHELL_PLUS_SUBCLASSES_IMPORTz# Shell Plus Subclasses Imports)rDr> SQL_TABLEr collect_subclasses)base_classes_to_import)perform_automatic_importsrMr rLr&r'import_subclassess z)import_objects..import_subclassescst}|dS)z Perform collision resolving and imports all models. When collisions are resolved we can perform imports and print information's, because it is last phase. This function updates imported_objects dictionary. N)r run_collision_resolver)modules_to_models)rmrrr&r' import_modelss z%import_objects..import_modelsc s|D]o\}}g}t|D]Q\}}ztd||f|<||kr'||n |d||fWqty_}z"drBtdsUt d||t |fWYd}~qd}~wwdsst d|d |fqdS) a Import elements from given dictionary. :param modules_to_classes: dictionary from module name to tuple. First element of tuple is model name, second is model alias. If both elements are equal than element is imported without alias. r]z %s (as %s) tracebackrMz*Failed to import '%s' from '%s' reason: %sNr:r;) itemsr^rrFrIr`rw print_excr>rBstrr?rG)modules_to_classesfull_module_pathrd model_labelsrhrir8)rNoptionsrLr&r'rrs,    z1import_objects..perform_automatic_importsc3s*D] }|jr|j|fVqdSr3)get_app_configs models_module get_models)apprUr&r'get_apps_and_modelss  z+import_objects..get_apps_and_modelsF)_document_registryTz# Shell Plus User Imports)rMr(r]z# Shell Plus Model Imports%sz SKIPPEDSHELL_PLUS_DJANGO_IMPORTSz# Shell Plus Django ImportsSHELL_PLUS_POST_IMPORTS) django.appsrVdjangorWready django.confr r`rDmongoengine.baserrIr>rorTr5r_r+r1r$rarFr#rrx)r~rLrW dont_load_clidont_load_confdont_load_any_modelsr\rsrvr mongoenginerimportsrSvrRmodrerc app_modelsrr&) rjrVrXrmrNrkrlr~rrrMr rLr'import_objectss                   r)F)r5rwtypingrrrrdjango.apps.configrdjango.utils.module_loadingrrrDJANGO_VERSION%django_extensions.collision_resolversr #django_extensions.import_subclassesr rupdateextend Exceptionr"r1rTrr&r&r&r's8        H