3
^ @ s d dl mZmZ ddlmZ ddlmZ ddlmZm Z ddl
m Z m Z ddl
mZ ed d
d d d
dddgZeeZdd ZG dd deZdS ) ) 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_cacheZpolicyKeyErrorr )userr r /usr/lib64/python3.6/users.pyuser_wrapper_factory) s r c @ s8 e Zd ZdZedZedZedZdd 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 || kr6|
}| }nd}d}|| krR|}|
}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 G s z-UsersDifference.diff_users..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))loginfoformatZ _set_diffZ left_policyZusersZ right_policy added_users
removed_usersdictmodified_usersrolesr
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 s