1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
# This file is part of rhizi, a collaborative knowledge graph editor.
# Copyright (C) 2014-2015 Rhizi
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from db_op import DBO_raw_query_set
import neo4j_util
class DBO_RDG__skill_graph(DBO_raw_query_set):
def __init__(self, rzdoc,
lim_n=50,
lim_r=10000,
prob_link_create=0.03,
name_set=['Bruce Lee', 'Kesuke Miyagi'],
skill_set=['Kung Foo', 'Karate'],
skill_level_set=['Novice', 'Intermediate', 'Expert']
):
"""
Random data generator:
- Name x Skill[proficiency: {Novice|Intermediate|Expert}]
"""
super(DBO_RDG__skill_graph, self).__init__()
rzdoc_label = neo4j_util.rzdoc__ns_label(rzdoc)
min_id = 10000 # separate group ids with numerical perfix
# create persons
q_arr = ['with {name_set} as n_attr_set', # TODO clean: foreach triggers SyntaxException: otherwise
'foreach (idx in range(0,%d)' % (lim_n - 1),
'|',
'create (n:%s:%s' % (rzdoc_label, 'Person'),
'{id: \'test-id_\' + toString(%d + idx),' % (min_id),
'name: n_attr_set[idx %% %d]}' % (len(name_set)),
'))',
]
q_param = {'name_set': name_set}
self.add_statement(q_arr, q_param)
min_id *= 2
# create skills
q_arr = ['with {skill_set} as skill_set', # TODO clean: foreach triggers SyntaxException: otherwise
'foreach (idx in range(0,%d)' % (len(skill_set) - 1),
'|',
'create (n:%s:%s' % (rzdoc_label, 'Skill'),
'{id: \'test-id_\' + toString(idx + %d),' % (min_id),
'name: skill_set[idx]}',
'))',
]
q_param = {'skill_set': skill_set}
self.add_statement(q_arr, q_param)
min_id *= 2
# create links
for skill_level in skill_level_set:
q_arr = ['match (n:%s),(m:%s)' % ('Person', 'Skill'),
'with n, m, rand() as rand',
'order by rand',
'limit %d' % (lim_r - 1),
'where rand() < %.2f' % (prob_link_create),
'create (n)-[r:%s' % (skill_level),
'{id: \'test-id_\' + toString(%d + toInt(%d * rand())),' % (min_id, lim_r * 100000), # aim for low id collision probability,
'proficiency: \'%s\'}' % (skill_level),
']->(m)',
'return collect(r.id)',
]
self.add_statement(q_arr, q_param)
|
