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
|
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)
|
