RepoMirrors/setools
1
0
Fork 0
mirror of https://github.com/SELinuxProject/setools synced 2025-04-11 03:51:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

NetworkXGraphEdgeDescriptor: Convert to proper abstract base class.

Browse Source
This commit is contained in:
Chris PeBenito 2018-11-10 10:49:53 -05:00
parent 6fef953597
commit ca82842a3d
1 changed files with 17 additions and 7 deletions

24
setools/descriptors.py
View File

@ -1,5 +1,5 @@
# Copyright 2015, Tresys Technology, LLC # Copyright 2015, Tresys Technology, LLC
# Copyright 2016, Chris PeBenito <pebenito@ieee.org> # Copyright 2016, 2018, Chris PeBenito <pebenito@ieee.org>
# #
# This file is part of SETools. # This file is part of SETools.
# #
@ -28,6 +28,7 @@ for more details.
""" """
import re import re
from abc import ABC, abstractmethod
from collections import defaultdict from collections import defaultdict
from weakref import WeakKeyDictionary from weakref import WeakKeyDictionary
@ -124,10 +125,10 @@ class CriteriaSetDescriptor(CriteriaDescriptor):
# #
class NetworkXGraphEdgeDescriptor: class NetworkXGraphEdgeDescriptor(ABC):
""" """
Descriptor base class for NetworkX graph edge attributes. Descriptor abstract base class for NetworkX graph edge attributes.
Parameter: Parameter:
name The edge property name name The edge property name
@ -145,13 +146,18 @@ class NetworkXGraphEdgeDescriptor:
if obj is None: if obj is None:
return self return self
return obj.G[obj.source][obj.target][self.name] try:
return obj.G[obj.source][obj.target][self.name]
except KeyError:
raise AttributeError(self.name)
@abstractmethod
def __set__(self, obj, value): def __set__(self, obj, value):
raise NotImplementedError pass
@abstractmethod
def __delete__(self, obj): def __delete__(self, obj):
raise NotImplementedError pass
class EdgeAttrDict(NetworkXGraphEdgeDescriptor): class EdgeAttrDict(NetworkXGraphEdgeDescriptor):
@ -163,7 +169,8 @@ class EdgeAttrDict(NetworkXGraphEdgeDescriptor):
if value is None: if value is None:
obj.G[obj.source][obj.target][self.name] = defaultdict(list) obj.G[obj.source][obj.target][self.name] = defaultdict(list)
else: else:
raise ValueError("{0} dictionaries should not be assigned directly".format(self.name)) raise AttributeError("{0} dictionaries should not be assigned directly".
format(self.name))
def __delete__(self, obj): def __delete__(self, obj):
obj.G[obj.source][obj.target][self.name].clear() obj.G[obj.source][obj.target][self.name].clear()
@ -184,6 +191,9 @@ class EdgeAttrIntMax(NetworkXGraphEdgeDescriptor):
current_value = obj.G[obj.source][obj.target][self.name] current_value = obj.G[obj.source][obj.target][self.name]
obj.G[obj.source][obj.target][self.name] = max(current_value, value) obj.G[obj.source][obj.target][self.name] = max(current_value, value)
def __delete__(self, obj):
obj.G[obj.source][obj.target][self.name] = 0
class EdgeAttrList(NetworkXGraphEdgeDescriptor): class EdgeAttrList(NetworkXGraphEdgeDescriptor):