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)