Network: Small code refactoring.

This commit is contained in:
Pierre-Antoine Rouby committed 2023-06-08 14:18:12 +02:00
1 parent 83b514db61
commit a6ddba484e
2 files changed
+56 -32

No files matched your search

+11 -9
View File
@@ -12,18 +12,18 @@ class Edge(object):
self._status = status
self.id = id
self._name = name if name != "" else f"{node1.name} -> {node2.name}"
self._name = name
self.node1 = node1
self.node2 = node2
self.enable = True
self._enable = True
def __getitem__(self, name):
ret = None
if name == "name":
ret = self._name
ret = self.name
elif name == "id":
ret = self.id
elif name == "node1":
@@ -31,7 +31,7 @@ class Edge(object):
elif name == "node2":
ret = self.node2.name
elif name == "enable":
ret = self.enable
ret = self._enable
return ret
@@ -45,16 +45,18 @@ class Edge(object):
elif name == "node2":
self.node2 = value
elif name == "enable":
self.enable = value
self._enable = value
self._status.modified()
@property
def name(self):
return self._name
return self._name if self._name != "" else f"{self.node1.name} -> {self.node2.name}"
def enable(self):
self.enable = True
def enable(self, enable=True):
self._enable = enable
self._status.modified()
def disable(self):
self.enable = False
self._enable = False
self._status.modified()