a
XC?hQ_ @ s
d Z ddlZddlZddlZddlZddlZddlZddlZej d Z
ej d Z ej d Z ej d Z
e e
fZejZdZdZdZd Zd
Zd Zd Zi Zd
d Zi Zdd ZdddZG dd dZ G dd dZ dd Z e dkrz
e Z W n e!y e"d Y n0 dS ) z3Find modules used by a script, using introspection. N
LOAD_CONST IMPORT_NAME
STORE_NAME STORE_GLOBAL c C s t | g | d S N)packagePathMap
setdefaultappend)Z packagenamepath r 1/opt/alt/python39/lib64/python3.9/modulefinder.pyAddPackagePath& s r c C s |t | < d S r )replacePackageMap)Zoldnamenewnamer r r ReplacePackage1 s r c C s t jj t jj| |}|du r8tdj| d| d|jt jju rVddddt ffS |jt jj
u rtddddt ffS |j }|j
| rdtj|ddtffS t|jt jjrt}n r? )rC sr r r __repr__q s
zModule.__repr__)NN)r=
__module__ __qualname__rD rG r r r r r< b s
r< c @ s e Zd Zd6ddZdd Zdd Zd d
Zd d Zd
d Zd7ddZ d8ddZ
dd Z dd Z d9ddZ
dd Zd d Zd d Zd d! Zd:d"d#Zd$d% Zd&d' Zd(d) Zd*d+ Zd;d,d-Zd.d/ Zd0d1 Zd2d3 Zd4d5 ZdS )< ModuleFinderNr c C sZ |d u rt j}|| _i | _i | _|| _d| _|d ur8|ng | _|d urJ|ng | _g | _d S )Nr ) sysr modules
badmodulesdebugindentexcludes
replace_pathsprocessed_paths)rC r rN rP rQ r r r rD | s zModuleFinder.__init__c G sV || j krRt| jD ]}tddd qt|dd |D ]}tt|dd q6t d S )N end)rN rangerO printrepr)rC levelstrargsiargr r r msg s
zModuleFinder.msgc G s, |d }|| j kr(| jd | _| j| d S Nr r rN rO r_ rC r\ rZ r r r msgin s
zModuleFinder.msginc G s, |d }|| j kr(| jd | _| j| d S r` ra rb r r r msgout s
zModuleFinder.msgoutc C sV | dd| t|*}ddtf}| d||| W d n1 sH0 Y d S )Nr
run_scriptr r __main__)r_ r4 r5 r. load_module)rC pathnamefpstuffr r r re s
zModuleFinder.run_scriptc C sh t j|\}}t j|\}}t|*}|dtf}| |||| W d n1 sZ0 Y d S )Nr )r) r splitr6 r4 r5 r. rg )rC rh dirr Zextri rj r r r load_file s
zModuleFinder.load_filer c C s\ | dd|||| | j||d}| ||\}}| ||}|sF|S |jrX| || d S )Nr import_hookrZ )r_ determine_parentfind_head_package load_tailr? ensure_fromlist) rC r callerfromlistrZ parentqtailmr r r rn s zModuleFinder.import_hookc C sN | dd|| |r |dkr,| dd d S |j}|dkr|jrH|d8 }|dkrx| j| }||u sfJ | dd| |S |d|k rtdd|dd | }| j| }| dd| |S |jr| j| }||u sJ | dd| |S d|v r>| d}|d | }| j| }|j|ks,J | dd| |S | dd d S ) N rp r zdetermine_parent -> Noner zdetermine_parent ->.z relative importpath too deep)
rc rd r= r? rL countr joinrk rfind)rC rt rZ Zpnamerv r] r r r rp sB
z ModuleFinder.determine_parentc C s | dd|| d|v r@|d}|d | }||d d }n|}d}|r\d|j|f }n|}| |||}|r| dd||f ||fS |r|}d }| |||}|r| dd||f ||fS | dd| td | d S )
Nrz rq r{ r r %s.%szfind_head_package ->"raise ImportError: No module namedNo module named )rc findr=
import_modulerd r )rC rv r r] headrx Zqnamerw r r r rq s.
z ModuleFinder.find_head_packagec C s | dd|| |}|r|d}|dk r2t|}|d | ||d d }}d|j|f }| |||}|s| dd| td| q| dd | |S )
Nrz rr r{ r r r r r z load_tail ->)rc r lenr= r rd r )rC rw rx ry r] r Zmnamer r r rr s
zModuleFinder.load_tailc C s | dd||| |D ]d}|dkrD|sz| |}|rz| ||d qt||sd|j|f }| |||}|std| qd S )Nrz rs *r r r )r_ find_all_submodulesrs hasattrr= r r )rC ry ru recursivesuballsubnameZsubmodr r r rs s
z ModuleFinder.ensure_fromlistc
C s |j s
d S i }g }|tjjd d 7 }|tjjd d 7 }|tjjd d 7 }|j D ]}zt|}W n$ ty | dd| Y qTY n0 |D ]R}d }|D ]0}t
|} || d |kr|d | } qq|r|dkr|||< qqT| S )Nr zcan't list directoryrD ) r? r r EXTENSION_SUFFIXESSOURCE_SUFFIXESBYTECODE_SUFFIXESr) listdirOSErrorr_ r keys)
rC ry rL suffixesrl namesr modZsuffnr r r r s.
z ModuleFinder.find_all_submodulesc C s | dd||| z| j| }W n ty2 Y n0 | dd| |S || jv r`| dd d S |r~|jd u r~| dd d S z | ||o|j|\}}}W n" ty | ddd Y d S 0 z | ||||}W |r| n|r| 0 |rt
||| | dd| |S )Nr r zimport_module ->zimport_module -> None) rc rL KeyErrorrd rM r? find_moduler rg closesetattr)rC Zpartnamefqnamerv ry ri rh rj r r r r , s<
zModuleFinder.import_modulec
C s* |\}}}| dd||od| |tkrF| ||}| dd| |S |tkr`t| |d} n~|tkrz | }
tj
|
|i W n< t y } z$| ddt | | W Y d } ~ n
d } ~ 0 0 t
t|
dd } nd } | |}||_| r| jr| | } | |_| | | | dd| |S )Nr rg ri zload_module ->execzraise ImportError: )rc r+ load_packagerd r. compilereadr2 r _bootstrap_external
_classify_pycr r[ marshalloads
memoryview
add_moduler> rQ replace_paths_in_coder@ scan_code) rC r ri rh file_infor: modetypery codataexcr r r rg L s4
zModuleFinder.load_modulec C s< || j vri | j |< |r*d| j | |j< nd| j | d< d S )Nr -)rM r= )rC r rt r r r _add_badmodulei s
zModuleFinder._add_badmodulec C sH || j v r| || d S z| j|||d W n typ } z*| ddt| | || W Y d }~nd }~0 ty } z*| ddt| | || W Y d }~nd }~0 0 |rD|D ]}|d | }|| j v r| || qz| j|||g|d W q ty@ } z*| ddt| | || W Y d }~qd }~0 0 qd S )Nro r z ImportError:z SyntaxError:r{ )rM r rn r r_ r[ SyntaxError)rC r rt ru rZ r_ r fullnamer r r _safe_import_hookq s,
"
z ModuleFinder._safe_import_hookc c s |j }|j}|j}dd t|D }t|D ]\}\}}|tv rTd|| ffV q.|tkr.|dkr.||d d ||d d krtkr.n q.|||d d } |||d d }
| dkrd|
|| ffV q.d| |
|| ffV q.q.d S ) Nc S s" g | ]\}}}|t kr||fqS r ) EXTENDED_ARG).0_opr^ r r r
s z-ModuleFinder.scan_opcodes..storer r r absolute_importrelative_import) co_codeco_names co_constsdisZ_unpack_opargs enumerate STORE_OPSr r ) rC r coder constsZopargsr] r ZopargrZ ru r r r scan_opcodes s&
zModuleFinder.scan_opcodesc C s |j }| j}||D ]F\}}|dkr8|\}d|j|< q|dkr|\}}d} |d urpd|v rbd} dd |D }| j|||dd | r\d }
|jr| j|jd | }
|
d u r| j|}
|
d ur|j|
j |j |
j |
j
d u rd|j |< n
d|j |< q|d
krT|\} }}|r0| j|||| d n"| j || d} | j| jd |dd qt |q|j
D ]"}
t|
t|rd| |
| qdd S ) Nr r r r r c S s g | ]}|d kr|qS )r r )r fr r r r z*ModuleFinder.scan_code..ro r{ r )r r rA r r? rL getr= updaterB r@ rp RuntimeErrorr r, r r )rC r ry r scannerZwhatr\ r ru Z have_starZmmrZ rv cr r r r sH
zModuleFinder.scan_codec C s | dd|| t|}|r"|}| |}||_|g|_|jt|g |_| d|j\}}}z2| |||| | dd| |W |r|
S n|r|
0 d S )Nr r rD zload_package ->) rc r r r r> r? r
r rg rd r )rC r rh r ry ri Zbufrj r r r r s(
zModuleFinder.load_packagec C s* || j v r| j | S t| | j |< }|S r )rL r< )rC r ry r r r r s
zModuleFinder.add_modulec C sn |d ur|j d | }n|}|| jv r<| dd| t||d u rd|tjv r^d d ddtffS | j}t||S )Nr{ r zfind_module -> Excludedr ) r= rP rd r rK builtin_module_namesr$ r r; )rC r r rv r r r r r s
zModuleFinder.find_modulec C s t t dd t dd t| j }|D ]B}| j| }|jrRt ddd n t ddd t d| |jpnd q0| \}}|rt t d
|D ]*}t| j| }t d |d d
| q|rt t ddd t d |D ]*}t| j| }t d |d d
| qdS )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)NameZFile)----r PrT rU ry z%-25sr zMissing modules:?z
imported fromz, z7Submodules that appear to be missing, but could also bez#global names in the parent package:N) rX sortedrL r r? r> any_missing_mayberM r} )rC r keyry missingmayber Zmodsr r r report s0
zModuleFinder.reportc C s | \}}|| 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 )rC r r r r r any_missing s zModuleFinder.any_missingc C s g }g }| j D ]}|| jv r q|d}|dk r<|| q||d d }|d| }| j|}|dur|| j | v r|| q||jv rq|jr|| q|| q|| q| | ||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) rM rP r~ r rL r rA rB sort)rC r r r r] r ZpkgnameZpkgr r r r ( s0
z ModuleFinder.any_missing_maybec C s t j|j }}| jD ]*\}}||r||t|d } qDq| jr|| jvr||krr| dd||f n| dd|f | j
| t |j }t
t|D ](}t|| t|r| || ||< q|jt||dS )Nr z co_filename %r changed to %rz co_filename %r remains unchanged)r co_filename)r) r normpathr rQ
startswithr rN rR rd r listr rW r, r r replacetuple)rC r Z new_filenameZoriginal_filenamer rr r] r r r r V s&
z"ModuleFinder.replace_paths_in_code)Nr NN)NNr )r )r )r )N) r= rH rI rD r_ rc rd re rm rn rp rq rr rs r r rg r r r r r r r r r r r r r r r rJ z s2
#
1
".rJ c
C s dd l } z | tjdd d\}}W n0 | jyV } zt| W Y d }~d S d }~0 0 d}d}g }g }|D ]Z\}} |dkr|d }|dkrd}|dkr|| tj }|dkrd}|dkrl|| ql|sd }
n|d }
tj d d } tj
|
| d< || } |dkr,td
| D ]} td t | qt | ||}
|dd D ]`}|dkrZd}qD|r|d d d
kr|
|d d d dg n
|
| n
|
| qD|
|
|
|
S )Nr r zdmp:qx:z-dz-mz-pz-qz-xzhello.pyzpath:rS z.*r )getoptrK argverrorrX rk r) pathsepr r r* rY rJ rn rm re r )r Zoptsr\ r_ rN ZdomodsZaddpathZexcludeoaZscriptr itemmfr^ r r r testn sX
r rf z
[interrupted])N)#__doc__r Z importlib._bootstrap_externalr importlib.machineryr r) r4 rK Zopmapr r r r r r r3 r. r2 r0 r+ r$ r& r
r r r r; r< rJ r r= r KeyboardInterruptrX r r r r sH
- w;