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 testGetSuccessor(self): logging.debug("size of db is %s" % edb.get_common_place_db().find().count()) self.assertIsNotNone(edb.get_common_place_db().find_one({"_id": self.testEnd.get_id()})) probs = np.zeros( (7, 24) ) probs[self.time0.weekday(), 3] = 10 props = { "user_id" : self.testUserId, "start_place" : self.testStart.get_id(), "end_place" : self.testEnd.get_id(), "start_loc" : self.testStart.location, "end_loc" : self.testEnd.location, "probabilites" : probs, "trips" : () } trip = esdctp.make_new_common_trip(props) esdctp.save_common_trip(trip) suc = esdcpq.get_succesor(self.testUserId, self.testStart.get_id(), self.time0) self.assertEqual(suc, self.testEnd.get_id())
def testGetSuccessor(self): logging.debug("size of db is %s" % edb.get_common_place_db().find().count()) self.assertIsNotNone(edb.get_common_place_db().find_one({"_id": self.testEnd.get_id()})) probs = np.zeros( (7, 24) ) probs[self.time0.weekday(), 3] = 10 props = { "user_id" : self.testUserId, "start_place" : self.testStart.get_id(), "end_place" : self.testEnd.get_id(), "start_loc" : self.testStart.location, "end_loc" : self.testEnd.location, "probabilites" : probs, "trips" : (), "start_times": [], "durations": [] } trip = esdctp.make_new_common_trip(props) esdctp.save_common_trip(trip) suc = esdcpq.get_succesor(self.testUserId, self.testStart.get_id(), self.time0) self.assertEqual(suc, self.testEnd.get_id())
def get_common_place_at_location(loc): db = edb.get_common_place_db() return make_common_place(db.find_one({"location": loc}))
def clear_existing_places(user_id): db = edb.get_common_place_db() db.remove({'user_id': user_id})
def get_common_place_from_db(common_place_id): db = edb.get_common_place_db() json_obj = db.find_one({"_id" : common_place_id}) return make_common_place(json_obj)
def get_all_common_places_for_user(user_id): db = edb.get_common_place_db() return db.find({"user_id" : user_id})
def save_common_place(common_place): db = edb.get_common_place_db() db.save(common_place)
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 save_common_place(common_place): edb.save(edb.get_common_place_db(), common_place)
def tearDown(self): edb.get_common_trip_db().drop() edb.get_common_place_db().drop()
def get_common_places(user_id): db = edb.get_common_place_db() place_jsons = db.find({"user_id" : user_id}) return [esdcpq.make_common_place(place_json) for place_json in place_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()
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_place_from_db(common_place_id): db = edb.get_common_place_db() json_obj = db.find_one({"_id": common_place_id}) return make_common_place(json_obj)
def get_common_places(user_id): db = edb.get_common_place_db() place_jsons = db.find({"user_id": user_id}) return [esdcpq.make_common_place(place_json) for place_json in place_jsons]
def get_all_common_places_for_user(user_id): db = edb.get_common_place_db() return db.find({"user_id": user_id})
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())