a 8Xh2<@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__rrh/var/www/viveiro_mudafortebrasil/venv/lib/python3.9/site-packages/django_extensions/management/shells.pyr !sr cCsftt|d}z8z||tdWWStyF|dYWS0Wnty`|YS0dS)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' .N)listreversedsplitindexr ValueError IndexError)mod_nameZrpartsrrr get_app_name%s  rFc Csi}|D]}t|tr |}z@t|tr|drzt|}WnLty}z4|st|d||fWYd}~WqWYd}~n d}~00t dd|j Ds|st|d|Wq|st| d||j D]}t|tj r |j D]"}|jp |j} t|j|| <qt|tjrt|jiidd |j D} |j D]}|jp`|j} z@|jd krt| D]} t| | || <qxnt| |j|| <Wn:ty}z tt| t|WYd}~n d}~00qPqnxtjd td d t|tr>t|} | ||dd<|s6t| d|Wqn"t|ttfrHt|d krHt|dts|st|d|dWqt|dttfr^t dd|dDr^t|dii|d} g} |dD]Z}zt| |||<Wn6ty&|s"t|d||d|fYn 0| |q|sFt| d|dd!| fnt|dtr"|dd krt|dii|d} t| D]} t| | || <q|s t| d|dnRtt|dii|dg|d} | ||d<|sFt| d|d|dfn$|s`t|d|d|dfn|s`t|d|Wqty|st|d|Yq0q|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 [zimport_items..z"Only specify import statements: %rz%scSsg|] }|jqSr)name)r r$rrr ir#z import_items..*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"r#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 %r)"rr)strip startswithrparse ExceptionprintERRORallr! SQL_COLTYPErnamesasnamer$ importlib import_moduler __import__moduledirgetattrAttributeError ImportErrorwarningswarnr rrtuplelenappendjoin) Zimport_directivesstyle quiet_loadimported_objects directivenodeexcr!r$r6Zimported_objectkZimported_namesrrr import_itemsBs   $       ( , "" "   rLcshddlmddlm}js$|ddlm dg}t dg}||dv}d t d it d it d i}iifd d  fdd}fdd} fddfdd} d} zddl m } d} Wnt yYn0|rN s$t dt| d} | D]\} }|| <q:| r|s| D]`\}}|dd}t|j}|vsbd||fvrqb|jg|j|qb|sJ| D]x\}}|sqt|j}|vrq|D]F}d||jfvrq|jr|jg|j|jqq| srt d|rjdnd |t d!drĈ st d"tt d} | D]\} }|| <qt d#i}|r st d$t| d} | D]\} }|| <qt d%i}|rd s:t d&t| d} | D]\} }|| <qPS)'Nrapps)setup)settings dont_loadZSHELL_PLUS_DONT_LOADr&rFZSHELL_PLUS_MODEL_ALIASESZSHELL_PLUS_APP_PREFIXESSHELL_PLUS_PRE_IMPORTScsi}tD]\}}t|}|i}|}t|D]\}d||fvrTq>||}|sx|rtd||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)sorteditemsrget setdefaultrC)Zmodels_to_importapp_modmodelsapp_nameZ app_aliasesprefix model_namealias) app_prefixesrQ load_models model_aliasesrr&get_dict_from_names_to_possible_modelss      z>import_objects..get_dict_from_names_to_possible_modelscs6tdg}|r2s"tdt|dS)NZSHELL_PLUS_SUBCLASSES_IMPORTz# Shell Plus Subclasses Imports)r<r1 SQL_TABLEr Zcollect_subclasses)Zbase_classes_to_import)perform_automatic_importsrFrPrErrimport_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)rZrun_collision_resolver)Zmodules_to_models)rarcrr import_modelssz%import_objects..import_modelsc s|D]\}}g}t|D]\}}z>td||f|<||krN||n|d||fWqty}zDdrtdst d||t |fWYd}~qd}~00qdst 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. rSz %s (as %s) tracebackrFz*Failed to import '%s' from '%s' reason: %sNr+r,) rUrTrrCr>rVrf print_excr1r2r)r4rD)Zmodules_to_classesZfull_module_pathrYZ model_labelsr\r]r*)rGoptionsrErrrcs"    z1import_objects..perform_automatic_importsc3s(D]}|jr|j|fVqdSr)Zget_app_configsZ models_moduleZ get_models)ZapprMrrget_apps_and_models s z+import_objects..get_apps_and_modelsF)_document_registryTz# Shell Plus User Pre Imports)rFrrSz# 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)Z django.appsrNZdjangorOreadyZ django.confrPrVr<Zmongoengine.baserjr>r1rbrLrUrrr rWrCr rm)rhrErOZ dont_load_cliZdont_load_confZdont_load_any_modelsrRrdreriZ mongoenginerjZimportsrKvr$modrZrXZ app_modelsrnror) r^rNrQrarGr_r`rhrcrFrPrErimport_objectss                      rs)F)rrfr?r7typingrrrrZdjango.apps.configrZdjango.utils.module_loadingrZ%django_extensions.collision_resolversrZ#django_extensions.import_subclassesr Z#django_extensions.utils.deprecationr rmr0r rrLrsrrrrs       l