def tearDown(self): edb.get_timeseries_db().remove({"user_id": self.testUUID}) edb.get_analysis_timeseries_db().remove({"user_id": self.testUUID}) edb.get_timeseries_db().remove({"user_id": "new_fake"}) edb.get_analysis_timeseries_db().remove({"user_id": "new_fake"}) edb.get_common_trip_db().drop() edb.get_common_place_db().drop()
def del_objects(args): del_query = {} if args.user_id != "all": del_query['user_id'] = uuid.UUID(args.user_id) if args.date is None: print("Deleting all analysis information for query %s" % del_query) print edb.get_analysis_timeseries_db().remove(del_query) print edb.get_common_place_db().remove(del_query) print edb.get_common_trip_db().remove(del_query)
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 get_common_trip_from_db(user_id, start_place_id, end_place_id): db = edb.get_common_trip_db() json_obj = db.find_one({ "user_id": user_id, "start_place": start_place_id, "end_place": end_place_id }) return make_common_trip_from_json(json_obj)
def save_common_trip(common_trip): db = edb.get_common_trip_db() probs = _2d_array_to_mongo_format(common_trip.probabilites) db.insert({ "user_id" : common_trip.user_id, "start_place" : common_trip.start_place, "end_place" : common_trip.end_place, "start_loc" : common_trip.start_loc, "end_loc" : common_trip.end_loc, "trips" : common_trip["trips"], "probabilites" : probs })
def save_common_trip(common_trip): db = edb.get_common_trip_db() probs = _2d_array_to_mongo_format(common_trip.probabilites) db.insert({ "user_id": common_trip.user_id, "start_place": common_trip.start_place, "end_place": common_trip.end_place, "start_loc": common_trip.start_loc, "end_loc": common_trip.end_loc, "trips": common_trip["trips"], "probabilites": probs })
def save_common_trip(common_trip): db = edb.get_common_trip_db() probs = _np_array_to_json_format(common_trip.probabilites) db.insert({ "user_id": common_trip.user_id, "start_place": common_trip.start_place, "end_place": common_trip.end_place, "start_loc": common_trip.start_loc, "end_loc": common_trip.end_loc, "trips": common_trip["trips"], "probabilites": probs, "start_times": common_trip["start_times"], "durations": common_trip["durations"] })
def save_common_trip(common_trip): db = edb.get_common_trip_db() probs = _np_array_to_json_format(common_trip.probabilites) db.insert({ "user_id" : common_trip.user_id, "start_place" : common_trip.start_place, "end_place" : common_trip.end_place, "start_loc" : common_trip.start_loc, "end_loc" : common_trip.end_loc, "trips" : common_trip["trips"], "probabilites" : probs, "start_times": common_trip["start_times"], "durations": common_trip["durations"] })
def get_all_common_trips_for_user(user_id): db = edb.get_common_trip_db() return db.find({"user_id" : user_id})
def get_common_trip_from_db(user_id, start_place_id, end_place_id): db = edb.get_common_trip_db() json_obj = db.find_one({"user_id" : user_id, "start_place" : start_place_id, "end_place" : end_place_id}) return make_common_trip_from_json(json_obj)
def get_common_trips(user_id): db = edb.get_common_trip_db() trip_jsons = db.find({"user_id" : user_id}) return [esdctq.make_new_common_trip(trip_json) for trip_json in trip_jsons]
def tearDown(self): edb.get_common_trip_db().remove({'user_id': self.testUserId}) edb.get_analysis_timeseries_db().remove({'user_id': self.testUserId}) edb.get_common_place_db().remove({'user_id': self.testUserId})
def purgeAnalysisData(): print edb.get_analysis_timeseries_db().remove() print edb.get_common_place_db().remove() print edb.get_common_trip_db().remove() print edb.get_pipeline_state_db().remove()
reset_collection(edb.get_analysis_timeseries_db(), user.uuid, new_uuid) logging.debug("Resetting client...") reset_collection(edb.get_client_db(), user.uuid, new_uuid) logging.debug("Resetting client_stats_backup...") reset_collection(edb.get_client_stats_db_backup(), user.uuid, new_uuid) logging.debug("Resetting server_stats_backup...") reset_collection(edb.get_server_stats_db_backup(), user.uuid, new_uuid) logging.debug("Resetting result_stats_backup...") reset_collection(edb.get_result_stats_db_backup(), user.uuid, new_uuid) logging.debug("Resetting edb.get_common_place_db...") reset_collection(edb.get_common_place_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_common_trip_db...") reset_collection(edb.get_common_trip_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_habitica_db...") reset_collection(edb.get_habitica_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_pipeline_state_db...") reset_collection(edb.get_pipeline_state_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_profile_db...") reset_collection(edb.get_profile_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_timeseries_db...") reset_collection(edb.get_timeseries_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_timeseries_error_db...") reset_collection(edb.get_timeseries_error_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_usercache_db...") reset_collection(edb.get_usercache_db(), user.uuid, new_uuid)
def purgeAnalysisData(): print(edb.get_analysis_timeseries_db().remove()) print(edb.get_common_place_db().remove()) print(edb.get_common_trip_db().remove()) print(edb.get_pipeline_state_db().remove())
edb.get_uuid_db().update({"uuid" : user.uuid}, {"$set": {"uuid" : new_uuid}}) logging.debug("Resetting alternatives...") reset_collection(edb.get_alternatives_db(), user.uuid, new_uuid) logging.debug("Resetting analysis...") reset_collection(edb.get_analysis_timeseries_db(), user.uuid, new_uuid) logging.debug("Resetting client...") reset_collection(edb.get_client_db(), user.uuid, new_uuid) logging.debug("Resetting client_stats_backup...") reset_collection(edb.get_client_stats_db_backup(), user.uuid, new_uuid) logging.debug("Resetting server_stats_backup...") reset_collection(edb.get_server_stats_db_backup(), user.uuid, new_uuid) logging.debug("Resetting result_stats_backup...") reset_collection(edb.get_result_stats_db_backup(), user.uuid, new_uuid) logging.debug("Resetting edb.get_common_place_db...") reset_collection(edb.get_common_place_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_common_trip_db...") reset_collection(edb.get_common_trip_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_habitica_db...") reset_collection(edb.get_habitica_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_pipeline_state_db...") reset_collection(edb.get_pipeline_state_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_profile_db...") reset_collection(edb.get_profile_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_timeseries_db...") reset_collection(edb.get_timeseries_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_timeseries_error_db...") reset_collection(edb.get_timeseries_error_db(), user.uuid, new_uuid) logging.debug("Resetting edb.get_usercache_db...") reset_collection(edb.get_usercache_db(), user.uuid, new_uuid)
def get_common_trips(user_id): db = edb.get_common_trip_db() trip_jsons = db.find({"user_id": user_id}) return [esdctq.make_new_common_trip(trip_json) for trip_json in trip_jsons]
def tearDown(self): edb.get_common_trip_db().drop() edb.get_section_new_db().drop() edb.get_trip_new_db().drop()
def clear_existing_trips(user_id): db = edb.get_common_trip_db() db.remove({'user_id': user_id})
def setUp(self): edb.get_common_trip_db().drop() edb.get_section_new_db().drop() edb.get_trip_new_db().drop()
def tearDown(self): edb.get_common_trip_db().drop() edb.get_common_place_db().drop()
def get_all_common_trips_for_user(user_id): db = edb.get_common_trip_db() return db.find({"user_id": user_id})