diff options
| -rw-r--r-- | src-py/db_controller.py | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/src-py/db_controller.py b/src-py/db_controller.py index f43c9673..a86de1a8 100644 --- a/src-py/db_controller.py +++ b/src-py/db_controller.py @@ -351,6 +351,51 @@ class DBO_rm_node_set(DB_op): q = ' '.join(q_arr) # TODO: use id param upon neo4j support: q_params = {'id_set': id_set} self.add_statement(q) +class DBO_rz_clone(DB_op): + def __init__(self, filter_label=None, limit=128): + """ + clone rhizi + + @return: a dict: {'node_set': n_set, + 'link_set': l_set } + where l_set is a list of (src.id, dst.id, link) tuples + """ + super(DBO_rz_clone, self).__init__() + + self.limit = limit + self.skip = 0 + + q_arr = ['match (n)' if not filter_label else 'match (n:%s)' % (filter_label), + 'optional match (n)-[r]->(m)', + 'with n,r,m', + 'order by n.id', + 'skip %d' % (self.skip), + 'limit %d' % (self.limit), + 'return n,collect([n.id, m.id, r])'] + + q = ' '.join(q_arr) + self.add_statement(q) + + def process_result_set(self): + ret_n_set = [] + ret_l_set = [] + for _, _, row_set in self: + for row in row_set: + itr = iter(row) + n = itr.next() + + l_set = itr.next() + for l in l_set: + assert 3 == len(l) # (n.id, m.id, r) tuples + if None == l[1]: + # as link matching is optional, collect may yield empty sets + continue + ret_l_set.append(l) + ret_n_set.append(n) + + return {'node_set': ret_n_set, + 'link_set': ret_l_set } + class DB_Controller: """ neo4j DB controller |