3
�^4 � @ s� d dl mZmZ ddlmZ ddlmZ ddlmZm Z ddl
m Z ddl m
Z
dd lmZ ed
d d d
g�ZG dd� de�ZG dd� de �ZdS )� )� defaultdict�
namedtuple� )� MLSRuletype� )�DiffResultDescriptor)�
Difference�Wrapper)� RangeWrapper)�class_wrapper_factory)� type_or_attr_wrapper_factoryZmodified_mlsrule�ruleZ
added_defaultZremoved_defaultc @ sP e Zd ZdZed�Zed�Zed�Zee �Z
ee �Z dd� Z dd� Z
dd� Zd S )
�MLSRulesDifferencez;Determine the difference in MLS rules between two policies.�diff_range_transitionsc C s� | j jdj| �� | j s"| j r*| j� | j| j| jtj t
�| j| jtj t
��\}}}g }x:|D ]2\}}t |j �t |j �krf|j
t||j |j �� qfW || _|| _|| _dS )zGGenerate the difference in range_transition rules between the policies.zPGenerating range_transition differences from {0.left_policy} to {0.right_policy}N)�log�info�format�_left_mls_rules�_right_mls_rules�_create_mls_rule_listsZ _set_diffZ_expand_generatorr Zrange_transition�MLSRuleWrapperr
�default�append�modified_mlsrule_record�added_range_transitions�removed_range_transitions�modified_range_transitions)�selfZaddedZremovedZmatchedZmodifiedZ left_ruleZ
right_rule� r � /usr/lib64/python3.6/mlsrules.pyr / s& z)MLSRulesDifference.diff_range_transitionsc C s� | j jdj| �� x$| jj� D ]}| j|j j|� q W | j jdj| �� x$| jj� D ]}| j |j j|� qVW | j jd� dS )z$Create rule lists for both policies.z,Building MLS rule lists from {0.left_policy}z-Building MLS rule lists from {0.right_policy}z"Completed building MLS rule lists.N)
r �debugr Z left_policyZmlsrulesr �ruletyper Z right_policyr )r r
r r r r P s z)MLSRulesDifference._create_mls_rule_listsc C s6 | j jd� d| _d| _d| _| jj� | jj� dS )z%Reset diff results on policy changes.z Resetting MLS rule differencesN)r r r r r r �clearr )r r r r � _reset_diff^ s
z MLSRulesDifference._reset_diffN)�__name__�
__module__� __qualname__�__doc__r r r r r �listr r r r r# r r r r r # s !r c @ s4 e Zd ZdZdZdd� Zdd � Zd
d � Zd d
� ZdS )r z'Wrap MLS rules to allow set operations.r! �source�target�tclassc C s8 || _ t|j�| _t|j�| _t|j�| _t|�| _d S )N)�originr r) r* r r+ �hash�key)r r
r r r �__init__p s
zMLSRuleWrapper.__init__c C s | j S )N)r. )r r r r �__hash__w s zMLSRuleWrapper.__hash__c C s | j |j k S )N)r. )r �otherr r r �__lt__z s zMLSRuleWrapper.__lt__c C s$ | j |j ko"| j|jko"| j|jkS )N)r) r* r+ )r r1 r r r �__eq__} s zMLSRuleWrapper.__eq__N)r! r) r* r+ ) r$ r% r& r' � __slots__r/ r0 r2 r3 r r r r r j s r N)� collectionsr r � r Z descriptorsr �
differencer r Zmlsr
Zobjclassr �typesr r r r r r r r �<module> s G