RepoMirrors/setools
1
0
Fork 0
mirror of https://github.com/SELinuxProject/setools synced 2025-03-30 23:36:23 +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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
setools/descriptors.py
View File

@ -1,5 +1,5 @@
# 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.
#
@ -28,6 +28,7 @@ for more details.
"""
import re
from abc import ABC, abstractmethod
from collections import defaultdict
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:
name The edge property name
@ -145,13 +146,18 @@ class NetworkXGraphEdgeDescriptor:
if obj is None:
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):
raise NotImplementedError
pass
@abstractmethod
def __delete__(self, obj):
raise NotImplementedError
pass
class EdgeAttrDict(NetworkXGraphEdgeDescriptor):
@ -163,7 +169,8 @@ class EdgeAttrDict(NetworkXGraphEdgeDescriptor):
if value is None:
obj.G[obj.source][obj.target][self.name] = defaultdict(list)
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):
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]
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):