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)
Beispiel #3
0
    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
        }) 
Beispiel #6
0
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())
Beispiel #16
0
            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)
Beispiel #17
0
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 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 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()
Beispiel #23
0
 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})