import sys, random, time, moment
from datetime import datetime
from db import CDatabase
#from msgdelivery import sendJoin
mydb = CDatabase()
res = mydb.buildConnection()
res = mydb.selectCollection("xmateHistoryPost")
mydb.removeData({})
print list(mydb.getData({})["content"])
res = mydb.selectCollection("xmatePost")
mydb.removeData({})
print list(mydb.getData({})["content"])
res = mydb.selectCollection("xmateUser")
mydb.removeData({})
print list(mydb.getData({})["content"])
res = mydb.selectCollection("xmateMessage")
mydb.removeData({})
print list(mydb.getData({})["content"])
#print list(mydb.getData({})["content"])
#match_list = {"fbid":"3"}
#ndata = {"gender":"female", "age":28, "history_partner":[], "schedule_list":[],"history_events":[],"unprocessed_message":[],"fbid":"3"}
#res = mydb.updateData({},{"conflict_list":[]})
#print list(mydb.getData(match_list)["content"])[0]
def generateSchedule():
user_db = CDatabase()
aa = user_db.buildConnection()
user_db.selectCollection('xmateUser')
# print(aa)
# get all users in a list
users = user_db.getData({})['content']
user_db.closeConnection()
# exercise type
exer_type = ["DoTA","Hotpot","Tennis", "Coding","Running"]
# initialize schedule dict and history dict to find participating users
scheduleMem = {}
historyMem = {}
for user in users:
uid = user['uid']
if "schedule_list" in user:
# print(user['schedule_list'])
scheduleList = user['schedule_list']
for schedule in scheduleList:
if schedule in scheduleMem:
scheduleMem[schedule].append(uid)
else:
scheduleMem[schedule] = [uid]
if "history_events" in user:
# print(user['history_events'])
historyList = user['history_events']
for history in historyList:
if history in historyMem:
historyMem[history].append(uid)
else:
historyMem[history] = [uid]
# print(scheduleMem)
# print(historyMem)
schedule_res = []
history_res = []
# randomly generate schedule and history schedule
for i in range(2000):
# schedule
if i in scheduleMem:
schedule = {}
schedule['sid'] = i
schedule['type'] = random.choice(exer_type)
schedule['status'] = "coming"
# latitude range for ithaca [42.400000, 42.500000]
# longitude range for ithaca[-76.560000,-76.410000]
schedule['location'] = {"city":"Ithaca"}
schedule['location']['latitude'] = random.uniform(42.40,42.50)
schedule['location']['longitude'] = random.uniform(-76.56, -76.41)
# time, within the past week
schedule['post_datetime'] = getNextDateTime(1456622834.756554, 10080)
schedule['time_range'] = {}
schedule['time_range']['start_time'], schedule['time_range']['end_time'] = getExerRange(schedule['post_datetime'],86400, 7200)
# member
schedule['member'] = scheduleMem[i]
schedule['owner'] = schedule['member'].pop(0)
schedule['creator'] = schedule['owner']
schedule['related_member'] = []
schedule_res.append(schedule)
print(schedule)
# history
if i in historyMem:
history = {}
history['sid'] = i
history['type'] = random.choice(exer_type)
# Location
history['location'] = {"city":"Ithaca"}
history['location']['latitude'] = random.uniform(42.40,42.50)
history['location']['longitude'] = random.uniform(-76.56, -76.41)
# Time
history['post_datetime'] = getNextDateTime(1455413234.756554, 10080)
history['time_range'] = {}
history['time_range']['start_time'], history['time_range']['end_time'] = getExerRange(schedule['post_datetime'],86400, 7200)
# member
history['member'] = historyMem[i]
history['owner'] = history['member'].pop(0)
history['creator'] = history['owner']
history_res.append(history)
print(history)
print(len(schedule_res))
print(len(history_res))
# write to schedule db
schedule_db = CDatabase()
schedule_db.buildConnection()
schedule_db.selectCollection('xmatePost')
schedule_db.insertManyData(schedule_res)
schedule_db.closeConnection()
# write to history schedule db
history_db = CDatabase()
history_db.buildConnection()
history_db.selectCollection('xmateHistoryPost')
history_db.insertManyData(history_res)
history_db.closeConnection()