o /hF@sddlZddlZddlZddlZddlmZmZmZmZddl m Z ddl m Z ddl mZddlmZddlmZgdZGd d d eZd d ZdddZddZdS)N)DictListTupleUnion)MODELS_MODULE_NAME) import_string)CollisionResolvingRunner)SubclassesFinder)RemovedInNextVersionWarning)z#from django.core.cache import cachez from django.conf import settingsz.from django.contrib.auth import get_user_modelz!from django.db import transactionzRfrom django.db.models import Avg, Case, Count, F, Max, Min, Prefetch, Q, Sum, Whenz!from django.utils import timezonezfrom django.urls import reversez7from django.db.models import Exists, OuterRef, Subqueryc@s eZdZdS)ObjectImportErrorN)__name__ __module__ __qualname__rr^/var/www/html/env_mimamsha/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' .)listreversedsplitindexr ValueError IndexError)mod_namerpartsrrr get_app_name%s  rFc Csi}|D];}t|tr|}zt|tr|drzt|}Wn&tyI}z|s?t|d||fWYd}~WqWYd}~nd}~wwt dd|j Dsa|s_t|d|Wq|slt| d||j D]|}t|tj r|j D]}|jp|j} t|j|| <qzt|tjrt|jiidd |j D} |j D]G}|jp|j} z|jd krt| D] } t| | || <qnt| |j|| <Wqty}ztd |jd |jd t| t|d}~wwqon;tjdtddt|trt|} | ||dd<|st| d|Wqt|ttfrt|dkrt|dts>|st|d|Yqw|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 )zfrom zimport zError parsing: %r %sNcss"|] }t|tjtjfVqdSN) isinstanceastImport ImportFrom).0bodyrrr ]s zimport_items..z"Only specify import statements: %rz%scSsg|]}|jqSr)name)r!r$rrr tsz import_items..*zCouldn't find z in z. Only found: zuOld style import definitions are deprecated. You should use the new style which is similar to normal Python imports. ) stacklevelrrz import %sz7Unable to import %r: module name must be of type stringrcss|]}t|tVqdSr)rstr)r!errrr#s  z.Unable to import %r from %r: %r does not existfrom %s import %s, zfrom %s import *z2Unable to import %r from %r: names must be stringsz*Unable to import %r: names must be stringszUnable to import %r)"rr)strip startswithrparse ExceptionprintERRORallr" SQL_COLTYPErnamesasnamer$ importlib import_moduler __import__moduledirgetattrAttributeError ImportErrorwarningswarnr rrtuplelenappendjoin) import_directivesstyle quiet_loadimported_objects directivenodeexcr"r$r6imported_objectkimported_namesrrr import_itemsBsF             !        rOcsDddlmddlm}js|ddlm dg}t dg}||dv}d t d it d it d i}iifd d  fdd}fdd} fddfdd} d} z ddl m } d} Wn t yYnw|r st dt| d} | D]\} }|| <q| r|s| 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!drP s;t d"tt d} | D] \} }|| <qFt d#i}|rx sct d$t| d} | D] \} }|| <qnt d%i}|r st d&t| d} | D] \} }|| <qS)'Nrapps)setup)settings dont_loadSHELL_PLUS_DONT_LOADr&rGSHELL_PLUS_MODEL_ALIASESSHELL_PLUS_APP_PREFIXESSHELL_PLUS_PRE_IMPORTScsi}tD]F\}}t|}|i}|}t|D].}d||fvr*q||}|s<|r:d||f}n|}||g||d||fqq|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)sorteditemsrget setdefaultrC)models_to_importapp_modmodelsapp_name app_aliasesprefix model_namealias) app_prefixesrT load_models model_aliasesrr&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)r<r1 SQL_TABLEr collect_subclasses)base_classes_to_import)perform_automatic_importsrGrSrFrrimport_subclasses%s  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)rrun_collision_resolver)modules_to_models)rirnrr import_models.s 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. rYz %s (as %s) tracebackrGz*Failed to import '%s' from '%s' reason: %sNr+r,) r[rZrrCr>r\rs print_excr1r2r)r4rD)modules_to_classesfull_module_pathr` model_labelsrdrer*)rHoptionsrFrrrn9sD       z1import_objects..perform_automatic_importsc3s*D] }|jr|j|fVqdSr)get_app_configs models_module get_models)apprPrrget_apps_and_models]s  z+import_objects..get_apps_and_modelsF)_document_registryTz# Shell Plus User Pre Imports)rGrrYz# Shell Plus Model Imports%sz SKIPPEDSHELL_PLUS_DJANGO_IMPORTSz# Shell Plus Django ImportsSHELL_PLUS_IMPORTSz# Shell Plus User ImportsSHELL_PLUS_POST_IMPORTSz# Shell Plus User Post Imports) django.appsrQdjangorRready django.confrSr\r<mongoengine.baser~r>r1rkrOr[rrr r]rCr r)rxrFrR dont_load_clidont_load_confdont_load_any_modelsrXrorrr} mongoenginer~importsrMvr$modrar_ app_modelsrrr) rfrQrTrirHrgrhrxrnrGrSrFrimport_objectss           $            r)F)rrsr?r7typingrrrrdjango.apps.configrdjango.utils.module_loadingr%django_extensions.collision_resolversr#django_extensions.import_subclassesr #django_extensions.utils.deprecationr rr0r rrOrrrrrs         1