{|j\ d Z ddlZddlZddlZddlZddlZddlZddlZdZ dZ
dZ dZ dZ
dZdZi Zd Zi Zd
Zdd Z G d d
Z G d d Zd Zedk( r e Zyy# e$ r ed Y yw xY w)z3Find modules used by a script, using introspection. N c N t j | g j | y N)packagePathMap
setdefaultappend) packagenamepaths %/usr/lib64/python3.12/modulefinder.pyAddPackagePathr s k2.55d; c |t | < y r
)replacePackageMap)oldnamenewnames r ReplacePackager ) s
!(g r c t j j j t j j j | | }| t dj
| | |j t j j u r ddddt ffS |j t j j u r ddddt ffS |j }|j j | r)dt j j! | ddt" ffS t% |j t j j& rt( }nvt% |j t j j* rt, }nAt% |j t j j. rt0 }n ddddt2 ffS t5 j6 | }t j j9 | d }|||d|ffS )zDAn importlib reimplementation of imp.find_module (for our purposes).NzNo module named {name!r})name rb) importlib machinery
PathFinderinvalidate_caches find_spec ImportErrorformatloaderBuiltinImporter
_C_BUILTINFrozenImporter
_PY_FROZENorigin
is_packageosr dirname_PKG_DIRECTORY
isinstanceSourceFileLoader
_PY_SOURCEExtensionFileLoader _C_EXTENSIONSourcelessFileLoader _PY_COMPILED
_SEARCH_ERRORio open_codesplitext)r r spec file_pathkindfilesuffixs r _find_moduler> - s ""446 ) ) 3 3D$ ?D |4;;;FTRR {{i))999TBJ/// {{i))888TBJ/// I {{ d#RWW__Y/"b.1III$++y 2 2 C CD DKK!4!4!H!H I DKK!4!4!I!I J TBM222
<< "D
WW
i
(
,F VT4 0 00r c e Zd ZddZd Zy)ModuleNc X || _ || _ || _ d | _ i | _ i | _ y r
)__name____file____path____code__ globalnames starimports)selfr r< r s r __init__zModule.__init__\ s2
r c d| j }| j |d| j z }| j |d| j z }|dz }|S )NzModule(, ))rB rC rD )rH ss r __repr__zModule.__repr__i sR == * == $dmm--A == $dmm--A
Gr )NN)rB
__module__ __qualname__rI rN r r r@ r@ Z s r r@ c e Zd ZddZd Zd Zd Zd Zd Zd dZ d d Z
d
Z d Z d d Z
d
Zd Zd Zd Zd dZd Zd Zd Zd Zd dZd Zd Zd Zd Zy) ModuleFinderNc |t j }|| _ i | _ i | _ || _ d| _ ||ng | _ ||ng | _ g | _ y )Nr ) sysr modules
badmodulesdebugindentexcludes
replace_pathsprocessed_paths)rH r rX rZ r[ s r rI zModuleFinder.__init__t sX <88D
$,$8b
.;.G]R ! r c || j k r\t | j D ] }t dd t |d |D ] }t t | d t y y )N end)rX rangerY printrepr)rH levelstrargsiargs r msgzModuleFinder.msg sW DJJ 4;;'e% ( #3 d3iS) G
r c t |d }|| j k r$| j dz | _ | j | y y Nr r rX rY rj rH rg re s r msginzModuleFinder.msgin 8 Q DJJ ++/DK DHHdO r c t |d }|| j k r$| j dz
| _ | j | y y rl rm rn s r msgoutzModuleFinder.msgout rp r c | j dd| t j | 5 }ddt f}| j d||| d d d y # 1 sw Y y xY w)Nr
run_scriptr r __main__)rj r6 r7 r0 load_module)rH pathnamefpstuffs r rt zModuleFinder.run_script sL L(+
\\(
#rz*E ZXu = $
#
#s AAc t j j | \ }}t j j | \ }}t j
| 5 }|dt f}| j |||| d d d y # 1 sw Y y xY w)Nr )r+ r splitr8 r6 r7 r0 rv )rH rw dirr extrx ry s r load_filezModuleFinder.load_file sj GGMM(+ TGG$$T* c
\\(
#r$
+E T2x 7 $
#
#s BB
c | j dd|||| | j || }| j || \ }}| j || }|s|S |j r| j || y )Nr import_hookre )rj determine_parentfind_head_package load_tailrD ensure_fromlist) rH r callerfromlistre parentqtailms r r zModuleFinder.import_hook su M45A&&vU&;((64 NN1d #H :: H -r c | j dd|| |r|dk( r| j dd y |j }|dk\ r|j r|dz }|dk( r*| j | }||u sJ | j dd| |S |j d |k r t
d dj |j d d | }| j | }| j dd| |S |j r*| j | }||u sJ | j dd| |S d|v rK|j d }|d | }| j | }|j |k( sJ | j dd| |S | j dd y ) N r r zdetermine_parent -> Noner zdetermine_parent ->.z relative importpath too deep)
ro rr rB rD rV countr" joinr{ rfind)rH r re pnamer rh s r r z ModuleFinder.determine_parent s
1(&%8! KK5 6 A:
z e, ''' A 4f=
{{3 %'!"@AAHHU[[ -gv 67E\\%(F KK0& 9M ??\\%(FV# ## KK0& 9M %< C A"1IE\\%(F??e+ ++ KK0& 9M A12r c | j dd|| d|v r |j d }|d | }||dz d }n|}d}|r|j d|}n|}| j ||| }|r| j dd||f ||fS |r2|}d }| j ||| }|r| j dd||f ||fS | j dd| t d|z ) Nr r r r r zfind_head_package ->"raise ImportError: No module namedNo module named )ro findrB
import_modulerr r" )rH r r rh headr qnamer s r r z ModuleFinder.find_head_package s
1)648 $; #A8D!:DDD %5EE tUF 3 KK1At9 =d7N EF""47A A 54yA$w A;UC,u455r c L | j dd|| |}|rx|j d }|dk r t | }|d | ||dz d }}|j d|}| j ||| }|s!| j dd| t
d|z |rx| j dd| |S ) Nr r r r r r r z load_tail ->)ro r lenrB r rr r" )rH r r r rh r mnames r r zModuleFinder.load_tail s
1k1d+
#A1u#d)a bq4!:$D zz40E""42A A CUK!"4u"<==
A~q)r c | j dd||| |D ]s }|dk( r+|r | j | }|s | j ||d 3t || r@|j d|}| j ||| }|rgt
d|z y )Nr r *r r r )rj find_all_submodulesr hasattrrB r r" )rH r r recursivesuballsubnamesubmods r r z ModuleFinder.ensure_fromlist s %q(I>Ccz 2215C ,,QQ7 Q_%&ZZ5 ++C!< %&87&BCC r c
|j sy i }g }|t j j d d z
}|t j j d d z
}|t j j
d d z
}|j D ]R } t
j | }|D ]5 }d }|D ] }t | } || d |k( s|d | } n |s+|dk7 s1|||< 7 T |j S # t $ r | j dd| Y w xY w)Nr zcan't list directoryrI ) rD r r EXTENSION_SUFFIXESSOURCE_SUFFIXESBYTECODE_SUFFIXESr+ listdirOSErrorrj r keys)
rH r rV suffixesr| namesr modsuffns
r r z ModuleFinder.find_all_submodules s zz I''::1==I''77::I''99!<<::C
3 $DD AQBCyD( "3QBi %
3*,#&GCL
||~
2C8
s C## DDc r | j dd||| | j | }| j dd| |S # t $ r Y nw xY w|| j v r| j dd y |r |j
| j dd y | j
||xr |j
| \ }}}n"# t $ r | j ddd Y y w xY w | j |||| }|r)|j n# |r|j w w xY w|r
t ||| | j dd| |S )Nr r zimport_module ->zimport_module -> None) ro rV rr KeyErrorrW rD find_moduler" rv closesetattr)rH partnamefqnamer r rx rw ry s r r zModuleFinder.import_module$ sD
1ox@ V$A
KK-q 1H
T__ $ KK2 3 foo- KK2 3 "&"2"28393Mfoov#W B% KK-t 4 Xu=A
FHa ( A)1-s- ; AA
%B3 3 CCC= =Dc |\ }}}| j dd||xr d| |t k( r'| j || }| j dd| |S |t k( r t |j
|d } n_|t k( rT |j
}
t j j |
|i t j t |
dd } nd } | j! | }||_ | r6| j$ r| j' | } | |_ | j+ | | | j dd| |S # t $ r%} | j ddt | z | d } ~ ww xY w)Nr rv rx zload_module ->execzraise ImportError: )ro r- load_packagerr r0 compilereadr4 r _bootstrap_external
_classify_pycr" rf marshalloads
memoryview
add_modulerC r[ replace_paths_in_coderE scan_code) rH r rx rw file_infor= modetyper codataexcs r rv zModuleFinder.load_moduleD sN &d
1mVR[D(C > !!!&(3A KK+Q /H : Hf5B
\
!
wwy--;;D&"M z$ / 45BB OOF #
!!//3AJ NN2q ! A'+
A 4s3x ?J
s ?1D9 9 E' E""E'c || j vri | j |< |r d| j | |j < y d| j | d< y )Nr -)rW rB )rH r r s r _add_badmodulezModuleFinder._add_badmodulea sI t &$&DOOD ! 56DOOD !&// 2)*DOOD !# &r c || j v r| j || y | j ||| |rH|D ]B }|dz |z }|| j v r| j || , | j |||g| D y y # t $ r8}| j ddt | | j || Y d }~d }~ww xY w# t $ r8}| j ddt | | j || Y d }~y d }~wt $ r8}| j ddt | | j || Y d }~y d }~ww xY w)Nr r r z ImportError:z SyntaxError:)rW r r r" rj rf SyntaxError)rH r r r re r fullnamerj s r _safe_import_hookz ModuleFinder._safe_import_hooki s; 4?? " f - > T6 7 #C #czC /H 4??2 ++Hf= > ((vuE(J
$ ' > NCH= ++Hf==> . HHQC 1 f - - . HHQC 1 f - - .s; C (B C .B>>C E.D E.EEc # K t j | D ] }d|ff t j | D ] \ }}}|dk( r d||ff d|||ff y w)Nstorer absolute_importrelative_import)dis_find_store_names
_find_imports)rH r r re r s r scan_opcodeszModuleFinder.scan_opcodes sm ))"-DD7" " .%(%6%6r%: !D%z'(D)999'%4)@@@ &;s AAc |j }| j } || D ] \ }}|dk( r|\ }d|j |< |dk( r|\ }}d} |d|v rd} |D
cg c]
}
|
dk7 s |
}}
| j |||d | sbd } |j r+| j
j
|j dz |z } | | j
j
| } | i|j j | j |j j | j | j d|j |< #d|j |< 4|dk( rS|\ } }}|r| j |||| X| j || }
| j |
j d |d t | |j D ]* }t |t | s| j! || , y c c}
w ) Nr r r r r r r r )co_coder rF r rD rV getrB updaterG rE r RuntimeError co_constsr. r r )rH r r codescannerwhatrg r r have_starfmmre r cs r r zModuleFinder.scan_code s zz##!"+JD$w &'
d#**!%$ 'h$% +3 @8aqCx8H @&&tQ&B Bzz "\\ - -ajj3.>.E Fz !\\ - -d 3~
,,R^^<
,,R^^< ;;.23AMM$ /./
d+**(,%x**4HE*J ! 2 21E 2 BF**6??D(RS*T #4((Q &T A!T"X&q!$ A As
G3%G3c | j dd|| t j | }|r|}| j | }||_ |g|_ |j
t j |g z |_ | j d|j
\ }}} | j |||| | j dd| ||r|j S S # |r|j w w xY w)Nr r rI zload_package ->) ro r r r rC rD r r rv rr r )rH r rw r r rx bufry s r r zModuleFinder.load_package s
1nfh7#''/ F OOF #
Z
ZZ."4"4VR"@@
))*ajjAC VRe 4 KK,a 0
r
s (C C(c t || j v r| j | S t | x| j |< }|S r
)rV r@ )rH r r s r r zModuleFinder.add_module s9 T\\ !<<' '#)&>1 V qr c ||j dz |z }n|}|| j v r | j dd| t | |*|t j
v r d d ddt ffS | j }t || S )Nr r zfind_module -> Excludedr ) rB rZ rr r" rU builtin_module_namesr& r r> )rH r r r r s r r zModuleFinder.find_module s s*4/HH t}} $ KK4h ? d# # <s/// dRZ$89999DD$''r c t t ddz t ddz t | j j }|D ]U }| j | }|j rt dd n
t dd t d|z |j
xs d W | j
\ }}|r_t t d
|D ]E }t | j | j }t d |d d
j | G |rmt t dd t d |D ]E }t | j | j }t d |d d
j | G yy)zPrint a report to stdout, listing the found modules with their
paths, as well as modules that are missing, or seem to be missing.
z
%-25s %s)NameFile)----r Pr_ r` r z%-25sr zMissing modules:?z
imported fromrK z7Submodules that appear to be missing, but could also bez#global names in the parent package:N) rc sortedrV r rD rC any_missing_mayberW r )rH r keyr missingmayber modss r reportzModuleFinder.report sG
l --.
l --.dll'')*C S!Azzcs#cs# 'C-!1r 2
//1 G $ % dood 3 8 8 :;c4 $))D/B G KQT U 7 8 dood 3 8 8 :;c4 $))D/B r c 2 | j \ }}||z S )zReturn a list of modules that appear to be missing. Use
any_missing_maybe() if you want to know which modules are
certain to be missing, and which *may* be missing.
)r )rH r r s r any_missingzModuleFinder.any_missing
s
//1 r c g }g }| j D ] }|| j v r|j d }|dk r|j | :||dz d }|d| }| j j | }|b|| j | v r|j | ||j v r|j r|j | |j | |j | |j |j ||fS )a Return two lists, one with modules that are certainly missing
and one with modules that *may* be missing. The latter names could
either be submodules *or* just global names in the package.
The reason it can't always be determined is that it's impossible to
tell which names are imported when "from module import *" is done
with an extension module, short of actually importing it.
r r r N) rW rZ r r
rV r rF rG sort)rH r r r rh r pkgnamepkgs r r z ModuleFinder.any_missing_maybe s OODt}}$
3A1ut$1Q34jG2AhG,,""7+Cdood 33 NN4( /__ LL& NN4(t$= $>
~ r c t j j |j x}}| j D ]) \ }}|j | s||t
| d z } n | j r]|| j vrO||k7 r| j dd|d| n| j dd|d | j j | t |j }t t
| D ]2 }t || t | s | j! || ||< 4 |j# t% | | S )Nr z co_filename z changed to z remains unchanged)r co_filename)r+ r normpathr r[
startswithr rX r\ rr r
listr rb r. r r replacetuple)rH r new_filenameoriginal_filenamer rconstsrh s r r z"ModuleFinder.replace_paths_in_codeC s& +-77+;+;BNN+KK (&&DAq ++A. #4SVW#= = '
::+43G3GG 00 A'8 H I A'8 ; < ' '(9 :bll#s6{#A&)T"X. 6 6vay Aq $ zzE&M|zLLr )Nr NN)NNr )r )r r
) rB rO rP rI rj ro rr rt r~ r r r r r r r rv r r r r r r r r r r r rQ r r rS rS r s
" > 8 !F66
D 8 @:+>4A/%b* ($ CD , \Mr rS c d dd l } | j t j dd d \ }}d}d}g }g }|D ][ \ }} |dk( r|dz }|dk( rd}|dk( r"|| j t j z }|dk( rd}|dk( sK|j | ] |sd }
n|d }
t j d d } t j j |
| d< || z } |dkD r't d
| D ] } t d t | t | || }
|dd D ]O }|dk( rd} |r2|d d d
k( r|
j |d d d dg -|
j | ?|
j | Q |
j |
|
j! |
S # | j $ r}t | Y d }~y d }~ww xY w)Nr r zdmp:qx:z-dz-mz-pz-qz-xzhello.pyzpath:r^ z.*r )getoptrU argverrorrc r{ r+ pathsepr
r r, rd rS r r~ rt r )r optsrg rj rX domodsaddpathexcludeoascriptr itemmfri s r testr [ s ]]388AB<;
d
E
FGG1 9AIE 9F 9
33G 9E 9 NN1 a 88A;Dggoof%DG T>D qy
gD %d $
dE7 +BABx $;F 23x4 s3Bxu5s# LL MM&IIK
Ig <<
c
s &F F/ F**F/ru z
[interrupted]r
) __doc__r importlib._bootstrap_externalr importlib.machineryr r+ r6 rU r5 r0 r4 r2 r- r& r( r r r r r> r@ rS r rB r KeyboardInterruptrc rQ r r