!A?h\ d Z ddlZddlZddlZddlZddlZddlZddlZdZ dZ
dZ dZ dZ
dZdZi Zd Zi Zd
Zdd Z G d d
Z G d d Zd Zedk r$ e ZdS # e$ r ed Y dS w xY wdS )z3Find modules used by a script, using introspection. N c b t | g | d S N)packagePathMap
setdefaultappend) packagenamepaths 9/opt/alt/python-internal/lib64/python3.11/modulefinder.pyAddPackagePathr s, k2..55d;;;;; c |t | < d S r
)replacePackageMap)oldnamenewnames r ReplacePackager ) s !(g r c t j j t j j | | }|%t d | | |j t j j u r
ddddt ffS |j t j j
u r
ddddt ffS |j }|j
| r*dt j | ddt" ffS t% |j t j j rt( }net% |j t j j rt, }n9t% |j t j j rt0 }n
ddddt2 ffS t5 j | }t j | 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 "44666 ) 3 3D$ ? ?D |4;;;FFTRRRR {i)999TBJ/// {i)888TBJ/// I { d## JRW__Y//"b.1III$+y 2 CDD
3 DK!4!H I I 3 DK!4!I J J 3 TBM222
< " "D
W
i
(
(
,F VT4 0 00r c e Zd ZddZd ZdS )ModuleNc Z || _ || _ || _ d | _ i | _ i | _ d S r
)__name____file____path____code__ globalnames starimports)selfr r< r s r __init__zModule.__init__\ s8
r c t d| j }| j
|d| j z }| j
|d| j z }|dz }|S )NzModule(, ))rB rC rD )rH ss r __repr__zModule.__repr__i sT == * = $dmm--A = $dmm--A
Gr )NN)rB
__module__ __qualname__rI rN r r r@ r@ Z s7 r r@ c e Zd Zd dZd Zd Zd Zd Zd Zd d
Z d d Z
d Z d
Z d dZ
d Zd Zd Zd Zd dZd Zd Zd Zd Zd!dZd Zd Zd Zd ZdS )" ModuleFinderNr c | t j }|| _ i | _ i | _ || _ d| _ ||ng | _ ||ng | _ g | _ d S )Nr ) sysr modules
badmodulesdebugindentexcludes
replace_pathsprocessed_paths)rH r rX rZ r[ s r rI zModuleFinder.__init__t s` <8D
$,$8b
.;.G]]R ! r c || j k rlt | j D ]}t dd t |d |D ] }t t | d !t d S d S )N end)rX rangerY printrepr)rH levelstrargsiargs r msgzModuleFinder.msg s DJ 4;''
&
&e%%%%% #3
*
*d3iiS))))) GGGGG
r c b |d }|| j k r| j dz | _ | j | d S d S Nr r rX rY rj rH rg re s r msginzModuleFinder.msgin > Q DJ +/DK DHdOOOO r c b |d }|| j k r| j dz
| _ | j | d S d S rl rm rn s r msgoutzModuleFinder.msgout rp r c | dd| t j | 5 }ddt f}| d||| d d d d S # 1 swxY w Y d S )Nr
run_scriptr r __main__)rj r6 r7 r0 load_module)rH pathnamefpstuffs r rt zModuleFinder.run_script s L(+++
\(
#
# >rz*E ZXu = = = > > > > > > > > > > > > > > > > > >s #A A #A c , t j | \ }}t j | \ }}t j | 5 }|dt f}| |||| d d d d S # 1 swxY w Y d S )Nr )r+ r splitr8 r6 r7 r0 rv )rH rw dirr extrx ry s r load_filezModuleFinder.load_file s GMM(++ TG$$T** c
\(
#
# 8r$
+E T2x 7 7 7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8s #B B
B
r c | dd|||| | || }| || \ }}| || }|s|S |j r| || d S )Nr import_hookre )rj determine_parentfind_head_package load_tailrD ensure_fromlist) rH r callerfromlistre parentqtailms r r zModuleFinder.import_hook s M45AAA&&vU&;;((664 NN1d # # H : . H - - -tr c H | dd|| |r|dk r| dd d S |j }|dk r|j r|dz }|dk r,| j | }||u sJ | dd| |S | d |k rt
d d | d d | }| j | }| dd| |S |j r,| j | }||u sJ | dd| |S d|v rR| d }|d | }| j | }|j |k sJ | dd| |S | dd d S ) 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(&%888 ! KK5 6 6 64 A::
zz e, '''' A 4f===
{{3 %''!"@AAAHHU[[ - -gvg 677E\%(F KK0& 9 9 9M ? \%(FV#### KK0& 9 9 9M %<< C A"1"IE\%(F?e++++ KK0& 9 9 9M A1222tr c | dd|| d|v r-| d }|d | }||dz d }n|}d}|r
|j d|}n|}| ||| }|r | dd||f ||fS |r:|}d }| ||| }|r | dd||f ||fS | 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 sA
1)64888 $;; #A8D!:DDDD %5EEE tUF 3 3 KK1At9 = = =d7N EF""477A
A 54yAAA$w A;UCCC,u4555r c | dd|| |}|r| d }|dk rt | }|d | ||dz d }}|j d|}| ||| }|s)| dd| t
d|z || 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+++
> #A1uu#d))a bqb4!:$D zzz440E""422A
> A CUKKK!"4u"<=== >
A~q)))r c > | dd||| |D ]}|dk r1|s.| | }|r| ||d 9t || s7|j d|}| ||| }|st
d|z d S )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>>>
D
DCczz 8 22155C 8 ,,QQ777 Q__
D%&ZZZ5 ++C!<< D%&87&BCCC
D
Dr c |j sd S i }g }|t j j d d z
}|t j j d d z
}|t j j d d z
}|j D ]} t
j | }n'# t $ r | dd| Y ;w xY w|D ]A}d }|D ]-}t | } || d |k r
|d | } n.|r |dk r|||< B| S )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 sU z F I':111==I'7::I'9!!!<<: ' 'C
3
2C888
'
' $ DD AQBCCyD(( "3QB3i ) '3*,,#&GCL
' ||~~ s ,B!B%$B%c | dd||| | j | }| dd| |S # t $ r Y nw xY w|| j v r| dd d S |r |j | dd d S | ||o|j | \ }}}n(# t $ r | ddd Y d S w xY w | |||| }|r| n # |r| w w xY w|rt ||| | 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$ s
1ox@@@ V$A
KK-q 1 1 1H D
T_ $ $ KK2 3 3 34 fo- KK2 3 3 34 "&"2"28393Mfov#W #W B%% KK-t 4 4 444 Xu==A
) FHa ( ( ( A)1---s/
A
A
A"B7 7!C C D D(c F |\ }}}| dd||od| |t k r/| || }| dd| |S |t k r$t | |d } n|t k r | }
t j
|
|i n:# t $ r-} | ddt | z | d } ~ ww xY wt j t |
dd } nd } | | }||_ | r9| j r| | } | |_ | | | | dd| |S )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 s &d
1mVR[D(CCC > ! !!!&(33A KK+Q / / /H : Hf55BB
\
!
!
wwyy-;;D&"MMMM
A 4s3xx ?JJJ
z$ / / 455BBB OOF # #
"!
4//33AJ NN2q ! ! ! A'+++s 5C
D(C??Dc ~ || j vr
i | j |< |rd| j | |j < d S d| j | d< d S )Nr -)rW rB )rH r r s r _add_badmodulezModuleFinder._add_badmodulea sS t & &$&DOD ! +56DOD !&/ 2 2 2)*DOD !# & & &r c || j v r| || d S | ||| |r|D ]}|dz |z }|| j v r| || * | |||g| F# t $ rD}| ddt | | || Y d }~d }~ww xY wd S d S # t $ rE}| ddt | | || Y d }~d S d }~wt $ rE}| ddt | | || Y d }~d S 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 - - - F > T6 7 7 7
># > >C #czC /H 4?22 ++Hf=== > ((vuE(JJJJ& > > > NCHH=== ++Hf========>
>
> > > . . . HHQC 1 1 1 f - - - - - - - - - . . . HHQC 1 1 1 f - - - - - - - - - .s; C *B
C:CC
E6$:D$$
E61:E11E6c # K t j | D ] }d|ffV
t j | D ] \ }}}|dk r d||ffV d|||ffV d S )Nstorer absolute_importrelative_import)dis_find_store_names
_find_imports)rH r r re r s r scan_opcodeszModuleFinder.scan_opcodes s )"-- # #DD7" " " " "%(%6r%:%: A A !D%zz'(D)999999'%4)@@@@@@ A Ar c |j }| j } || D ]k\ }}|dk r|\ }d|j |< |dk r|\ }}d} |d|v rd} d |D }| |||d | rd }
|j r%| j |j dz |z }
|
| j | }
|
P|j |
j |j |
j |
j
d|j |< d|j |< |d k rZ|\ } }}|r| |||| '| || } | | j d |d ^t | |j
D ]5}
t |
t | r| |
| 6d S )
Nr r r r r c g | ]
}|d k | S )r rQ ).0fs r