khFddlZddlZddlZddlZddlmZmZmZmZddl m Z ddl m Z ddl mZddlmZddlmZgdZGd d eZd Zdd ZdZdS)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, SubqueryceZdZdS)ObjectImportErrorN)__name__ __module__ __qualname__Z/var/www/histauto/venv/lib/python3.11/site-packages/django_extensions/management/shells.pyr r !sDrr ctt|d} ||tdzS#t $r |dcYSwxYw#t $r|cYSwxYw)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_namerpartss r get_app_namer%s"(8>>#..// 0 0F &,,'9::Q>? ?   !9     s)"AA*'A-)A**A-- A<;A<Fc i}|D]9}t|tr|} t|tr2|dr t j|}nF#t $r9}|s-t|d|d|Yd}~Yd}~nd}~wwxYwtd|j Ds(|s%t|d|z|s%t| d|z|j D]H}t|tj r4|j D],}|jp|j} t!j|j|| <-t|tjrt!j|jiid|j D} |j D]}|jp|j} |jd kr&t+| D]} t-| | || <nt-| |j|| <[#t.$rC}td |jd |jd t+| t1|d}~wwxYwJnt3jd t6dt|trVt'|} | ||dd<|s%t| d|zt|t:t<frt?|dkrt|dts/|s+t|d|dzwt|dt:t<frtd|dDrt'|dii|d} g} |dD]o} t-| |||<| |,#t.$r7|s2t|d|d|dd|dYlwxYw|sAt| d|ddd!| nlt|dtr|dd krqt'|dii|d} t+| D]} t-| | || <|s+t| d|dznt-t'|dii|dg|d} | ||d<|s4t| d|dd|dn_|s5t|d|dd|ddn'|s%t|d|z#t0$r+|s%t|d |zY7wxYw|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 zimport zError parsing:  Nc3bK|]*}t|tjtjfV+dSN) isinstanceastImport ImportFrom).0bodys r zimport_items..]sFGKJtcj#.%ABBrz"Only specify import statements: %rz%scg|] }|j Sr)name)r)r-s r z import_items..ts1S1S1S$)1S1S1Sr*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 stringrc3@K|]}t|tVdSr$)r%str)r)es rr+zimport_items..sEGG/0 1c**GGGGGGrzUnable to import z from z: z does not existr! import , zfrom %s import *z: names must be stringsz*Unable to import %r: names must be stringszUnable to import %r)"r%r3strip startswithr&parse ExceptionprintERRORallr* SQL_COLTYPEr'namesasnamer- importlib import_moduler( __import__moduledirgetattrAttributeError ImportErrorwarningswarnr rrtuplelenappendjoin) import_directivesstyle quiet_loadimported_objects directivenodeexcr*r-r@imported_objectkimported_namess r import_itemsrYBs&]F]F i % % *!))IZ F)S))V i.B.B$//V !9Y//DD !!!%!ekkkIIIss*STTUUU !!!!!!OSy &!KK Dy P !?%++D9,<==>>> I77D!$ 33$(JD%)[%=DIF7@7N $ 88,V44"$777*3*> KR1S1S 1S1S1S++%)J77D%)[%=DIF7#'9#3#3-0-A-A%*%*>E,;Q?*?*(8(;(;%* @G(7@&@&$4V$<$2777 %)N)N)N )N)N7:?7K7K)N)N!"!"!"'2#&6&6 67/7B L/  i--X&0&;&;O@O$Y__S%9%9!%<=%Je// i0GHHIII D%=99Rc)nnPQ>Q>Q%ilC88 !) ! % )6'0l %3!"!"!!)A,u >>=3GG4=aLGGGDD= +5Yq\2r9UV<*X*X)+$-aL < M,Yq\:#-" %$)$5$5$5+4Q<<<1)G%&%&!"!"!" *! % '0|||Yq\\\%C!"!"&!KK Ly X  F F F Fekk"7)"CDDEEE F s,XA43X4 B7>*B2(X-X2B77AXCXA H&%X& I3 0>I. .I3 3B XA9XQXQF4X1X=<X=cddlmddlm}js |ddlmdg}tdg}||zdv}dtd itd itd i}iifd fd }fd}fdfd} d} ddl m } d} n#t$rYnwxYw|rUs"t dt|} | D] \} }|| < | r|s| D]}\}}|dd}t#|j}|vs |d|vrB|jg|j|~|s| D]|\}}|st#|j}|vr!|D]X}|d|jvr|jr@|jg|j|jY}|s)t d|rdndz|tddrZs"t dtt,} | D] \} }|| < tdi}|rUs"t dt|} | D] \} }|| < tdi}|rUs"t d t|} | D] \} }|| < S)!Nr)apps)setup)settings dont_loadSHELL_PLUS_DONT_LOADr/rQSHELL_PLUS_MODEL_ALIASESSHELL_PLUS_APP_PREFIXESSHELL_PLUS_PRE_IMPORTSci}t D]\}}t|} |i}|}t|D]e}|d| vr ||}|s |r|d|}n|}||g|||d|f|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. r_)sorteditemsrget setdefaultrM) models_to_importapp_modmodelsapp_name app_aliasesprefix model_namealias app_prefixesr^ load_models model_aliasess r&get_dict_from_names_to_possible_modelsz>import_objects..get_dict_from_names_to_possible_modelss%k&7&7&9&9:: P POGV#G,,H'++Hb99K!%%h//F$Vnn P P &hh 3y@@# 33+++166:: > * ++E2666 '..'''::/NOOOO P rctdg}|rPs"tdt|dSdS)NSHELL_PLUS_SUBCLASSES_IMPORTz# Shell Plus Subclasses Imports)rFr; SQL_TABLEr collect_subclasses)base_classes_to_importperform_automatic_importsrQr]rPs rimport_subclassesz)import_objects..import_subclasses%s!(3QSU!V!V !  Jeoo&GHHIII % % !788KKMM       rcpt}|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_modelsrtrzs r import_modelsz%import_objects..import_models.sG 566MM 2 2 4 4   "!"344444rc|D]H\}}g}t|D]\}} t|d||<||kr||n||d|dU#t$r}drt jds8t d|d|dt|Yd }~d }~wwxYwds;t d |d d |Jd S) 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. rz (as ) tracebackrQzFailed to import 'z' from 'z ' reason: Nr!r5r6) rfrerrMrHrgr print_excr;r<r3r>rN) modules_to_classesfull_module_pathrk model_labelsrorpr4rRoptionsrPs rrzz1import_objects..perform_automatic_imports9s);(@(@(B(B   $ fL%+F^^  ! E.;#3#3#3ZZ@//$U+"U**$++J7777$++JJJ,NOOO"   {{;//.!+---";;|44!KKK#-::/?/?/?Q!I ;;|,, %%%+++TYY|-D-D-DF-  sAB D  A5DD c3KD]&}|jr|j|fV'dSr$)get_app_configs models_module get_models)appr[s rget_apps_and_modelsz+import_objects..get_apps_and_models]sV'')) : :C  :')9)99999 : :rF)_document_registryTz# Shell Plus User Pre Imports)rQrz# 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.appsr[djangor\ready django.confr]rgrFmongoengine.baserrHr;rwrYrfrrrrhrMrr) rrPr\ dont_load_clidont_load_confdont_load_any_modelsrbr{rr mongoenginerimportsrWvr-modrlrj app_modelsrrrqr[r^rtrRrrrsrzrQr]s `` @@@@@@@@@@rimport_objectsrs       : $$$$$$KK R00MX'=rBBN.I)+\**JH&@"EEM8%>CCL$X/GLLK        < 5 5 5 5 5 5"""""""H::::: K 777777      $ D %//"ABB C C C5uTTTMMOO $ $DAq"# Q  5/5+1133 5 5ID#::c??2&D#CN33H9$$HHHdd)C (Q(Q  " "3>2 6 6 6  ' . .t 4 4 4 4 E#6#6#8#8 E E GZ #G$455H9$$! E E&hh 5BB>E**3>2>>>/66s|DDD E   OO: ; ;19zzr ;   MOOOx4d;;$ B %//"?@@ A A A8%JWWWMMOO $ $DAq"# Q   +?DD$ @ %//"=>> ? ? ?15ZPPPMMOO $ $DAq"# Q  &h0I2NN$ E %//"BCC D D D6*UUUMMOO $ $DAq"# Q   sC C'&C')F)r&rrIrAtypingrrrrdjango.apps.configrdjango.utils.module_loadingr%django_extensions.collision_resolversr #django_extensions.import_subclassesr #django_extensions.utils.deprecationr rr:r rrYrrrrrsI 211111555555JJJJJJ@@@@@@KKKKKK           :mmmm`~~~~~r