U
e5d�; � @ s� d dl Z d dlZd dlZd dlZd dlmZmZmZ ddddgZ i Z
G dd� d�Z G d d� de �Z G d
d� de �Z
d d � Zd d
d�Zd dd�Zd dd�Zd dd�Zdd� Zdd� Zdd� Zdd� Zedkr�e� dS ) � N)�NAME�DEDENT�OP�
readmodule�
readmodule_ex�Class�Functionc @ s e Zd Zdd� Zdd� ZdS )�_Objectc C s( || _ || _|| _|| _|| _i | _d S �N)�module�name�file�lineno�parent�children��selfr r r
r r � r � /usr/lib64/python3.8/pyclbr.py�__init__6 s z_Object.__init__c C s || j |< d S r
)r )r r �objr r r � _addchild> s z_Object._addchildN)�__name__�
__module__� __qualname__r r r r r r r 4 s r c @ s e Zd Zddd�ZdS )r Nc C s t �| |||||� d S r
)r r r r r r r D s zFunction.__init__)N)r r r r r r r r r B s c s& e Zd Zd� fdd� Zdd� Z� ZS )r Nc s0 t �| |||||� |d kr g n|| _i | _d S r
)r r �super�methods)r r r r r
r r �� __class__r r r J s zClass.__init__c C s || j |< d S r
)r )r r r r r r �
_addmethodO s zClass._addmethod)N)r r r r r �
__classcell__r r r r r H s c C s: t | j|| j|| �}| �||� t| t�r6| �||� |S r
)r r r
r �
isinstancer r )�ob� func_namer Znewfuncr r r �_nest_functionS s
r$ c C s& t | j||| j|| �}| �||� |S r
)r r r
r )r" �
class_namer r Znewclassr r r � _nest_class[ s r& c C s6 i }t | |pg ��� D ]\}}t|t�r|||< q|S r
)� _readmodule�itemsr! r )r �path�res�key�valuer r r r a s
c C s t | |p
g �S r
)r' )r r) r r r r m s c C s� |d k rd|| f }n| }|t kr*t | S i }| tjkrL|d krL|t | < |S | �d�}|dkr�| d |� }| |d d � }t|||�}|d k r�d||f }d|kr�td�|���t||d |�S d } |d k r�|}
n
|tj }
tj �
||
�} | d k�rt d|��|d��|t |< | j d k �r$| j |d< z| j
�|�} W n ttfk
�rR | Y S X | d k�rb|S | j
�|�}
t|||
| ||�S ) Nz%s.%s�.r � �__path__zNo package named {}zno module named )r )�_modules�sys�builtin_module_names�rfindr' � ImportError�formatr) � importlib�util�_find_spec_from_path�ModuleNotFoundError�submodule_search_locations�loader�
get_source�AttributeError� get_filename� _create_tree)r r) � inpackage�
fullmodule�tree�i�packageZ submoduler �fZ search_path�spec�source�fnamer r r r' v sJ
r' c !
C sH t �|�}g }t�|j�}�z|D �]�\} }
} } }
| tkr`| \}}|r^|d d |kr^|d= qBq"|
dkr�| \}}|r�|d d |kr�|d= qpt|�dd� \} }} | tkr�q"d }|r�|d d }t|||�}nt | |||�}|||< |�
||f� q"|
dk�r�| \}}|�r(|d d |k�r(|d= �qt|�dd� \} }} | tk�rJq"t|�dd� \} }
} d }|
dk�r�g }d}g }t|�dd� \} }
} |
dk�r |dk�r d � |�}||k�r�|| }nL|� d
�}t
|�dk�r|d }|d }|tk�rt| }||k�r|| }|�
|� g }|
dk�r0|d7 }nZ|
d k�rR|d8 }|dk�r��q�n8|
d
k�rh|dk�rhn"| ttfk�rz|dk�rz|�
|
� �qz|}|�r�|d d }t||||�} nt| ||||�} | ||< |�
| |f� q"|
dk�rh| d dk�rht|�} | D ]d\} } zL|d k�r t| |� n2zt| ||� W n tk
�rP t| g � Y nX W n Y nX �q q"|
dkr"| d dkr"t|�\} }
| r"|
dk�r�q"t|�}zt| ||�}W n Y q"Y nX |D ]X\}} ||k�r�|| || �p�|< n0|dk�r�|D ] }|d dk�r�|| ||< �q��q�q"W n tk
�r: Y nX |�� |S )N�����r. �defr � �class�()�)�,� r- �����rN rO �import�from�*�_)�io�StringIO�tokenize�generate_tokens�readliner �nextr r$ r �append�join�split�lenr0 r r& r � _getnamelistr' r4 �_getname�
StopIteration�close)!rA r) rH rG rB r@ rE �stack�g� tokentype�token�startZ_end�_liner Z
thisindentr# Zcur_funcZcur_objr% Zinherit�names�levelr �n�c�m�dZ cur_class�modules�modZ_mod2Zn2r r r r? � s�
� �
r? c C sl g }t | �\}}|sqh|dkr,t | �\}}nd }|�||f� |dkr\d|kr\t| �d }q>|dkrqhq|S )N�asrO �
r. )ra r\ r[ )re rj r rg Zname2r r r r` E s r` c C s� g }t | �dd� \}}|tkr0|dkr0d |fS |�|� t | �dd� \}}|dkrXq�t | �dd� \}}|tkrvq�|�|� q:d�|�|fS )Nr � rT r- )r[ r r\ r] )re �partsrf rg r r r ra [ s
ra c
C sX dd l } ztjd }W n t}Y nX | j�|�rj| j�|�g}| j�|�}|�� � d�rn|d d� }ng }t
||�}dd� }t |� � |dd�}d }|�rT|�
� }t|t�r�q�t|d
�s�d|_t|t�r�t |j� � |dd�}|D ]} |j| | _q�|�|� t|t��r,td �d |j |j|j|j�� q�t|t�r�td
�d |j |j|j�� q�d S )Nr r. z.py�����c S s t | dd�S )Nr r )�getattr)�ar r r �| � z_main..T)r+ �reversert �indentz{}class {} {} {}� z {}def {} {}) �osr1 �argv�__file__r) �exists�dirname�basename�lower�endswithr �sorted�values�popr! �list�hasattrr| r r �extendr �printr5 r r r r )
r~ rq r) rB Z
lineno_keyZobjsZ indent_levelr Znew_objsr" r r r �_mainm sL
�
�
r� �__main__)N)N)N)N)rV r1 �importlib.utilr6 rX rg r r r �__all__r0 r r r r$ r& r r r' r? r` ra r� r r r r r �) s*
@ &