3
�^� � @ s� d Z ddlZddlmZmZ ddlmZ ddlmZ G dd� d�Z G dd � d e �Z
G d
d � d e�Z G d d
� d
e �Z G dd� de �Z
G dd� de �ZG dd� d�ZdS )z�
SETools descriptors.
These classes override how a class's attributes are get/set/deleted.
This is how the @property decorator works.
See https://docs.python.org/3/howto/descriptor.html
for more details.
� N)�ABC�abstractmethod)� defaultdict)�WeakKeyDictionaryc @ s, e Zd ZdZd dd�Zd
dd�Zdd� ZdS ) �CriteriaDescriptora�
Single item criteria descriptor.
Keyword Parameters:
name_regex The name of instance's regex setting attribute;
used as name_regex below. If unset,
regular expressions will never be used.
lookup_function The name of the SELinuxPolicy lookup function,
e.g. lookup_type or lookup_boolean.
default_value The default value of the criteria. The default
is None.
enum_class The class of enumeration which supports a
lookup class method.
Read-only instance attribute use (obj parameter):
policy The instance of SELinuxPolicy
name_regex This attribute is read to determine if
the criteria should be looked up or
compiled into a regex. If the attribute
does not exist, False is assumed.
Nc C sJ |s|s|st d��|o| s&t d��|| _|| _|| _|| _t� | _d S )NzQA simple attribute should be used if there is no regex, lookup function, or enum.z9Lookup functions and enum classes are mutually exclusive.)�AssertionError�regex�
default_value�lookup_function�
enum_classr � instances)�selfZ
name_regexr
r r � r �#/usr/lib64/python3.6/descriptors.py�__init__C s zCriteriaDescriptor.__init__c C s |d kr | S | j j|| j�S )N)r �
setdefaultr )r
�obj�objtyper r r �__get__R s zCriteriaDescriptor.__get__c C s� |sd | j |< nn| jr6t|| jd�r6tj|�| j |< nH| jrZt|j| j�}||�| j |< n$| jrt| jj|�| j |< n
|| j |< d S )NF) r r �getattr�re�compiler
�policyr �lookup)r
r �valuer r r r �__set__X s zCriteriaDescriptor.__set__)NNNN)N)�__name__�
__module__� __qualname__�__doc__r r r r r r r r + s
r c @ s e Zd ZdZdd� ZdS )�CriteriaSetDescriptorz!Descriptor for a set of criteria.c s� |sd �j |< n��jr6t|�jd�r6tj|��j |< nd�jrht|j�j�� t� fdd�|D ���j |< n2�jr�t�fdd�|D ���j |< nt|��j |< d S )NFc 3 s | ]}� |�V qd S )Nr )�.0�v)r r r � <genexpr>q s z0CriteriaSetDescriptor.__set__.<locals>.<genexpr>c 3 s | ]}� j j|�V qd S )N)r r )r! r" )r
r r r# s s ) r r r r r r
r �setr )r
r r r )r r
r r j s z CriteriaSetDescriptor.__set__N)r r r r r r r r r r f s r c @ s: e Zd ZdZdd� Zd dd�Zedd� �Zed d
� �ZdS ) �NetworkXGraphEdgeDescriptora'
Descriptor abstract base class for NetworkX graph edge attributes.
Parameter:
name The edge property name
Instance class attribute use (obj parameter):
G The NetworkX graph
source The edge's source node
target The edge's target node
c C s
|| _ d S )N)�name)r
�propnamer r r r � s z$NetworkXGraphEdgeDescriptor.__init__Nc C sH |d kr | S y|j |j |j | j S tk
rB t| j��Y nX d S )N)�G�source�targetr& �KeyError�AttributeError)r
r r r r r r � s z#NetworkXGraphEdgeDescriptor.__get__c C s d S )Nr )r
r r r r r r � s z#NetworkXGraphEdgeDescriptor.__set__c C s d S )Nr )r
r r r r �
__delete__� s z&NetworkXGraphEdgeDescriptor.__delete__)N) r r r r r r r r r- r r r r r% � s
r% c @ s e Zd ZdZdd� Zdd� ZdS )� EdgeAttrDictz7A descriptor for edge attributes that are dictionaries.c C s: |d kr&t t�|j|j |j | j< ntdj| j���d S )Nz0{0} dictionaries should not be assigned directly)r �listr( r) r* r& r, �format)r
r r r r r r � s zEdgeAttrDict.__set__c C s |j |j |j | j j� d S )N)r( r) r* r& �clear)r
r r r r r- � s zEdgeAttrDict.__delete__N)r r r r r r- r r r r r. � s r. c @ s e Zd ZdZdd� Zdd� ZdS )�EdgeAttrIntMaxz�
A descriptor for edge attributes that are non-negative integers that always
keep the max assigned value until re-initialized.
c C s\ |d kr"d|j |j |j | j< n6|j |j |j | j }t||�|j |j |j | j< d S )Nr )r( r) r* r& �max)r
r r Z
current_valuer r r r � s zEdgeAttrIntMax.__set__c C s d|j |j |j | j< d S )Nr )r( r) r* r& )r
r r r r r- � s zEdgeAttrIntMax.__delete__N)r r r r r r- r r r r r2 � s r2 c @ s e Zd ZdZdd� Zdd� ZdS )� EdgeAttrListz0A descriptor for edge attributes that are lists.c C s6 |d kr"g |j |j |j | j< ntdj| j���d S )Nz){0} lists should not be assigned directly)r( r) r* r& �
ValueErrorr0 )r
r r r r r r � s zEdgeAttrList.__set__c C s |j |j |j | j j� d S )N)r( r) r* r& r1 )r
r r r r r- � s zEdgeAttrList.__delete__N)r r r r r r- r r r r r4 � s r4 c @ s2 e Zd ZdZdd� Zd dd�Zdd� Zd d
� ZdS ) �PermissionMapDescriptoraN
Descriptor for Permission Map mappings.
Parameter:
name The map setting name.
validator A callable for validating the setting.
Instance class attribute use (obj parameter):
perm_map The full permission map.
class_ The mapping's object class
perm The mapping's permission
c C s || _ || _d S )N)r& � validator)r
r' r7 r r r r � s z PermissionMapDescriptor.__init__Nc C s$ |d kr | S |j |j |j | j S )N)�perm_map�class_�permr& )r
r r r r r r � s z PermissionMapDescriptor.__get__c C s" | j |�|j|j |j | j< d S )N)r7 r8 r9 r: r& )r
r r r r r r � s z PermissionMapDescriptor.__set__c C s t �d S )N)r, )r
r r r r r- � s z"PermissionMapDescriptor.__delete__)N)r r r r r r r r- r r r r r6 � s
r6 )r r �abcr r � collectionsr �weakrefr r r r% r. r2 r4 r6 r r r r �<module> s
;#