B
�/g^!; � @ 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.7/pyclbr.py�__init__6 s z_Object.__init__c C s || j |< d S )N)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 )N)r r )r r
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 )N)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 )N)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 )N)r r
r r �
isinstancer r )�ob� func_namer
Znewfuncr r r �_nest_functionS s
r! c C s& t | j||| j|| �}| �||� |S )N)r r
r r )r �
class_namer
r Znewclassr r r � _nest_class[ s r# c C s: i }x0t | |pg ��� D ]\}}t|t�r|||< qW |S )N)� _readmodule�itemsr r )r
�pathZres�key�valuer r r r a s
c C s t | |p
g �S )N)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 �
||
�} |t |< | j d k �r| j |d< y | j �
|�} | d k�r$|S W n ttfk
�r@ |S X | j �|�}
t|||
| ||�S )Nz%s.%s�.r � �__path__zNo package named {})�_modules�sys�builtin_module_names�rfindr$ � ImportError�formatr& � importlib�util�_find_spec_from_path�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 sF
r$ c !
C sj t �|�}g }t�|j�}�y*�x"|D �]\} }
} } }
| tkrh| \}}x |rd|d d |krd|d= qHW q&|
dk�r| \}}x |r�|d d |kr�|d= q|W t|�dd� \} }} | tkr�q&d }|r�|d d }t|||�}nt | |||�}|||< |�
||f� q&|
dk�r�| \}}x$|�r8|d d |k�r8|d= �qW t|�dd� \} }} | tk�r\q&t|�dd� \} }
} d }|
dk�r�g }d}g }�xt|�dd� \} }
} |
dk�r2|dk�r2d � |�}||k�r�|| }nL|� d
�}t
|�dk�r$|d }|d }|tk�r$t| }||k�r$|| }|�
|� g }|
dk�rF|d7 }nX|
d k�rf|d8 }|dk�r�P n8|
d
k�r||dk�r|n"| ttfk�r�|dk�r�|�
|
� �q�W |}|�r�|d d }t||||�} nt| ||||�} | ||< |�
| |f� q&|
dk�r�| d dk�r�t|�} xl| D ]d\} } yL|d k�r8t| |� n2yt| ||� W n tk
�rh t| g � Y nX W n Y nX �qW q&|
dkr&| d dkr&t|�\} }
| r&|
dk�r�q&t|�}yt| ||�}W n w&Y nX xd|D ]\\}} ||k�r|| || �p|< n4|dk�r�x(|D ] }|d dk�r|| ||< �qW �q�W q&W W n tk
�r\ Y nX |�� |S )N�����r* �defr � �class�()�)�,� r) �����rI rJ �import�from�*�_)�io�StringIO�tokenize�generate_tokens�readliner �nextr r! r �append�join�split�lenr, r r# r � _getnamelistr$ r0 �_getname�
StopIteration�close)!r< r& rC rB r= r; r@ �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 st g }xjt | �\}}|sP |dkr.t | �\}}nd }|�||f� x |dkr`d|kr`t| �d }qBW |dkrP qW |S )N�asrJ �
r* )r\ rW rV )r` re r rb Zname2r r r r[ B s r[ c C s� g }t | �dd� \}}|tkr0|dkr0d |fS |�|� xJt | �dd� \}}|dkrZP t | �dd� \}}|tkrxP |�|� q<W d�|�|fS )Nr � rO r) )rV r rW rX )r` �partsra rb r r r r\ X s
r\ c
C sb dd l } ytjd }W n t}Y nX | j�|�rj| j�|�g}| j�|�}|�� � d�rn|d d� }ng }t
||�}dd� }t |� � |dd�}d }x�|�r\|�
� }t|t�r�q�t|d
�s�d|_t|t��rt |j� � |dd�}x|D ]} |j| | _q�W |�|� t|t��r4td �d |j |j|j|j�� q�t|t�r�td
�d |j |j|j�� q�W d S )Nr r* z.py�����c S s t | dd�S )Nr
r )�getattr)�ar r r �<lambda>y � z_main.<locals>.<lambda>T)r' �reversero �indentz{}class {} {} {}� z {}def {} {}) �osr- �argv�__file__r& �exists�dirname�basename�lower�endswithr �sorted�values�popr �list�hasattrrw r r �extendr �printr1 r r r
r )
ry rl r& r= Z
lineno_keyZobjsZ indent_levelr Znew_objsr r r r �_mainj s@
r� �__main__)N)N)N)N)rQ r- �importlib.utilr2 rS rb r r r �__all__r, r r r r! r# r r r$ r: r[ r\ r� r r r r r �<module>) s*
= &