3
^B) @ s d dl mZmZ ddlmZ ddlmZmZmZ eddddgZ ed dddgZ
ed
d d d
dgZ ee Z
ee Zdd Zdd ZG dd deZG dd deZG dd deZG dd deZG dd d eZG d d d eZd S ) ) defaultdict
namedtuple )DiffResultDescriptor)
Difference
SymbolWrapperWrapperZmodified_category
added_aliasesremoved_aliasesmatched_aliasesZmodified_sensitivityZmodified_levelleveladded_categoriesremoved_categoriesmatched_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_cachepolicyKeyErrorr )categoryc r /usr/lib64/python3.6/mls.pycategory_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 ZdZedZedZedZdd Zdd Z dS )CategoriesDifferencezX s z7CategoriesDifference.diff_categories..c s s | ]}t |V qd S )N)r )r r r r r r Y s F)unwrapN)
loginfoformat _set_diff left_policy
categories right_policyr
r dictmodified_categoriesaliasesmodified_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 ZdZedZedZedZdd 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..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_sensitivitiesremoved_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 ZdZedZedZedZdd 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..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_levelsremoved_levelsmodified_levelsr% appendmodified_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 hashkey)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__..)rA r r setr% )r+ r r r r rD s zLevelWrapper.__init__c C s
t | jS )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.
lowhighc 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 | jS )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 s, *,,!