diff options
Diffstat (limited to 'src-py/db_controller.py')
| -rw-r--r-- | src-py/db_controller.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/src-py/db_controller.py b/src-py/db_controller.py index f4bfbb75..2c18427c 100644 --- a/src-py/db_controller.py +++ b/src-py/db_controller.py @@ -141,14 +141,19 @@ class DBO_attr_diff_commit(DB_op): q = ("match (n {id: {id}}) " + "set n += {attr_set}" + - "remove " + rm_attr_str + - " return n.id, n") + "remove " + rm_attr_str + " " + + "return n.id, n") q_param_set = {'id': id_attr, 'attr_set': n_attr_diff['attr_write']} self.add_statement(q, q_param_set) def on_completion(self, data): - pass + ret = {} + for _, _, r_set in self: + for row in r_set: + n_id, n = [v for v in row] # we expect a [n_id, n] array + ret[n_id] = n + return ret class DBO_add_node_set(DB_op): def __init__(self, node_map): |