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()
Esempio n. 2
0
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)
Esempio n. 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 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())
Esempio n. 5
0
 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})
Esempio n. 10
0
def clear_existing_places(user_id):
    db = edb.get_common_place_db()
    db.remove({'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)
Esempio n. 13
0
def save_common_place(common_place):
    edb.save(edb.get_common_place_db(), common_place)
Esempio n. 14
0
def save_common_place(common_place):
    db = edb.get_common_place_db()
    db.save(common_place)
Esempio n. 15
0
 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()
Esempio n. 19
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)
Esempio n. 20
0
def get_common_place_at_location(loc):
    db = edb.get_common_place_db()
    return make_common_place(db.find_one({"location": loc}))
Esempio n. 21
0
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)
Esempio n. 22
0
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]
Esempio n. 23
0
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())