-
Notifications
You must be signed in to change notification settings - Fork 1
/
entityloader.py
159 lines (111 loc) · 4.06 KB
/
entityloader.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
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
__author__ = 'alacambra'
import neo4j
import Entities
import StringIO
from Entities import Label
execution = True
class Connection:
connection = None
cursor = None
def __init__(self, host="localhost", port=7474):
self.host = host
self.port = port
def set_connection_url(self, host, port):
self.host = host
self.port = port
self.create();
return self
def set_connection_port(self, port):
self.port = port
self.create()
return self
def create(self):
if self.connection is not None:
self.connection.close()
self.connection = neo4j.connect("http://" + self.host + ":" + str(self.port));
self.cursor = self.connection.cursor()
return self
def commit(self):
self.connection.commit();
def get_cursor(self):
if self.cursor is None:
raise Exception("cursor is not initialized")
return self.cursor
connection = Connection().create()
def create_channel(total_items=10):
query_builder = StringIO.StringIO()
channel_gen = Entities.Channel(query_builder)
channel_items_rel_gen = Entities.ChannelItemsRelation(query_builder)
user_gen = Entities.User(query_builder, "CH_O_U")
channel_creator_ref = user_gen.create_user()
items_gen = Entities.ChannelItem(query_builder, channel_creator_ref)
channel_ref, channel_uuid = channel_gen.create_channel()
channel_items_rel_gen.set_channel_owner(channel_ref, channel_creator_ref)
first_item, first_channel_item_uuid = items_gen.create_first_item()
channel_items_rel_gen.set_first_item(channel_ref, first_item)
second_item = None
step = 20
for i in range(0, total_items - 1):
if i%step == 0 and i != 0:
print "-"*50 + str(i)
execute(query_builder.getvalue())
connection.commit()
query_builder.truncate(0)
query_builder.seek(0)
query_builder.write("match (last:" + Label.uuid + " {" + Label.uuid + ":'" + last_channel_item_uuid + "'})")
second_item = "last"
if second_item is not None:
first_item = second_item
second_item, last_channel_item_uuid = items_gen.create_channel_item(Label.bce if i % 4 else Label.container)
channel_items_rel_gen.connect_items(first_item, second_item)
execute(query_builder.getvalue())
connection.commit()
query_builder.truncate(0)
query_builder.seek(0)
channel_query = "MATCH (channel: " + Label.uuid + " {" + Label.uuid + ":'" + channel_uuid + "'}), (last:" \
+ Label.uuid + " {" + Label.uuid + ":'" + last_channel_item_uuid +"'})\n"
query_builder.write(channel_query)
channel_items_rel_gen.set_last_item("channel", "last")
i = 0
print "-"*50 + str(i)
execute(query_builder.getvalue())
connection.commit()
def clear_all():
query = "MATCH n return count(n) as total"
result = connection.get_cursor().execute(query)
(total, ) = result.next()
step = 50
print str(total) + " nodes to delete:"
total_removed = 0
for i in range(0, total, step):
print " - "*25 + str(total_removed)
query = "match n with n LIMIT " + str(step) + " optional match n-[r]-() delete r, n return count(n) as removed"
result = execute(query)
(removed, ) = result.next()
total_removed += removed
connection.commit()
def execute(query):
# print query
if execution:
result = connection.get_cursor().execute(query)
# connection.commit()
return result;
else:
print query
def setup_db(clear_db=False):
if clear_db:
clear_all()
# uniqueness
unique_fields = {
'uuid': 'uuid',
'person': 'email'
}
for constraint in unique_fields.items():
query = "CREATE CONSTRAINT ON (node:" + constraint[0] + ") ASSERT node." + constraint[1] + " IS UNIQUE"
execute(query)
connection.commit()
if __name__ == "__main__":
print "hello"
# clear_all();
for i in range(0, 1):
create_channel()