OwlCyberSecurity - MANAGER
Edit File: modulefinder.cpython-310.pyc
o �?OgQ_ � @ 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� Zddd�ZG dd� d�Z G dd� d�Z dd� Z e dkr�ze � 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� setdefault�append)Z packagename�path� r �3/opt/alt/python310/lib64/python3.10/modulefinder.py�AddPackagePath& s r c C s |t | < d S r )�replacePackageMap)Zoldname�newnamer 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 jj�rZt}n t|jt jj�ret}nt|jt jj�rpt}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� PathFinder�invalidate_caches� find_spec� ImportError�format�loader�BuiltinImporter� _C_BUILTIN�FrozenImporter� _PY_FROZEN�origin� is_package�osr �dirname�_PKG_DIRECTORY� isinstance�SourceFileLoader� _PY_SOURCE�ExtensionFileLoader� _C_EXTENSION�SourcelessFileLoader� _PY_COMPILED� _SEARCH_ERROR�io� open_code�splitext)r r �specZ file_path�kind�file�suffixr r r � _find_module5 s* r; c @ s e Zd Zddd�Zdd� ZdS )�ModuleNc C s( || _ || _|| _d | _i | _i | _d S r )�__name__�__file__�__path__�__code__� globalnames� starimports)�selfr r9 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= r> 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 Zd6dd�Zdd� Zdd� Zd d � Zd d � Zd d� Zd7dd�Z d8dd�Z dd� Z dd� Z d9dd�Z 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� badmodules�debug�indent�excludes� replace_paths�processed_paths)rC r rN rP rQ r r r rD | s zModuleFinder.__init__c G sZ || j kr+t| j�D ]}tddd� q t|dd� |D ] }tt|�dd� qt� d S d S )N� � ��end)rN �rangerO �print�repr)rC �level�str�args�i�argr r r �msg� s �zModuleFinder.msgc G s0 |d }|| j kr| jd | _| j|� d S d S �Nr r �rN rO r_ �rC r\ rZ r r r �msgin� � �zModuleFinder.msginc G s0 |d }|| j kr| jd | _| j|� d S d S r` ra rb r r r �msgout� rd zModuleFinder.msgoutc C sX | � dd|� t�|��}ddtf}| �d|||� W d � d S 1 s%w Y d S )Nr � run_scriptr r �__main__)r_ r4 r5 r. � load_module)rC �pathname�fp�stuffr r r rf � 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 �splitr6 r4 r5 r. rh )rC ri �dirr Zextrj rk r r r � load_file� s "�zModuleFinder.load_filer c C s\ | � dd||||� | j||d�}| �||�\}}| �||�}|s#|S |jr,| �||� d S )Nr � import_hook�rZ )r_ �determine_parent�find_head_package� load_tailr? �ensure_fromlist) rC r �caller�fromlistrZ �parent�q�tail�mr r r ro � s zModuleFinder.import_hookc C sJ | � dd||� |r|dkr| �dd� d S |j}|dkrb|jr$|d8 }|dkr<| j| }||u s3J �| �dd|� |S |�d�|k rGtd��d�|�d�d | � �}| j| }| �dd|� |S |jry| j| }||u spJ �| �dd|� |S d|v r�|� d�}|d |� }| j| }|j|ks�J �| �dd|� |S | �dd� d S ) N� rq r zdetermine_parent -> Noner zdetermine_parent ->�.z relative importpath too deep) rc re r= r? rL �countr �joinrl �rfind)rC ru rZ Zpnamerw r] r r r rq � 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{ rr r| r r �%s.%szfind_head_package ->�"raise ImportError: No module named�No module named )rc �findr= � import_modulere r )rC rw r r] �headry Zqnamerx r r r rr � 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{ rs r| r r r� r� r� z load_tail ->)rc r� �lenr= r� re r )rC rx ry rz r] r� Zmnamer r r rs � 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{ rt �*r r� r� )r_ �find_all_submodulesrt �hasattrr= r� r )rC rz rv � recursive�sub�all�subnameZsubmodr r r rt 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 directoryrD ) r? r r �EXTENSION_SUFFIXES�SOURCE_SUFFIXES�BYTECODE_SUFFIXESr) �listdir�OSErrorr_ r� �keys) rC rz rL �suffixesrm �namesr �modZsuff�nr 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 |r�t |||� | �dd|� |S )Nr r� zimport_module ->zimport_module -> None) rc rL �KeyErrorre rM r? � find_moduler rh �close�setattr)rC Zpartname�fqnamerw rz rj ri rk 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 rh rj zload_module ->�execzraise ImportError: � )rc r+ � load_packagere r. �compile�readr2 r �_bootstrap_external� _classify_pycr r[ �marshal�loads� memoryview� add_moduler> rQ �replace_paths_in_coder@ � scan_code) rC r� rj ri � file_infor: �mode�typerz �co�data�excr r r rh L s8 �� zModuleFinder.load_modulec C s> || j vr i | j |< |rd| j | |j< d S d| j | d< d S )Nr �-)rM r= )rC r ru 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 )Nrp r z ImportError:z SyntaxError:r| )rM r� ro r r_ r[ � SyntaxError)rC r ru rv 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 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||f�qS r )� EXTENDED_ARG)�.0�_�opr^ r r r � <listcomp>� s �z-ModuleFinder.scan_opcodes.<locals>.<listcomp>�storer r r �absolute_import�relative_import) �co_code�co_names� co_consts�disZ_unpack_opargs� enumerate� STORE_OPSr r ) rC r� �coder� �constsZopargsr] r� ZopargrZ rv r r r � scan_opcodes� s& �0��zModuleFinder.scan_opcodesc C s| |j }| j}||�D ]�\}}|dkr|\}d|j|< q |dkr~|\}}d} |d ur6d|v r/d} dd� |D �}| j|||dd� | r}d } |jrQ| j�|jd | �} | d u r[| j�|�} | d urx|j�| j� |j �| j � | j d u rwd|j |< q d|j |< q |d kr�|\} }}|r�| j|||| d� q | j || d�} | j| jd |dd� q t |��|j D ]} t| t|��r�| �| |� q�d S ) Nr� r r� r r� c S s g | ]}|d kr|�qS )r� r )r� �fr r r r� � s z*ModuleFinder.scan_code.<locals>.<listcomp>rp r| r� )r� r� rA r� r? rL �getr= �updaterB r@ rq � RuntimeErrorr� r, r� r� )rC r� rz r� �scannerZwhatr\ r rv Z have_starZmmrZ rw �cr r r r� � sP � � ��zModuleFinder.scan_codec C s� | � dd||� t�|�}|r|}| �|�}||_|g|_|jt�|g � |_| �d|j�\}}}z| �||||� | � dd|� |W |rJ|� � S S |rR|� � w w )Nr r� rD zload_package ->) rc r r� r� r> r? r r� rh re r� )rC r� ri r rz rj Zbufrk r r r r� � s$ � �zModuleFinder.load_packagec C s* || j v r | j | S t|� | j |< }|S r )rL r< )rC r� rz r r r r� � s zModuleFinder.add_modulec C sn |d ur |j d | }n|}|| jv r | �dd|� t|��|d u r2|tjv r/d d ddtffS | j}t||�S )Nr| r zfind_module -> Excludedr ) r= rP re r rK �builtin_module_namesr$ r r; )rC r r rw r� r r r r� � s zModuleFinder.find_modulec C s t � t dd � t dd � t| j�� �}|D ]!}| j| }|jr)t ddd� nt ddd� t d| |jp7d � q| �� \}}|rat � t d � |D ]}t| j| �� �}t d |d d �|�� qK|r�t � t ddd� t d� |D ]}t| j| �� �}t d |d d �|�� qrdS dS )z�Print 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 rz 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� �keyrz �missing�mayber Zmodsr r r �report� s4 �zModuleFinder.reportc C s | � � \}}|| S )z�Return 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 ]V}|| jv rq|�d�}|dk r |�|� q||d d� }|d|� }| j�|�}|durX|| j | v rC|�|� q||jv rIq|jrR|�|� 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 � } nq | jrH|| jvrH||kr9| � dd||f � n | � dd|f � | j� |� t |j �}t t|��D ]}t|| t|��rg| �|| �||< qS|jt|�|d�S )Nr z co_filename %r changed to %rz co_filename %r remains unchanged)r� � co_filename)r) r �normpathr� rQ � startswithr� rN rR re r �listr� rW r, r� r� �replace�tuple)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 ) r= rH rI rD r_ rc re rf rn ro rq rr rs rt r� r� rh r� r� r� r� r� r� r� r� r� r� r� r r r r rJ z s4 # 1 " .rJ c C s� dd l } z| � tjdd � d�\}}W n | jy+ } z t|� W Y d }~d S d }~ww d}d}g }g }|D ]-\}} |dkrB|d }|dkrHd}|dkrT|| �tj� }|dkrZd}|dkrc|�| � q6|sid } n|d } tj d d � } tj � | �| d< || } |dkr�td � | D ] } td t | �� q�t | ||�} |dd � D ]+}|dkr�d}q�|r�|d d � d kr�| � |d d � d dg� q�| � |� q�| �|� q�| �| � | �� | S )Nr r zdmp:qx:z-dz-mz-pz-qz-xzhello.pyzpath:rS ���z.*r� )�getoptrK �argv�errorrX rl r) �pathsepr r r* rY rJ ro rn rf r� )r� Zoptsr\ r_ rN ZdomodsZaddpathZexclude�o�aZscriptr �item�mfr^ r r r �testn s^ �� � r� rg z [interrupted]r )#�__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 �<module> sN - w; ��