-
Notifications
You must be signed in to change notification settings - Fork 0
/
nea.py
60 lines (53 loc) · 2.88 KB
/
nea.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
import datetime
import random
import time
import pymongo
from pymongo import MongoClient
from bson.json_util import dumps
client = MongoClient('mongodb://localhost:27017/')
db = client['twitter']
dict1 = {0:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0},1:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0},2:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0},3:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0},4:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0},5:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0},6:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0}}
dict2 = {0:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0},1:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0},2:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0},3:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0},4:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0},5:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0},6:{0:0,1:0,2:0,3:0,4:0,5:0,6:0,7:0,8:0,9:0,10:0,11:0,12:0,13:0,14:0,15:0,16:0,17:0,18:0,19:0,20:0,21:0,22:0,23:0}}
for obj in db.geo_tagged.find():
geo = dumps(obj)[15][7][1]
for road in db.road_collection.find({location:{'$near':{'$geometry':geo,'$maxDistance': 1000}}}):
x = time.strftime("%w%H%M", time.localtime(int(dumps(obj)[27],15)))
day = int(x[0])
hour = int(x[1:3])
if int(x[1])==0:
hour = int(x[2])
min = int(x[3:])
if min<30:
dict1[day][hour]=dict1[day][hour]+5
else:
dict2[day][hour]=dict2[day][hour]+5
merged = db.merged
for d in range(0,7):
for h in range(0,24):
for s in range(0,2):
if d==0:
q='Sun'
if d==1:
q='Mon'
if d==2:
q='Tue'
if d==3:
q='Wed'
if d==4:
q='Thu'
if d==5:
q='Fri'
if d==6:
q='Sat'
if random.random()<=(1/3):
rand = 'L'
else:
if random.random()>(2/3):
rand = 'H'
else:
rand = 'M'
if s==0:
merged_obj = {'day':q,'hour':'H'+str(h),'minute':'M0','tweetden':dict1[d][h],'class':rand}
else:
merged_obj = {'day':q,'hour':'H'+str(h),'minute':'M30','tweetden':dict2[d][h],'class':rand}
merged.insert_one(merged_obj).inserted_id