Your IP : 216.73.217.88
B
A!p\ � @ sZ d Z ddlmZ ddlmZ ddlmZ G dd� d�ZG dd� de�ZG d d
� d
e�Z d S ) z+Utilities for creating VCG and Dot diagrams� )� is_exception)�
VCGPrinter)�
DotBackendc @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�
DiagramWriterz,base class for writing project diagrams
c C s$ || _ |\| _| _| _| _d | _d S )N)�config� pkg_edges� inh_edges� imp_edges�association_edges�printer)�selfr �styles� r �H/opt/alt/python37/lib/python3.7/site-packages/pylint/pyreverse/writer.py�__init__ s zDiagramWriter.__init__c C sh xb|D ]Z}|j �� �dd�}d|| jjf }| �||� |jdkrN| �|� n
| �|� | � � qW dS )z9write files for <project> according to <diadefs>
� �_z%s.%s�classN)
�title�strip�replacer Z
output_format� set_printerZTYPE�
write_classes�write_packages� close_graph)r Zdiadefs�diagram�basename� file_namer r r �write s
zDiagramWriter.writec C sx xBt t|�� dd� d��D ]&\}}| jj|| �|�dd� ||_qW x.|�d�D ] }| jj|j j|j
jf| j � qPW dS )zwrite a package diagramc S s | j S )N)r )�xr r r �<lambda>. � z.DiagramWriter.write_packages.<locals>.<lambda>)�key�box)�label�shape�dependsN) � enumerate�sorted�modulesr � emit_node� get_title�fig_id�get_relationships� emit_edge� from_object� to_objectr )r r �i�obj�relr r r r + s "
z DiagramWriter.write_packagesc C s� x>t t|jdd� d��D ]$\}}| jj|f| �|�� ||_qW x.|�d�D ] }| jj|j j|j
jf| j � qLW x.|�d�D ] }| jj|j j|j
jf| j � q|W x8|�d�D ]*}| jj|j j|j
jfd|j
i| j�� q�W dS ) zwrite a class diagramc S s | j S )N)r )r r r r r : r! z-DiagramWriter.write_classes.<locals>.<lambda>)r" ZspecializationZ
implementsZ associationr$ N)r' r( �objectsr r* �
get_valuesr, r- r. r/ r0 r r �namer
)r r r1 r2 r3 r r r r 7 s
zDiagramWriter.write_classesc C s t �dS )z set printerN)�NotImplementedError)r r r r r r r P s zDiagramWriter.set_printerc C s t �dS )zget project titleN)r7 )r r2 r r r r+ T s zDiagramWriter.get_titlec C s t �dS )z get label and shape for classes.N)r7 )r r2 r r r r5 X s zDiagramWriter.get_valuesc C s t �dS )zfinalize the graphN)r7 )r r r r r \ s zDiagramWriter.close_graphN) �__name__�
__module__� __qualname__�__doc__r r r r r r+ r5 r r r r r r s
r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� DotWriterz=write dot graphs from a diagram definition and a project
c C sD t ddd�t ddd�t dddd�t ddd d
d �g}t�| ||� d S ) N�none�open)� arrowtail� arrowhead�empty�nodeZdashed)r? r@ �style�greenZdiamond�solid)� fontcolorr? r@ rC )�dictr r )r r r
r r r r e s
zDotWriter.__init__c C s" t dd�}t||d�| _|| _dS )z9initialize DotWriter and add options for layout.
ZBT)Zrankdir)Zadditional_paramN)rG r r r )r r r Zlayoutr r r r p s
zDotWriter.set_printerc C s |j S )zget project title)r )r r2 r r r r+ w s zDotWriter.get_titlec C s~ |j }|jdkrd| }| jjsZd|d�|j�f }x |jD ]}d||jf }q<W d| }t|j �rrt
d|dd �S t
|dd
�S ) z`get label and shape for classes.
The label contains all attributes and methods
� interfaceu «interface»\n%sz%s|%s\l|z\lz%s%s()\lz{%s}Zred�record)rF r$ r% )r$ r% ) r r% r �only_classnames�join�attrs�methodsr6 r rB rG )r r2 r$ �funcr r r r5 { s
zDotWriter.get_valuesc C s | j �| j� dS )z$print the dot graph into <file_name>N)r Zgenerater )r r r r r � s zDotWriter.close_graphN) r8 r9 r: r; r r r+ r5 r r r r r r< a s r< c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� VCGWriterz=write vcg graphs from a diagram definition and a project
c C sH t dddd�t dddd�t ddddd�t dddd �g}t�| ||� d S )
NrE r= r )�
arrowstyle�backarrowstyle�
backarrowsize�
Zdotted)rP rQ Z linestylerR rD )rP rQ Z textcolor)rG r r )r r r
r r r r � s zVCGWriter.__init__c C sJ t |d�| _t| j�| _| jj|ddddd� | jj| j_| jj| j_dS )z$initialize VCGWriter for a UML graphzw+Zdfs�yes�no)r ZlayoutalgorithmZlate_edge_labelsZ port_sharingZmanhattan_edgesN) r> �
graph_filer r Z
open_graphrB r* Zedger. )r r r r r r r � s zVCGWriter.set_printerc C s
d|j S )z get project title in vcg formatz\fb%s\fn)r )r r2 r r r r+ � s zVCGWriter.get_titlec
C s� t |j�rd|j }n
d|j }|jdkr0d}nd}| jjs�|j}dd� |jD �}tdd � |jg| | D ��}d
|d }d ||f }x|D ]}d
||f }q�W |r�d ||f }x|D ]} d|| f }q�W t ||d�S )z`get label and shape for classes.
The label contains all attributes and methods
z \fb\f09%s\fnz\fb%s\fnrH Zellipser# c S s g | ]
}|j �qS r )r6 )�.0rN r r r �
<listcomp>� s z(VCGWriter.get_values.<locals>.<listcomp>c s s | ]}t |�V qd S )N)�len)rW r6 r r r � <genexpr>� s z'VCGWriter.get_values.<locals>.<genexpr>r � z%s\n\f%sz
%s\n\f08%sz %s\n\f10%s())r$ r% )
r rB r r% r rJ rL rM �maxrG )
r r2 r$ r% rL rM �maxlen�line�attrrN r r r r5 � s&
zVCGWriter.get_valuesc C s | j �� | j�� dS )zclose graph and fileN)r r rV �close)r r r r r � s
zVCGWriter.close_graphN) r8 r9 r: r; r r r+ r5 r r r r r rO � s rO N)
r; Zpylint.pyreverse.utilsr Zpylint.pyreverse.vcgutilsr Z pylint.graphr r r< rO r r r r �<module> s L0