o
9jQ_ @ 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 dkrze Z W dS e!y e"d Y dS w 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) packagenamepath r #/usr/lib/python3.10/modulefinder.pyAddPackagePath& s r c C s |t | < d S r )replacePackageMap)oldnamenewnamer r r ReplacePackage1 s r c C s t jj t jj| |}|du r tdj| d| d|jt jju r+ddddt ffS |jt jj
u r:ddddt ffS |j }|j
| rOdtj|ddtffS t|jt jjrZt}n t|jt jjret}nt|jt jjrpt}nddddtffS t|}tj |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_pathkindfilesuffixr r r _find_module5 s*
r> c @ s e Zd ZdddZdd ZdS )ModuleNc C s( || _ || _|| _d | _i | _i | _d S r )__name____file____path____code__ globalnames starimports)selfr r< r r r r __init__d s
zModule.__init__c C sL d| j f }| jd ur|d| jf }| jd ur |d| jf }|d }|S )Nz Module(%rz, %r))r@ rA rB )rF sr r r __repr__q s
zModule.__repr__)NN)r@
__module__ __qualname__rG rJ 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! Zd8d"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 ur |ng | _|d ur%|ng | _g | _d S )Nr ) sysr modules
badmodulesdebugindentexcludes
replace_pathsprocessed_paths)rF r rQ rS rT r r r rG | s
zModuleFinder.__init__c G sZ || j kr+t| jD ]}tddd q
t|dd |D ]
}tt|dd qt d S d S )N end)rQ rangerR printrepr)rF levelstrargsiargr r r msg s
zModuleFinder.msgc G s0 |d }|| j kr| jd | _| j| d S d S Nr r rQ rR rb rF r_ r] r r r msgin
zModuleFinder.msginc G s0 |d }|| j kr| jd | _| j| d S d S rc rd re r r r msgout rg zModuleFinder.msgoutc C sX | dd| t|}ddtf}| d||| W d d S 1 s%w Y d S )Nr
run_scriptr r __main__)rb r6 r7 r0 load_module)rF pathnamefpstuffr r r ri s
"zModuleFinder.run_scriptc C sj t j|\}}t j|\}}t|}|dtf}| |||| W d d S 1 s.w Y d S )Nr )r+ r splitr8 r6 r7 r0 rk )rF rl dirr extrm rn r r r load_file s
"zModuleFinder.load_filer c C s\ | dd|||| | j||d}| ||\}}| ||}|s#|S |jr,| || d S )Nr import_hookr] )rb determine_parentfind_head_package load_tailrB ensure_fromlist) rF r callerfromlistr] parentqtailmr r r rs s zModuleFinder.import_hookc C sJ | dd|| |r|dkr| dd d S |j}|dkrb|jr$|d8 }|dkr<| j| }||u s3J | dd| |S |d|k rGtdd|dd | }| j| }| dd| |S |jry| j| }||u spJ | dd| |S d|v r| d}|d | }| j| }|j|ksJ | dd| |S | dd d S ) N ru r zdetermine_parent -> Noner zdetermine_parent ->.z relative importpath too deep)
rf rh r@ rB rO countr" joinro rfind)rF ry r] pnamer{ r` r r r ru sB
z ModuleFinder.determine_parentc C s | dd|| d|v r |d}|d | }||d d }n|}d}|r.d|j|f }n|}| |||}|rF| dd||f ||fS |rb|}d }| |||}|rb| dd||f ||fS | dd| td | )
Nr rv r r r %s.%szfind_head_package ->"raise ImportError: No module namedNo module named )rf findr@
import_modulerh r" )rF r{ r r` headr} qnamer| r r r rv s.
z ModuleFinder.find_head_packagec C s | dd|| |}|rG|d}|dk rt|}|d | ||d d }}d|j|f }| |||}|sE| dd| td| |s | dd | |S )
Nr rw r r r r r r z load_tail ->)rf r lenr@ r rh r" )rF r| r} r~ r` r mnamer r r rw s
zModuleFinder.load_tailc C s | dd||| |D ]2}|dkr"|s!| |}|r!| ||d q t||s=d|j|f }| |||}|s=td| q d S )Nr rx *r r r )rb find_all_submodulesrx hasattrr@ r r" )rF r~ rz recursivesuballsubnamesubmodr r r rx s
z ModuleFinder.ensure_fromlistc
C s |j sd S i }g }|tjjd d 7 }|tjjd d 7 }|tjjd d 7 }|j D ]F}zt|}W n tyC | dd| Y q*w |D ])}d }|D ]}t
|} || d |krd|d | } nqL|ro|dkro|||< qFq*| S )Nr zcan't list directoryrG ) rB r r EXTENSION_SUFFIXESSOURCE_SUFFIXESBYTECODE_SUFFIXESr+ listdirOSErrorrb r keys)
rF r~ rO suffixesrp namesr modsuffnr r r r s6
z ModuleFinder.find_all_submodulesc C s | dd||| z| j| }W n ty Y n
w | dd| |S || jv r0| dd d S |r?|jd u r?| dd d S z| ||oG|j|\}}}W n ty_ | ddd Y d S w z| ||||}W |rp| n|rx| w w |rt
||| | dd| |S )Nr r zimport_module ->zimport_module -> None) rf rO KeyErrorrh rP rB find_moduler" rk closesetattr)rF partnamefqnamer{ r~ rm rl rn r r r r , sD
zModuleFinder.import_modulec
C s |\}}}| dd||o
d| |tkr#| ||}| dd| |S |tkr0t| |d} n9|tkrgz| }
tj
|
|i W n t yZ } z | ddt | | d } ~ ww t
t|
dd } nd } | |}||_| r| jr{| | } | |_| | | | dd| |S )Nr rk rm zload_module ->execzraise ImportError: )rf r- load_packagerh r0 compilereadr4 r _bootstrap_external
_classify_pycr" r^ marshalloads
memoryview
add_modulerA rT replace_paths_in_coderC scan_code) rF r rm rl file_infor= modetyper~ codataexcr r r rk L s8
zModuleFinder.load_modulec C s> || j vr
i | j |< |rd| j | |j< d S d| j | d< d S )Nr -)rP r@ )rF r ry r r r _add_badmodulei s
zModuleFinder._add_badmodulec C sJ || j v r
| || d S z
| j|||d W nC ty9 } z| ddt| | || W Y d }~d S d }~w tyZ } z| ddt| | || W Y d }~d S d }~ww |r|D ]C}|d | }|| j v rs| || q_z | j|||g|d W q_ ty } z| ddt| | || W Y d }~q_d }~ww d S d S )Nrt r z ImportError:z SyntaxError:r )rP r rs r" rb r^ SyntaxError)rF r ry rz r] rb r fullnamer r r _safe_import_hookq s:
z ModuleFinder._safe_import_hookc c s |j }|j}|j}dd t|D }t|D ]`\}\}}|tv r+d|| ffV q|tkrx|dkrx||d d ||d d krItkrxn q|||d d } |||d d }
| dkrmd|
|| ffV n
d| |
|| ffV qqd S ) Nc S s" g | ]
\}}}|t kr||fqS r ) EXTENDED_ARG).0_opra r r r