3
Klf @ sp d Z ddlmZ ddlmZ ejr.ddlmZ G dd dZdd 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)fdmodulesortrequires)self r /usr/lib/python3.6/output.py__init__& s zModuleWriter.__init__c C sJ || _ | jrt| j x.tj| j ddD ]\}}|jdt| q(W d S )NT)Z showdepthz%s
)r r sort_filterr Zwalktreewritestr)r
r r nodedepthr 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)xyZxlZylvr 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| jt|jS )Nr )r src_typesZ tgt_typesZ obj_classesr r Zperms)abretr r r
avrule_cmpK s r! c C s8 | j d |j d kr*t| j d |j d S t| j|jS )Nr )argsr Zifname)r r r r r
ifcall_cmpZ s r# c C sf t | tjr8t |tjr"t| |S t| jd g|jS n*t |tjrNt| |S t| j|jd gS d S )Nr )
isinstancer
InterfaceCallr# r r" r ZAVRuler! )r r r r r rule_cmp` s
r& c C s t | j|jS )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}||kr|r|jtj |}tj } | jjd| |j| |j| qW |j| g }
|
j| j |
jt j
td t|
rftj } | jjd |j| |j|
x$| jD ]} | |krx|j| qxW || _d S )N)keyr z ============= %s ==============z"============= ROLES ==============)Zmodule_declarationsappendr Commentr extendZavrulesZinterface_callsr r
cmp_to_keyr& r$ r% r" firstr linesZ
role_typesr' r Zchildren) r cmodZrequireZrulesZcurZ sep_rulesZruler ZcommentZrasZchildr r r sort_noder sL
z sort_filter..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 s