3
�^� � @ sl d dl mZmZ ddlmZ ddlmZmZ ddlm Z edddd g�Z
ee �Z d
d � Z
G d d
� d
e�ZdS )� )� defaultdict�
namedtuple� )�DiffResultDescriptor)�
Difference�
SymbolWrapper)�type_wrapper_factoryZ
modified_role� added_types�
removed_types�
matched_typesc C s> yt | j | S tk
r8 t| �}|t | j | < |S X dS )z|
Wrap roles from the specified policy.
This caches results to prevent duplicate wrapper
objects in memory.
N)� _roles_cacheZpolicy�KeyErrorr )Zrole�r� r � /usr/lib64/python3.6/roles.py�role_wrapper_factory" s r c @ s8 e Zd ZdZed�Zed�Zed�Zdd� Zdd� Z dS )�RolesDifferencez7Determine the difference in roles between two policies.�
diff_rolesc C s� | j jdj| �� | jdd� | jj� D �dd� | jj� D ��\| _| _}t � | _
xX|D ]P\}}| jdd� |j � D �dd� |j � D ��\}}}|s�|rVt |||�| j
|< qVW dS )z6Generate the difference in roles between the policies.zDGenerating role differences from {0.left_policy} to {0.right_policy}c s s | ]}t |�V qd S )N)r )�.0r r r r � <genexpr>@ s z-RolesDifference.diff_roles.<locals>.<genexpr>c s s | ]}t |�V qd S )N)r )r r r r r r A s c s s | ]}t |�V qd S )N)r )r �tr r r r J s c s s | ]}t |�V qd S )N)r )r r r r r r K s N)
�log�info�formatZ _set_diffZ left_policyZrolesZ right_policy� added_roles�
removed_roles�dict�modified_roles�types�modified_roles_record)�selfZ
matched_rolesZ left_roleZ
right_roler r
r r r r r 9 s zRolesDifference.diff_rolesc C s" | j jd� d| _d| _d| _dS )z%Reset diff results on policy changes.zResetting role differencesN)r �debugr r r )r r r r � _reset_diffU s zRolesDifference._reset_diffN)
�__name__�
__module__� __qualname__�__doc__r r r r r r" r r r r r 1 s r N)� collectionsr r Z descriptorsr �
differencer r r r r r r r r r r r r �<module> s