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_portconZruleZ
added_contextZremoved_contextc @ s8 e Zd ZdZed�Zed�Zed�Zdd� Zdd� Z dS )�PortconsDifferencez:Determine the difference in portcons between two policies.�
diff_portconsc 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 portcons between the policies.zGGenerating portcon differences from {0.left_policy} to {0.right_policy}c s s | ]}t |�V qd S )N)�PortconWrapper)�.0�n� r
� /usr/lib64/python3.6/portcon.py� <genexpr>. s z3PortconsDifference.diff_portcons.<locals>.<genexpr>c s s | ]}t |�V qd S )N)r
)r r r
r
r r / s N)�log�info�formatZ _set_diffZ left_policyZportconsZ right_policy�added_portcons�removed_portcons�modified_portconsr �context�append�modified_portcon_record)�selfZmatched_portconsZ left_portconZ
right_portconr
r
r r ' s
z PortconsDifference.diff_portconsc C s" | j jd� d| _d| _d| _dS )z%Reset diff results on policy changes.z Resetting portcon differencesN)r �debugr r r )r r
r
r � _reset_diff> s z PortconsDifference._reset_diffN)
�__name__�
__module__� __qualname__�__doc__r r r r r r r
r
r
r r s r c @ s4 e Zd ZdZdZdd� Zdd� Zd d
� Zd d � Zd
S )r
z*Wrap portcon statements for diff purposes.�protocol�low�highc C s* || _ |j| _|j\| _| _t|�| _d S )N)�originr Zportsr! r" �hash�key)r Zoconr
r
r �__init__L s zPortconWrapper.__init__c C s | j S )N)r% )r r
r
r �__hash__R s zPortconWrapper.__hash__c C s | j |j k S )N)r# )r �otherr
r
r �__lt__U s zPortconWrapper.__lt__c C s$ | j |j ko"| j|jko"| j|jkS )N)r r! r" )r r( r
r
r �__eq__X s zPortconWrapper.__eq__N)r r! r" ) r r r r � __slots__r& r' r) r* r
r
r
r r
F s r
N) � collectionsr r r Z descriptorsr �
differencer r r r r
r
r
r
r �<module> s '