forked from mmsdivino/Tourisys
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdata_filter_sp.py
64 lines (45 loc) · 1.76 KB
/
data_filter_sp.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
import json
#SP - SP
sp_lower_lat = -23.784969
sp_lower_long = -46.809319
sp_upper_lat = -23.39566
sp_upper_long = -46.36499
# Salvador
ssa_lower_long = -38.699435
ssa_lower_lat = -13.017395
ssa_upper_long = -38.303414
ssa_upper_lat = -12.733537
ssa_places = set()
ssa_users = set()
# gPlusUserId
with open("poi_brasil.json") as poi_data_file:
line = poi_data_file.readline()
with open("ssa_pois.json", "w") as review_data_file:
while line:
line_json = eval(line)
if line_json['gps'] and line_json['gps'][0] >= ssa_lower_lat and line_json['gps'][1] >= ssa_lower_long and line_json['gps'][0] <= ssa_upper_lat and line_json['gps'][1] <= ssa_upper_long:
# print('Entrou')
review_data_file.write(line)
ssa_places.add(line_json['gPlusPlaceId'])
# else:
# print(line_json)
line = poi_data_file.readline()
print('Comecou a bagaceira dos reviews')
with open("reviews.json") as poi_data_file:
line = poi_data_file.readline()
with open("ssa_reviews.json", "w") as review_data_file:
while line:
line_json = eval(line)
if line_json['gPlusPlaceId'] in ssa_places:
ssa_users.add(line_json['gPlusUserId'])
review_data_file.write(line)
line = poi_data_file.readline()
print('Comecou a bagaceira dos usuarios')
with open("users.json") as poi_data_file:
line = poi_data_file.readline()
with open("ssa_users.json", "w") as review_data_file:
while line:
line_json = eval(line)
if line_json['gPlusUserId'] in ssa_users:
review_data_file.write(line)
line = poi_data_file.readline()