modelverse/evolution/attribute_ops.py

import sys
sys.path.append("../wrappers")
from wrappers import modelverse as mv
import commons


class AttributeAdd(object):
    def __init__(self):
        self._key = ""
        self._value = ""
        self._node_id = ""
        self._node_type = ""

    def execute(self, model, node_id, key, value, local):
        """
        Add a new attribute (key, value) to node with node_id in model.
        """
        self._key = key
        self._value = value
        self._node_id = node_id

        if not self._node_type:
            self._node_type = commons.get_node_type(model, node_id)

        if local:
            mv.transformation_execute_MANUAL("graph_ops/add_attribute", {"gm":model}, {"gm":model},
                                             callback=self._callback)
        else:
            for m in commons.all_models():
                nodes = commons.all_nodes_with_type(m, self._node_type)
                for nid in nodes:
                    self.execute(m, nid, key, value, local=True)

    def _callback(self, model):
        attr_id = mv.instantiate(model, "gm/Attribute")
        mv.attr_assign(model, attr_id, "key", self._key)
        mv.attr_assign(model, attr_id, "value", self._value)
        mv.instantiate(model, "gm/NodeAttribute", ("gm/"+self._node_id, attr_id))


class AttributeDelete(object):
    def __init__(self):
        self._key = ""
        self._node_id = ""
        self._node_type = ""

    def execute(self, model, node_id, key, local):
        """
        Deletes an attribute identified by its key from node_id in model.
        """
        self._key = key
        self._node_id = node_id

        if not self._node_type:
            self._node_type = commons.get_node_type(model, node_id)

        if local:
            mv.transformation_execute_MANUAL("graph_ops/del_attribute", {"gm":model}, {"gm":model},
                                             callback=self._callback)
        else:
            # delete all attributes with key
            for m in commons.all_models():
                nodes = commons.all_nodes_with_type(m, self._node_type)
                for nid in nodes:
                    self.execute(m, nid, key, local=True)

    def _callback(self, model):
        outgoings = mv.read_outgoing(model, "gm/"+self._node_id, "gm/NodeAttribute")
        for edge in outgoings:
            attr = mv.read_association_destination(model, edge)[0]
            attr_key = mv.read_attrs(model, attr)["key"]
            if attr_key == self._key:
                mv.delete_element(model, attr)
                break


class AttributeChange(object):
    """
    Remark: This is currently not needed since the UI does not have the functionality to update an attribute
    key in-place. Instead, an attribute must be deleted and added with the new key to effectively change the key.
    Therefore, this is only here for the sake of completeness.
    """
    def __init__(self):
        self._node_id = ""
        self._old_key = ""
        self._new_key = ""
        self._node_type = ""

    def execute(self, model, node_id, old_key, new_key, local):
        """
        Update the attribute key old_key of node_id in model to new_key.
        """
        self._node_id = node_id
        self._old_key = old_key
        self._new_key = new_key

        if not self._node_type:
            self._node_type = commons.get_node_type(model, node_id)

        if local:
            mv.transformation_execute_MANUAL("graph_ops/change_attribute", {"gm":model}, {"gm":model},
                                             callback=self._callback)

        else:
            for m in commons.all_models():
                nodes = commons.all_nodes_with_type(m, self._node_type)
                for nid in nodes:
                    self.execute(m, nid, old_key, new_key, local=True)

    def _callback(self, model):
        outgoings = mv.read_outgoing(model, "gm/"+self._node_id, "gm/NodeAttribute")
        for link in outgoings:
            attr = mv.read_association_destination(model, link)[0]
            attr_key = mv.read_attrs(model, attr)["key"]
            if attr_key == self._old_key:
                mv.attr_assign(model, attr, "key", self._new_key)
                break