3
�^B) � @ s� d dl mZmZ ddlmZ ddlmZmZmZ eddddg�Z ed dddg�Z
ed
d d d
dg�Z ee �Z
ee �Zdd� Zdd� ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd � d e�ZG d d � d e�Zd S ) � )� defaultdict�
namedtuple� )�DiffResultDescriptor)�
Difference�
SymbolWrapper�WrapperZmodified_category�
added_aliases�removed_aliases�matched_aliasesZmodified_sensitivityZmodified_level�level�added_categories�removed_categories�matched_categoriesc C s> yt | j | S tk
r8 t| �}|t | j | < |S X dS )z
Wrap category from the specified policy.
This caches results to prevent duplicate wrapper
objects in memory.
N)� _cats_cache�policy�KeyErrorr )�category�c� r �/usr/lib64/python3.6/mls.py�category_wrapper_factory+ s r c C s> yt | j | S tk
r8 t| �}|t | j | < |S X dS )z�
Wrap sensitivity from the specified policy.
This caches results to prevent duplicate wrapper
objects in memory.
N)� _sens_cacher r r )� sensitivityr r r r �sensitivity_wrapper_factory: s r c @ s8 e Zd ZdZed�Zed�Zed�Zdd� Zdd� Z dS )�CategoriesDifferencez<Determine the difference in categories between two policies.�diff_categoriesc C s� | j jdj| �� | jdd� | jj� D �dd� | jj� D ��\| _| _}t � | _
xH|D ]@\}}| j|j � |j � dd�\}}}|s�|rVt |||�| j
|< qVW dS )z;Generate the difference in categories between the policies.zHGenerating category differences from {0.left_policy} to {0.right_policy}c s s | ]}t |�V qd S )N)r )�.0r r r r � <genexpr>X s z7CategoriesDifference.diff_categories.<locals>.<genexpr>c s s | ]}t |�V qd S )N)r )r r r r r r Y s F)�unwrapN)
�log�info�format� _set_diff� left_policy�
categories� right_policyr
r �dict�modified_categories�aliases�modified_cat_record)�selfr Z
left_categoryZright_categoryr r
r r r r r Q s z$CategoriesDifference.diff_categoriesc C s" | j jd� d| _d| _d| _dS )z%Reset diff results on policy changes.z Resetting category differencesN)r �debugr
r r( )r+ r r r � _reset_diffk s z CategoriesDifference._reset_diffN)
�__name__�
__module__� __qualname__�__doc__r r
r r( r r- r r r r r I s r c @ s8 e Zd ZdZed�Zed�Zed�Zdd� Zdd� Z dS )�SensitivitiesDifferencez?Determine the difference in sensitivities between two policies.�diff_sensitivitiesc C s� | j jdj| �� | jdd� | jj� D �dd� | jj� D ��\| _| _}t � | _
xH|D ]@\}}| j|j � |j � dd�\}}}|s�|rVt |||�| j
|< qVW dS )z>Generate the difference in sensitivities between the policies.zKGenerating sensitivity differences from {0.left_policy} to {0.right_policy}c s s | ]}t |�V qd S )N)r )r �sr r r r � s z=SensitivitiesDifference.diff_sensitivities.<locals>.<genexpr>c s s | ]}t |�V qd S )N)r )r r4 r r r r � s F)r N)
r r! r" r# r$ Z
sensitivitiesr& �added_sensitivities�removed_sensitivitiesr' �modified_sensitivitiesr) �modified_sens_record)r+ Zmatched_sensitivitiesZ left_sensZ
right_sensr r
r r r r r3 { s z*SensitivitiesDifference.diff_sensitivitiesc C s" | j jd� d| _d| _d| _dS )z%Reset diff results on policy changes.z!Resetting sensitivity differencesN)r r, r5 r6 r7 )r+ r r r r- � s z#SensitivitiesDifference._reset_diffN)
r. r/ r0 r1 r r5 r6 r7 r3 r- r r r r r2 s s r2 c @ s8 e Zd ZdZed�Zed�Zed�Zdd� Zdd� Z dS )�LevelDeclsDifferencez8Determine the difference in levels between two policies.� diff_levelsc C s� | j jdj| �� | jdd� | jj� D �dd� | jj� D ��\| _| _}g | _ x\|D ]T\}}| jdd� |j
� D �dd� |j
� D ��\}}}|s�|rT| j j t ||||�� qTW dS )z7Generate the difference in levels between the policies.zJGenerating level decl differences from {0.left_policy} to {0.right_policy}c s s | ]}t |�V qd S )N)�LevelDeclWrapper)r r4 r r r r � s z3LevelDeclsDifference.diff_levels.<locals>.<genexpr>c s s | ]}t |�V qd S )N)r; )r r4 r r r r � s c s s | ]}t |�V qd S )N)r )r r r r r r � s c s s | ]}t |�V qd S )N)r )r r r r r r � s N)
r r! r" r# r$ Zlevelsr& � added_levels�removed_levels�modified_levelsr% �append�modified_level_record)r+ Zmatched_levelsZ
left_levelZ right_levelr
r r r r r r: � s z LevelDeclsDifference.diff_levelsc C s" | j jd� d| _d| _d| _dS )z%Reset diff results on policy changes.z!Resetting sensitivity differencesN)r r, r< r= r> )r+ r r r r- � s z LevelDeclsDifference._reset_diffN)
r. r/ r0 r1 r r< r= r> r: r- r r r r r9 � s r9 c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) r; z-Wrap level declarations to allow comparisons.r c C s || _ t|j�| _t|�| _d S )N)�originr r �hash�key)r+ r r r r �__init__� s zLevelDeclWrapper.__init__c C s | j S )N)rC )r+ r r r �__hash__� s zLevelDeclWrapper.__hash__c C s | j |j kS )N)r )r+ �otherr r r �__eq__� s zLevelDeclWrapper.__eq__c C s | j |j k S )N)r )r+ rF r r r �__lt__� s zLevelDeclWrapper.__lt__N) r. r/ r0 r1 � __slots__rD rE rG rH r r r r r; � s r; c @ s4 e Zd ZdZd
Zdd� Zdd� Zdd � Zd
d � Zd S )� LevelWrapperz!Wrap levels to allow comparisons.r r% c C s. || _ t|j�| _tdd� |j� D ��| _d S )Nc s s | ]}t |�V qd S )N)r )r r r r r r � s z(LevelWrapper.__init__.<locals>.<genexpr>)rA r r �setr% )r+ r r r r rD � s zLevelWrapper.__init__c C s
t | j�S )N)rB rA )r+ r r r rE � s zLevelWrapper.__hash__c C s2 y| j |j ko| j|jkS tk
r, dS X d S )NF)r r% �AttributeError)r+ rF r r r rG � s
zLevelWrapper.__eq__c C s2 y| j |j k o| j|jk S tk
r, dS X d S )NF)r r% rL )r+ rF r r r rH � s
zLevelWrapper.__lt__N)r r% ) r. r/ r0 r1 rI rD rE rG rH r r r r rJ � s rJ c @ s4 e Zd ZdZd
Zdd� Zdd� Zdd � Zd
d � Zd S )� RangeWrapperz�
Wrap ranges to allow comparisons.
This only compares the low and high levels of the range.
It does not detect additions/removals/modifications
to levels between the low and high levels of the range.
�low�highc C s" || _ t|j�| _t|j�| _d S )N)rA rJ rN rO )r+ Zrange_r r r rD s zRangeWrapper.__init__c C s
t | j�S )N)rB rA )r+ r r r rE s zRangeWrapper.__hash__c C s2 y| j |j ko| j|jkS tk
r, dS X d S )NF)rN rO rL )r+ rF r r r rG s
zRangeWrapper.__eq__c C s2 y| j |j k o| j|jk S tk
r, dS X d S )NF)rN rO rL )r+ rF r r r rH s
zRangeWrapper.__lt__N)rN rO ) r. r/ r0 r1 rI rD rE rG rH r r r r rM s rM N)� collectionsr r Z descriptorsr �
differencer r r r* r8 r@ r' r r r r r r2 r9 r; rJ rM r r r r �<module> s, *,,!