3
�^� � @ sh d dl mZ ddlmZ ddlmZ ddlmZmZ edddd g�Z G d
d � d e�Z
G d d
� d
e�Z dS )� )�
namedtuple� )�ContextWrapper)�DiffResultDescriptor)�
Difference�WrapperZmodified_nodeconZruleZ
added_contextZremoved_contextc @ s8 e Zd ZdZed�Zed�Zed�Zdd� Zdd� Z dS )�NodeconsDifferencez:Determine the difference in nodecons between two policies.�
diff_nodeconsc C s� | j jdj| �� | jdd� | jj� D �dd� | jj� D ��\| _| _}g | _ x<|D ]4\}}t
|j �t
|j �krT| j j t
||j |j �� qTW dS )z9Generate the difference in nodecons between the policies.zGGenerating nodecon differences from {0.left_policy} to {0.right_policy}c s s | ]}t |�V qd S )N)�NodeconWrapper)�.0�n� r
� /usr/lib64/python3.6/nodecon.py� <genexpr>/ s z3NodeconsDifference.diff_nodecons.<locals>.<genexpr>c s s | ]}t |�V qd S )N)r
)r r r
r
r r 0 s N)�log�info�formatZ _set_diffZ left_policyZnodeconsZ right_policy�added_nodecons�removed_nodecons�modified_nodeconsr �context�append�modified_nodecon_record)�selfZmatched_nodeconsZ left_nodeconZ
right_nodeconr
r
r r ( s
z NodeconsDifference.diff_nodeconsc C s" | j jd� d| _d| _d| _dS )z%Reset diff results on policy changes.z Resetting nodecon differencesN)r �debugr r r )r r
r
r � _reset_diff? s z NodeconsDifference._reset_diffN)
�__name__�
__module__� __qualname__�__doc__r r r r r r r
r
r
r r s r c @ s4 e Zd ZdZd
Zdd� Zdd� Zdd � Zd
d � Zd S )r
z*Wrap nodecon statements for diff purposes.�
ip_version�networkc C s$ || _ |j| _|j| _t|�| _d S )N)�originr r! �hash�key)r Zoconr
r
r �__init__M s zNodeconWrapper.__init__c C s | j S )N)r$ )r r
r
r �__hash__S s zNodeconWrapper.__hash__c C s | j |j k S )N)r" )r �otherr
r
r �__lt__V s zNodeconWrapper.__lt__c C s | j |j ko| j|jkS )N)r r! )r r' r
r
r �__eq__Y s zNodeconWrapper.__eq__N)r r! ) r r r r � __slots__r% r& r( r) r
r
r
r r
G s r
N) � collectionsr r r Z descriptorsr �
differencer r r r r
r
r
r
r �<module> s '