def set_up_trips(list_of_cluster_data, user_id):
# Import in here to avoid recursive imports
# TODO: This should really be moved to a separate class that creates the
# entire graph at one time
import emission.storage.decorations.common_place_queries as esdcpq
clear_existing_trips(user_id)
for dct in list_of_cluster_data:
start_loc = gj.Point(dct['start_coords'].coordinate_list())
end_loc = gj.Point(dct['end_coords'].coordinate_list())
start_place_id = esdcpq.get_common_place_at_location(start_loc).get_id()
end_place_id = esdcpq.get_common_place_at_location(end_loc).get_id()
#print 'dct["sections"].trip_id %s is' % dct["sections"][0]
probabilites = np.zeros((DAYS_IN_WEEK, HOURS_IN_DAY))
for sec in dct["sections"]:
probabilites[get_day(sec), get_start_hour(sec)] += 1
trip = make_new_common_trip()
trip.user_id = user_id
trip.start_place = start_place_id
trip.end_place = end_place_id
trip.start_loc = start_loc
trip.end_loc = end_loc
trip.probabilites = probabilites
trip.trips = [unc_trip.get_id() for unc_trip in dct["sections"]]
place_db = edb.get_place_db()
save_common_trip(trip)
def set_up_trips(list_of_cluster_data, user_id):
# Import in here to avoid recursive imports
# TODO: This should really be moved to a separate class that creates the
# entire graph at one time
import emission.storage.decorations.common_place_queries as esdcpq
clear_existing_trips(user_id)
for dct in list_of_cluster_data:
start_times = []
durations = []
start_loc = gj.Point(dct['start_coords'])
end_loc = gj.Point(dct['end_coords'])
start_place_id = esdcpq.get_common_place_at_location(
start_loc).get_id()
end_place_id = esdcpq.get_common_place_at_location(end_loc).get_id()
#print 'dct["sections"].trip_id %s is' % dct["sections"][0]
probabilites = np.zeros((DAYS_IN_WEEK, HOURS_IN_DAY))
for sec in dct["sections"]:
probabilites[get_day(sec), get_start_hour(sec)] += 1
start_times.append(get_start_time(sec))
durations.append(sec.data.duration)
trip = make_new_common_trip()
trip.user_id = user_id
trip.start_place = start_place_id
trip.end_place = end_place_id
trip.start_loc = start_loc
trip.end_loc = end_loc
trip.probabilites = probabilites
trip.trips = [unc_trip.get_id() for unc_trip in dct["sections"]]
trip.start_times = start_times
trip.durations = durations
save_common_trip(trip)
def setUp(self):
edb.get_common_place_db().drop()
edb.get_common_trip_db().drop()
self.testUserId = uuid.uuid4()
self.testLocation = gj.Point((122.1234, 37.1234))
self.testEnd = esdcpq.make_new_common_place(uuid.uuid4(), gj.Point((1,2.092)))
esdcpq.save_common_place(self.testEnd)
self.testEnd = esdcpq.get_common_place_at_location(self.testEnd.location)
self.testEnd.successors = ()
self.testStart = esdcpq.make_new_common_place(uuid.uuid4(), gj.Point((1,2)))
self.testStart.successors = (self.testEnd.get_id(),)
esdcpq.save_common_place(self.testEnd)
esdcpq.save_common_place(self.testStart)
self.time0 = datetime.datetime(1900, 1, 1, 1)
def setUp(self):
# self.clearRelatedDb()
self.testUserId = uuid.uuid4()
self.testEnd = esdcpq.make_new_common_place(uuid.uuid4(),
gj.Point((1, 2.092)))
esdcpq.save_common_place(self.testEnd)
self.testEnd = esdcpq.get_common_place_at_location(
self.testEnd.location)
self.testEnd.successors = ()
self.testStart = esdcpq.make_new_common_place(uuid.uuid4(),
gj.Point((1, 2)))
self.testStart.successors = (self.testEnd.get_id(), )
esdcpq.save_common_place(self.testEnd)
esdcpq.save_common_place(self.testStart)
self.time0 = datetime.datetime(1900, 1, 1, 1)
