3
�^� � @ s� d dl mZmZ ddlmZ ddlmZ ddlmZm Z ddl
m Z m Z ddl
mZ ed d
d d d
dddg�Zee�Zdd� ZG dd� de�ZdS )� )� defaultdict�
namedtuple� )� MLSDisabled� )�DiffResultDescriptor)�
Difference�
SymbolWrapper)� LevelWrapper� RangeWrapper)�role_wrapper_factoryZ
modified_user� added_roles�
removed_roles�
matched_roles� added_level�
removed_level� added_range�
removed_rangec C s> yt | j | S tk
r8 t| �}|t | j | < |S X dS )z|
Wrap users from the specified policy.
This caches results to prevent duplicate wrapper
objects in memory.
N)� _users_cacheZpolicy�KeyErrorr )�user�r� r � /usr/lib64/python3.6/users.py�user_wrapper_factory) s r c @ s8 e Zd ZdZed�Zed�Zed�Zdd� Zdd� Z dS )�UsersDifferencez7Determine the difference in users between two policies.�
diff_usersc C s� | j jdj| �� | jdd� | jj� D �dd� | jj� D ��\| _| _}t � | _
�x:|D �]0\}}| jdd� |j D �dd� |j D ��\}}}y$t |j
�}t|j�}|j
} |j}
W n$ tk
r� d}d}d} d}
Y nX y$t |j
�} t|j�} |j
}
|j}W n& tk
�r d} d} d}
d}Y nX || k�r6|
}| }nd}d}|| k�rR|}|
}nd}d}|�sp|�sp|�sp|rXt|||||||�| j
|< qXW dS ) z6Generate the difference in users between the policies.zDGenerating user differences from {0.left_policy} to {0.right_policy}c s s | ]}t |�V qd S )N)r )�.0r r r r � <genexpr>G s z-UsersDifference.diff_users.<locals>.<genexpr>c s s | ]}t |�V qd S )N)r )r r r r r r H s c s s | ]}t |�V qd S )N)r )r r r r r r R s c s s | ]}t |�V qd S )N)r )r r r r r r S s NzNone (MLS Disabled))�log�info�formatZ _set_diffZ left_policyZusersZ right_policy� added_users�
removed_users�dict�modified_users�rolesr
Z mls_levelr Z mls_ranger �modified_users_record)�selfZ
matched_usersZ left_userZ
right_userr
r r Zleft_level_wrapZleft_range_wrapZ
left_levelZ
left_rangeZright_level_wrapZright_range_wrapZ right_levelZ right_ranger r r r r r r r @ s`
zUsersDifference.diff_usersc C s" | j jd� d| _d| _d| _dS )z%Reset diff results on policy changes.zResetting user differencesN)r �debugr" r# r% )r( r r r � _reset_diff� s zUsersDifference._reset_diffN)
�__name__�
__module__� __qualname__�__doc__r r" r# r% r r* r r r r r 8 s Gr N)� collectionsr r Z exceptionr Z descriptorsr �
differencer r Zmlsr
r r& r r' r$ r r r r r r r �<module> s |