X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=PLC%2FSlices.py;h=42da1d4cb12d633056fc80a84bb1f65343dc7b0c;hb=c52399403d9a9c76b21fa343d88f3f5e92b0a62d;hp=e62bd639c5cb91de39d2fad02a4c2034c4b43b1b;hpb=e347fc823bbba9d88a3fddf07d5c21024dfd1e55;p=plcapi.git diff --git a/PLC/Slices.py b/PLC/Slices.py index e62bd63..42da1d4 100644 --- a/PLC/Slices.py +++ b/PLC/Slices.py @@ -125,16 +125,13 @@ class Slice(Row): self['person_ids'].remove(person_id) person['slice_ids'].remove(slice_id) - def add_node(self, node, is_foreign_node = False, commit = True): + def add_node(self, node, commit = True): """ Add node to existing slice. """ assert 'slice_id' in self - if not is_foreign_node: - assert isinstance(node, Node) - else: - assert isinstance(node, ForeignNode) + assert isinstance(node, (Node,ForeignNode)) assert 'node_id' in node slice_id = self['slice_id'] @@ -159,7 +156,7 @@ class Slice(Row): """ assert 'slice_id' in self - assert isinstance(node, Node) + assert isinstance(node, (Node,ForeignNode)) assert 'node_id' in node slice_id = self['slice_id']