3
�^� � @ sd d dl Z d dlmZmZ d dlmZ edddg�ZG dd� d�ZG d d
� d
e�ZG d d � d e�Z dS )
� N)�ABC�abstractmethod)�
namedtupleZ
modified_item�left�rightc @ sn e Zd ZdZdd� Zedd� �Zejdd� �Zedd� �Zejd d� �Zd
d � Z e
d d
� �Z e
ddd��Z dS )�
Differencez&Base class for all policy differences.c C s t jt�| _|| _|| _d S )N)�loggingZ getLogger�__name__�log� left_policy� right_policy)�selfr r � r �"/usr/lib64/python3.6/difference.py�__init__ s zDifference.__init__c C s | j S )N)� _left_policy)r
r r r r ' s zDifference.left_policyc C s$ | j jdj|�� || _| j� d S )Nz"Policy diff left policy set to {0})r
�info�formatr � _reset_diff)r
�policyr r r r + s c C s | j S )N)�
_right_policy)r
r r r r 1 s zDifference.right_policyc C s$ | j jdj|�� || _| j� d S )Nz#Policy diff right policy set to {0})r
r r r r )r
r r r r r 5 s c C s t �dS )z%Reset diff results on policy changes.N)�NotImplementedError)r
r r r r > s zDifference._reset_diffc c s. x(| D ] }x|j � D ]}||�V qW qW dS )z4Generator that yields a wrapped, expanded rule list.N)�expand)Z rule_list�WrapperZunexpanded_ruleZ
expanded_ruler r r �_expand_generatorB s
z Difference._expand_generatorNTc C s� t | �}t |�}|| }|| }t � }t|| |d�} t|| |d�}
t| �t|
�ksntdjt| �t|
����x<t| |
�D ].\} }| |ks�tdj| |���|j| |f� qzW |r�t dd� |D ��t dd� |D ��t dd� |D ��fS |||fS dS )a�
Standard diff of two sets.
Parameters:
left An iterable
right An iterable
Return:
tuple (added, removed, matched)
added Set of items in right but not left
removed Set of items in left but not right
matched Set of items in both left and right. This is
in the form of tuples with the matching item
from left and right
)�keyzDMatched items assertion failure (this is an SETools bug), {0} != {1}c s s | ] }|j V qd S )N)�origin)�.0�ir r r � <genexpr> s z'Difference._set_diff.<locals>.<genexpr>c s s | ] }|j V qd S )N)r )r r r r r r � s c s s | ]\}}|j |j fV qd S )N)r )r r r r r r r � s N)�set�sorted�len�AssertionErrorr �zip�add) r r r ZunwrapZ
left_itemsZ right_itemsZ added_itemsZ
removed_itemsZ
matched_itemsZleft_matched_itemsZright_matched_itemsr r r � _set_diffL s(
zDifference._set_diff)NT)
r �
__module__� __qualname__�__doc__r �propertyr �setterr r � staticmethodr r& r r r r r s
r c @ sH e Zd ZdZdZdd� Zedd� �Zedd � �Zed
d � �Z d d
� Z
dS )r z/Abstract base class for policy object wrappers.r r c C s dj | t| j��S )Nz&<{0.__class__.__name__}(Wrapping {1})>)r �reprr )r
r r r �__repr__� s zWrapper.__repr__c C s d S )Nr )r
r r r �__hash__� s zWrapper.__hash__c C s d S )Nr )r
�otherr r r �__eq__� s zWrapper.__eq__c C s d S )Nr )r
r0 r r r �__lt__� s zWrapper.__lt__c C s
| |k S )Nr )r
r0 r r r �__ne__� s zWrapper.__ne__N)r r ) r r' r( r) � __slots__r. r r/ r1 r2 r3 r r r r r � s r c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) �
SymbolWrapperz�
General wrapper for policy symbols, e.g. types, roles
to provide a diff-specific equality operation based
on its name.
�namec C s || _ t|�| _t| j�| _d S )N)r �strr6 �hashr )r
Zsymbolr r r r � s
zSymbolWrapper.__init__c C s | j S )N)r )r
r r r r/ � s zSymbolWrapper.__hash__c C s | j |j k S )N)r6 )r
r0 r r r r2 � s zSymbolWrapper.__lt__c C s | j |j kS )N)r6 )r
r0 r r r r1 � s zSymbolWrapper.__eq__N) r r' r( r) r4 r r/ r2 r1 r r r r r5 � s r5 )
r �abcr r � collectionsr Zmodified_item_recordr r r5 r r r r �<module> s k