U ad2<@sddlZddlZddlZddlZddlmZmZmZmZddl m Z ddl m Z ddl mZddlmZddlmZdd d d d d ddgZGdddeZddZdddZddZdS)N)DictListTupleUnion)MODELS_MODULE_NAME) import_string)CollisionResolvingRunner)SubclassesFinder)RemovedInNextVersionWarningz#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__rrG/tmp/pip-unpacked-wheel-portabvm/django_extensions/management/shells.pyr !sr cCsjtt|d}z:z||tdWWStk rH|dYWSXWntk rd|YSXdS)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%srFc Csi}|D]}t|tr |}z8t|tr|drzt|}WnFtk r}z(|st|d||fWYWqW5d}~XYnXt 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]}|jpX|j} z@|jd krt| D]} t| | || <qpnt| |j|| <Wn8tk r}ztt| t|W5d}~XYnXqHqܐnztjd td d t|tr4t|} | ||dd<|s,t| d|Wqn$t|ttfr@t|d kr@t|dts|s~t|d|dWqt|dttfrVt dd|dDrVt|dii|d} g} |dD]\}zt| |||<Wn8tk r|st|d||d|fYn X| |q|s>t| d|dd!| fnt|dtr|dd krt|dii|d} t| D]} t| | || <q|st| d|dnRtt|dii|dg|d} | ||d<|s>t| d|d|dfn$|sXt|d|d|dfn|sXt|d|Wqtk r|st|d|YqXq|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 [szimport_items..z"Only specify import statements: %rz%scSsg|] }|jqSr)name)r r#rrr isz 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"sz.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#r5Zimported_objectkZimported_namesrrr import_itemsBs          $ , "" "   rKcsjddlmddlm}js$|ddlm dg}t dg}||dk}d t d it d it d i}iifd d  fdd}fdd} fddfdd} d} zddl m } d} Wnt k r YnX|rP s&t dt| d} | D]\} }|| <q<| r|s| D]`\}}|dd}t|j}|ksdd||fkrqd|jg|j|qd|sL| D]x\}}|sqt|j}|krq|D]F}d||jfkrq|jr|jg|j|jqq| stt d|rldnd |t d!drƈ st d"tt d} | D]\} }|| <qt d#i}|r st d$t| d} | D]\} }|| <qt d%i}|rf s||}|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 setdefaultrB)Zmodels_to_importapp_modmodelsapp_nameZ app_aliasesprefix model_namealias) app_prefixesrP 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;r0 SQL_TABLEr Zcollect_subclasses)Zbase_classes_to_import)perform_automatic_importsrErOrDrrimport_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)r`rbrr import_modelssz%import_objects..import_modelsc s|D]\}}g}t|D]\}}z>td||f|<||krN||n|d||fWqtk r}z<drtdst d||t |fW5d}~XYqXqdst 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. rRz %s (as %s) tracebackrEz*Failed to import '%s' from '%s' reason: %sNr*r+) rTrSrrBr=rUre print_excr0r1r(r3rC)Zmodules_to_classesZfull_module_pathrXZ model_labelsr[r\r))rFoptionsrDrrrbs"    z1import_objects..perform_automatic_importsc3s(D]}|jr|j|fVqdSr)Zget_app_configsZ models_moduleZ get_models)ZapprLrrget_apps_and_models s z+import_objects..get_apps_and_modelsF)_document_registryTz# Shell Plus User Pre Imports)rErrRz# 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.appsrMZdjangorNreadyZ django.confrOrUr;Zmongoengine.baserir=r0rarKrTrrr rVrBr rl)rgrDrNZ dont_load_cliZdont_load_confZdont_load_any_modelsrQrcrdrhZ mongoengineriZimportsrJvr#modrYrWZ app_modelsrmrnr) r]rMrPr`rFr^r_rgrbrErOrDrimport_objectss                      rr)F)rrer>r6typingrrrrZdjango.apps.configrZdjango.utils.module_loadingrZ%django_extensions.collision_resolversrZ#django_extensions.import_subclassesr Z#django_extensions.utils.deprecationr rlr/r rrKrrrrrrs,       l