3
Kl�f� � @ sp d Z ddlmZ ddlmZ ejr.ddlmZ G dd� d�Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dS )ai
Classes and functions for the output of reference policy modules.
This module takes a refpolicy.Module object and formats it for
output using the ModuleWriter object. By separating the output
in this way the other parts of Madison can focus solely on
generating policy. This keeps the semantic / syntactic issues
cleanly separated from the formatting issues.
� )� refpolicy)�util)�cmpc @ s e Zd Zdd� Zdd� ZdS )� ModuleWriterc C s d | _ d | _d| _d| _d S )NT)�fd�module�sort�requires)�self� r � /usr/lib/python3.6/output.py�__init__& s zModuleWriter.__init__c C sJ || _ | jrt| j � x.tj| j dd�D ]\}}|jdt|� � q(W d S )NT)Z showdepthz%s
)r r � sort_filterr Zwalktree�write�str)r
r r �node�depthr r r r , s
zModuleWriter.writeN)�__name__�
__module__� __qualname__r
r r r r r r % s r c C s� t j| �}|j� t j|�}|j� t|�t|�krFt|d |d �S x4t||�D ]&}|d |d krRt|d |d �S qRW dS )N� r )r Z set_to_listr �lenr �zip)�x�yZxlZyl�vr r r �
id_set_cmp= s
r c C sd t | j|j�}|dkr|S t | j|j�}|dkr4|S t | j|j�}|dkrN|S tt| j�t|j��S )Nr )r � src_typesZ tgt_typesZ obj_classesr r Zperms)�a�b�retr r r �
avrule_cmpK s r! c C s8 | j d |j d kr*t| j d |j d �S t| j|j�S )Nr )�argsr Zifname)r r r r r �
ifcall_cmpZ s r# c C sf t | tj�r8t |tj�r"t| |�S t| jd g|j�S n*t |tj�rNt| |�S t| j|jd g�S d S )Nr ) �
isinstancer �
InterfaceCallr# r r" r ZAVRuler! )r r r r r �rule_cmp` s
r& c C s t | j|j�S )N)r Zrole)r r r r r �
role_type_cmpl s r' c C s& dd� }x| j � D ] }||� qW dS )z/Sort and group the output for readability.
c S s� g }x(| j � D ] }|j|� |jtj� � qW x| j� D ]}|j|� q8W |jtj� � g }|j| j� � |j| j� � |jt j
t �d� d }g }x||D ]t}t |tj
�r�|jd }n t j|j�}||k�r|r�|jtj� � |}tj� } | jjd| � |j| � |j|� q�W |j|� g }
|
j| j� � |
jt j
t�d� t|
��rftj� } | jjd� |j| � |j|
� x$| jD ]} | |k�rx|j| � �qxW || _d S )N)�keyr z ============= %s ==============z"============= ROLES ==============)Zmodule_declarations�appendr �Commentr �extendZavrulesZinterface_callsr r �
cmp_to_keyr& r$ r% r" �firstr �linesZ
role_typesr' r Zchildren) r �c�modZrequireZrulesZcurZ sep_rulesZruler ZcommentZrasZchildr r r � sort_noder sL
z sort_filter.<locals>.sort_nodeN)Znodes)r r1 r r r r r o s <r N)
�__doc__� r r ZPY3r r r r! r# r& r' r r r r r �<module> s