3 g2<@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__rra/var/www/tester-filtro-web/env/lib/python3.6/site-packages/django_extensions/management/shells.pyr !sr cCs\tt|jd}y0y||jtdStk r>|dSXWntk rV|SXdS)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}x|D]}t|tr$|j}yFt|to:|jdrytj|}Wn>tk r}z"|szt|jd||fw WYdd}~XnXt dd|j Ds|st|jd|w |st|j d|x|j D] }t|tj rx.|j D]$}|jp|j} tj|j|| <qWt|tjrtj|jiid d |j D} x|j D]}|jp^|j} yD|jd krx2t| D]} t| | || <qxWnt| |j|| <Wn8tk r}ztt| t|WYdd}~XnXqNWqWn~tjd td dt|tr@t|} | ||jdd<|s:t|j d|w n*t|ttfrRt|d krRt|dts|st|jd|dw t|dttfrdt dd|dDrdt|dii|d} g} xh|dD]\}yt| |||<Wn8tk r*|s&t|jd||d|fYn X| j |qW|sPt|j d|ddj!| fnt|dtr,|dd krt|dii|d} x t| D]} t| | || <qW|s*t|j d|dnRtt|dii|dg|d} | ||d<|sPt|j d|d|dfn$|sjt|jd|d|dfn|sjt|jd|Wq tk r|st|jd|Yq Xq W|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 from import zError parsing: %r %sNcss |]}t|tjtjfVqdS)N) 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|tVqdS)N)rstr)r!errrr#sz.Unable to import %r from %r: %r does not existzfrom %s import %sz, 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)"rr)strip startswithrparse ExceptionprintERRORallr" SQL_COLTYPErnamesasnamer$ importlib import_moduler __import__moduledirgetattrAttributeError ImportErrorwarningswarnr rrtuplelenappendjoin) Zimport_directivesstyle quiet_loadimported_objects directivenodeexcr"r$r4Zimported_objectkZimported_namesrrr import_itemsBs       ( ,"""   rJc sddlmddlm}js$|ddlm jdg}t dg}||dk}jd t d it d it d i}iifd d  fdd}fdd} fddfdd} d} yddl m } d} Wnt k r YnX|rT s&t j dt| d} x| jD]\} }|| <q>W| r| rxl| jD]`\}}|jdd'}t|j}|ksld||fkrqlj|jg|jj|qlW|s^x| D]|\}}|sqt|j}|krqxN|D]F}d||jfkr*q|jrj|jg|jj|jqWqW| st j d|r~dnd |t d!dr܈ st j d"tt d} x| jD]\} }|| <qWt d#i}|r0 st j d$t| d} x| jD]\} }|| <qWt d%i}|r sVt j d&t| d} x| jD]\} }|| <qnWS)(Nr)apps)setup)settings dont_loadZSHELL_PLUS_DONT_LOADr&rDZSHELL_PLUS_MODEL_ALIASESZSHELL_PLUS_APP_PREFIXESSHELL_PLUS_PRE_IMPORTScsi}xtjD]\}}t|}j|i}j|}xht|D]\}d||fkrXqB|j|}|s||rxd||f}n|}|j|g||jd||fqBWqW|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. z%s.%sz%s_%s)sorteditemsrget setdefaultrA)Zmodels_to_importapp_modmodelsapp_nameZ app_aliasesprefix model_namealias) app_prefixesrN load_models model_aliasesrr&get_dict_from_names_to_possible_modelss     z>import_objects..get_dict_from_names_to_possible_modelscs6tdg}|r2s"tjdt|jdS)NZSHELL_PLUS_SUBCLASSES_IMPORTz# Shell Plus Subclasses Imports)r:r/ SQL_TABLEr Zcollect_subclasses)Zbase_classes_to_import)perform_automatic_importsrDrMrCrrimport_subclassess  z)import_objects..import_subclassescstj}|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]r_rr import_modelssz%import_objects..import_modelscsx|jD]\}}g}xt|D]\}}y>td||f|<||krR|j|n|jd||fWq tk r}z>jdrtjjdstj d||t |fWYdd}~Xq Xq Wjds tj d|dj |fq WdS) 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. z%s.%sz %s (as %s) tracebackrDz*Failed to import '%s' from '%s' reason: %sNzfrom %s import %sz, ) rQrPrrAr<rRrb print_excr/r0r)r2rB)Zmodules_to_classesZfull_module_pathrUZ model_labelsrXrYr*)rEoptionsrCrrr_s    , z1import_objects..perform_automatic_importsc3s,x&jD]}|jr |j|jfVq WdS)N)Zget_app_configsZ models_moduleZ get_models)Zapp)rKrrget_apps_and_models sz+import_objects..get_apps_and_modelsF)_document_registryTz# Shell Plus User Pre Imports)rDrrz%s.%sz# 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.appsrKZdjangorLreadyZ django.confrMrRr:Zmongoengine.baserfr<r/r^rJrQrrr rSrAr rh)rdrCrLZ dont_load_cliZdont_load_confZdont_load_any_modelsrOr`rareZ mongoenginerfZimportsrIvr$modrVrTZ app_modelsrirjr) rZrKrNr]rEr[r\rdr_rDrMrCrimport_objectss                  ro)F)rrbr=r5typingrrrrZdjango.apps.configrZdjango.utils.module_loadingrZ%django_extensions.collision_resolversrZ#django_extensions.import_subclassesr Z#django_extensions.utils.deprecationr rhr.r rrJrorrrrs*      l