Your IP : 216.73.217.106
B
A!p\# � @ s| d Z ddlZddlmZmZ ddlmZ G dd� d�ZG dd� de�ZG d d
� d
e�Z G d d � d ee�Z
G d
d� de
�Z dS )zdiagram objects
� N)� is_interface� FilterMixIn)�decorated_with_propertyc @ s e Zd ZdZdS )�Figurez base class for counter handlingN)�__name__�
__module__� __qualname__�__doc__� r
r
�J/opt/alt/python37/lib/python3.7/site-packages/pylint/pyreverse/diagrams.pyr s r c @ s e Zd ZdZddd�ZdS )� Relationshipz=a relation ship from an object in the diagram to another
Nc C s& t �| � || _|| _|| _|| _d S )N)r �__init__� from_object� to_object�type�name)�selfr r �
relation_typer r
r
r r
s
zRelationship.__init__)N)r r r r r
r
r
r
r r s r c @ s e Zd ZdZddd�ZdS )�
DiagramEntityzAa diagram object, i.e. a label associated to an astroid node
�No nameNc C s t �| � || _|| _d S )N)r r
�title�node)r r r r
r
r r
' s
zDiagramEntity.__init__)r N)r r r r r
r
r
r
r r # s r c @ s� e Zd ZdZdZdd� Zdd� Zd dd �Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � ZdS )!� ClassDiagramz main class diagram handling
�classc C s8 t �| |� t�| � || _g | _i | _i | _g | _d S )N)r r
r r �objects�
relationships�_nodes�depends)r r �moder
r
r r
3 s
zClassDiagram.__init__c C s t | j�|d�dd� d�S )Nr
c S s | j j| jjfS )N)r Zfig_idr )�xr
r
r �<lambda>@ � z0ClassDiagram.get_relationships.<locals>.<lambda>)�key)�sortedr �get)r Zroler
r
r �get_relationships< s z ClassDiagram.get_relationshipsNc C s& t ||||�}| j�|g ��|� dS )z create a relation ship
N)r r �
setdefault�append)r r r r r �relr
r
r �add_relationshipC s z ClassDiagram.add_relationshipc C s2 x$| j �|d�D ]}|j|kr|S qW t|��dS )z'return a relation ship or None
r
N)r r$ r �KeyError)r r r r( r
r
r �get_relationshipI s
z ClassDiagram.get_relationshipc C s� g }dd� |� � D �}xbt|j� � �t|j� � � | D ]>\}}| �|�sLq8| �|�}|rld|d�|�f }|�|� q8W t|�S )z3return visible attributes, possibly with class namec S s, g | ]$\}}t |tj�rt|�r||f�qS r
)�
isinstance�astroid� FunctionDefr )�.0�n�mr
r
r �
<listcomp>U s z*ClassDiagram.get_attrs.<locals>.<listcomp>z%s : %sz, ) �items�list�instance_attrs_type� locals_type� show_attr� class_names�joinr' r# )r r �attrsZ
properties� node_nameZassociated_nodes�namesr
r
r � get_attrsQ s (
zClassDiagram.get_attrsc s& � fdd�|� � D �}t|dd� d�S )zreturn visible methodsc s0 g | ](}t |tj�rt|�s� �|j�r|�qS r
)r, r- r. r r7 r )r/ r1 )r r
r r2 i s z,ClassDiagram.get_methods.<locals>.<listcomp>c S s | j S )N)r )r0 r
r
r r o r! z*ClassDiagram.get_methods.<locals>.<lambda>)r" )�valuesr# )r r �methodsr
)r r � get_methodsf s
zClassDiagram.get_methodsc C s2 || j kst�t||�}|| j |< | j�|� dS )z create a diagram object
N)r �AssertionErrorr r r' )r r r �entr
r
r �
add_objectq s
zClassDiagram.add_objectc C sb g }xX|D ]P}t |tj�r |j}t |tj�r
t|d�r
| �|�s
|j|kr
|j}|�|� q
W |S )z'return class names if needed in diagramr ) r, r- �Instance�_proxied�ClassDef�hasattr�has_noder r' )r �nodesr< r r; r
r
r r8 y s
zClassDiagram.class_namesc C s
| j �� S )z,return the list of underlying nodes
)r �keys)r r
r
r rI � s zClassDiagram.nodesc C s
|| j kS )zAreturn true if the given node is included in the diagram
)r )r r r
r
r rH � s zClassDiagram.has_nodec C s
| j | S )z1return the diagram object mapped to node
)r )r r r
r
r �object_from_node� s z ClassDiagram.object_from_nodec C s dd� | j D �S )z%return all class nodes in the diagramc S s g | ]}t |jtj�r|�qS r
)r, r r- rF )r/ �or
r
r r2 � s z(ClassDiagram.classes.<locals>.<listcomp>)r )r r
r
r �classes� s zClassDiagram.classesc C s. x | � � D ]}|jj|kr
|S q
W t|��dS )z@return a class by its name, raise KeyError if not found
N)rM r r r* )r r �klassr
r
r �classe� s zClassDiagram.classec C st �xl| � � D �]^}|j}| �|�|_| �|�|_t|�r@d|_nd|_xH|jdd�D ]8}y | � |�}| �
||d� W qT t k
r� wTY qTX qTW xB|j D ]8}y | � |�}| �
||d� W q� t k
r� w�Y q�X q�W x�t
|j�� �t
|j�� � D ]v\}}xl|D ]d} | tjk�r�q t| tj��r(| j} y | � | �}
| �
|
|d|� W n t k
�r` �w Y nX �q W q�W q W dS ) z<extract relation ships between nodes in the diagram
Z interfacer F)ZrecursZspecialization�
implementsZ associationN)rM r r= r: r@ r? r �shapeZ ancestorsrK r) r* rP r4 r5 r3 r6 r- Z Uninferabler, rD rE ) r �objr Zpar_nodeZpar_objZ impl_nodeZimpl_objr r> �valueZassociated_objr
r
r �extract_relationships� s>
z"ClassDiagram.extract_relationships)N)r r r r �TYPEr
r% r) r+ r= r@ rC r8 rI rH rK rM rO rT r
r
r
r r - s
r c @ s< e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
S )�PackageDiagramz package diagram handling
�packagec C s dd� | j D �S )z&return all module nodes in the diagramc S s g | ]}t |jtj�r|�qS r
)r, r r- �Module)r/ rL r
r
r r2 � s z*PackageDiagram.modules.<locals>.<listcomp>)r )r r
r
r �modules� s zPackageDiagram.modulesc C s. x | � � D ]}|jj|kr
|S q
W t|��dS )zAreturn a module by its name, raise KeyError if not found
N)rY r r r* )r r �modr
r
r �module� s zPackageDiagram.modulec C sp xb| � � D ]V}|jj}||kr"|S |�� j}|d||f kr@|S |d|�dd�d |f kr
|S q
W t|��dS )zlreturn a module by its name, looking also for relative imports;
raise KeyError if not found
z%s.%s�.� r N)rY r r �root�rsplitr* )r r r rZ �mod_namerW r
r
r �
get_module� s
zPackageDiagram.get_modulec C s2 |� � j}| �|�}||jjkr.|jj�|� dS )z1add dependencies created by from-imports
N)r^ r r[ r r r' )r r Z from_moduler` rR r
r
r �add_from_depend� s
z PackageDiagram.add_from_dependc
C s� t �| � xJ| �� D ]>}y"| �|j�� �}| �||d� W q tk
rP wY qX qW x`| �� D ]T}d|_ xH|jj
D ]<}y| � ||j�}W n tk
r� wtY nX | �||d� qtW q`W dS )z<extract relation ships between nodes in the diagram
Z ownershiprW r N) r rT rM rK r r^ r) r* rY rQ r ra )r rR rZ Zdep_name�depr
r
r rT � s
z$PackageDiagram.extract_relationshipsN)
r r r r rU rY r[ ra rb rT r
r
r
r rV � s rV ) r r- Zpylint.pyreverse.utilsr r Zpylint.checkers.utilsr r r r r rV r
r
r
r �<module> s
"