src/local/domain/cri.rhizi.net/domain_cri.py


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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140

from db_op import DB_op

class DBO_random_data_generation__domain__CRI(DB_op):

    def __init__(self, lim_n=50,
                       lim_r=10000,
                       prob_link_create=0.03):
        """
        generate random data: CRI domain: Person x Skill
        """
        super(DBO_random_data_generation__domain__CRI, self).__init__()

        n_attr_set__skill_raw = ['Cryptocurrency',
                             'Database architecture',
                             'Web-development',
                             'Mobile-development',
                             'Machine learning',
                             'Guitar Playing',
                             'Image processing',
                             'Algebra',
                             'Calculus',
                             'Molecular-Biology',
                             'Graphic design',
                             'Geo-location services',
                             'Drone-building',
                             'Artificial-intelligence',
                             'Distributed information systems',
                             'Wordpress',
                             'Woodworking',
                             'Quality-control',
                             'Video-editing',
                             'Soldering',
                             'Network engineering',
                             'GIT',
                             'Electronic music',
                             'Network administration']

        n_attr_set__skill = []
        for s in n_attr_set__skill_raw:
            n_attr_set__skill.append(s)

        n_attr_set__name = ['John',
                            'William',
                            'James',
                            'Charles',
                            'George',
                            'Frank',
                            'Joseph',
                            'Thomas',
                            'Henry',
                            'Robert',
                            'Edward',
                            'Harry',
                            'Walter',
                            'Arthur',
                            'Fred',
                            'Albert',
                            'Samuel',
                            'David',
                            'Louis',
                            'Joe',
                            'Charlie',
                            'Clarence',
                            'Richard',
                            'Andrew',
                            'Daniel',
                            'Ernest',
                            'Mary',
                            'Anna',
                            'Emma',
                            'Elizabeth',
                            'Minnie',
                            'Margaret',
                            'Ida',
                            'Alice',
                            'Bertha',
                            'Sarah',
                            'Annie',
                            'Clara',
                            'Ella',
                            'Florence',
                            'Cora',
                            'Martha',
                            'Laura',
                            'Nellie',
                            'Grace',
                            'Carrie',
                            'Maude',
                            'Mabel',
                            'Bessie',
                            'Jennie',
                            'Gertrude',
                            'Julia']

        l_attr_set__level = ['Novice', 'Intermediate', 'Expert']


        min_id = 10000  # separate group ids with numerical perfix

        # create persons
        q_arr = ['with {n_attr_set__name} as n_attr_set',  # TODO clean: foreach triggers SyntaxException: otherwise
                 'foreach (idx in range(0,%d)' % (lim_n - 1),
                 '|',
                 'create (n:%s' % ('Person'),
                 '{id: \'test-id_\' + toString(%d + idx),' % (min_id),
                  'name: n_attr_set[idx %% %d]}' % (len(n_attr_set__name)),
                 '))',
                ]
        q_param = {'n_attr_set__name': n_attr_set__name}
        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(n_attr_set__skill) - 1),
                 '|',
                 'create (n:%s' % ('Skill'),
                 '{id: \'test-id_\' + toString(idx + %d),' % (min_id),
                  'name: skill_set[idx]}',
                 '))',
                ]
        q_param = {'skill_set': n_attr_set__skill}
        self.add_statement(q_arr, q_param)
        min_id *= 2

        # create links
        for skill_level in l_attr_set__level:
            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)