def cursorToList(cursor): cursorList = [] for x in range(0, cursor.count(), 1): dico = cursor[x] dico["_id"] = "" cursorList.append(dico) return cursorList
def test_remove_expired(self): num_projects = 10 num_queues = 10 messages_per_queue = 100 projects = ['gc-test-project-{0}'.format(i) for i in range(num_projects)] queue_names = ['gc-test-{0}'.format(i) for i in range(num_queues)] client_uuid = 'b623c53c-cf75-11e2-84e1-a1187188419e' messages = [{'ttl': 0, 'body': str(i)} for i in range(messages_per_queue)] for project in projects: for queue in queue_names: self.queue_controller.create(queue, project) self.controller.post(queue, messages, client_uuid, project) self.controller.remove_expired() for project in projects: for queue in queue_names: query = {'q': queue, 'p': project} cursor = self.driver.db.messages.find(query) count = cursor.count() # Expect that the most recent message for each queue # will not be removed. self.assertEquals(count, 1) message = next(cursor) self.assertEquals(message['k'], messages_per_queue)
def add_user_post(): client = MongoClient() db = client["giw"] _id=request.forms.get("_id") country=request.forms.get("country") zip=request.forms.get("zip") email=request.forms.get("email") gender=request.forms.get("gender") likes=request.forms.get("likes").split(",") password=request.forms.get("password") year=request.forms.get("year") cursor = db.users.find({"_id": _id}) if cursor.count() > 0: print "Fallo al insertar. Usuario existente." return "Usuario existente" db.users.insert_one( { "_id": _id , "address": { "country": country, "zip": zip }, "email": email, "gender": gender, "likes":likes, "password":password, "year":year}) print "Usuario insertado con exito." return "Usuario insertado con exito."
def add_user_post(): client = MongoClient() db = client["giw"] _id = request.forms.get("_id") country = request.forms.get("country") zip = request.forms.get("zip") email = request.forms.get("email") gender = request.forms.get("gender") likes = request.forms.get("likes").split(",") password = request.forms.get("password") year = request.forms.get("year") cursor = db.users.find({"_id": _id}) if cursor.count() > 0: print "Fallo al insertar. Usuario existente." return "Usuario existente" db.users.insert_one({ "_id": _id, "address": { "country": country, "zip": zip }, "email": email, "gender": gender, "likes": likes, "password": password, "year": year }) print "Usuario insertado con exito." return "Usuario insertado con exito."
def getAnswer(request, ide): # ide is questionID try: answer_json = global_answer_class.data.find( {'questionID': ide}) # get the users answers. question_json = global_question_class.getDictById( ide) # get the question of the answer if question_json['type'] == 1: answers = range(question_json['answers']['min'], question_json['answers']['max'] + 1, question_json['answers']['step']) else: answers = question_json['answers'] # get answer choices data = {} data["questionID"] = question_json['id'] data["text"] = question_json['text'] data["images"] = question_json['images'] data["video"] = question_json['video'] list_answers = [] # data output print(answers) for j in answers: cursor = global_answer_class.data.find({'answers': j}) print(cursor.count()) user = [] for i in cursor: user.append(i['userID']) list_answers.append({'value': j, 'users': user}) data["answers"] = list_answers return JsonResponse(data, safe=False) # return data except Exception as e: traceback.print_exc() return output('False')
def _handle_cursor(self, cursor): """ if a plugin-function returns a database-cursor, this function extracts the documents and packs them into a list. """ result_size = cursor.count() data = [] for result in cursor: data.append(result) return data
def getAllByRole(request, rooole): try: cursor = global_account_class.data.find({"role": rooole}) cursorList = [] for x in range(0, cursor.count(), 1): dico = cursor[x] dico["_id"] = "" cursorList.append(dico) return JsonResponse(cursorList, safe=False) except Exception as e: print(e) return output('False')
def insert_or_update(): client = MongoClient() db = client["giw"] # Getting post variables _id = request.forms.get("_id") country = request.forms.get("country") zipValue = request.forms.get("zip") email = request.forms.get("email") gender = request.forms.get("gender") likes = request.forms.get("likes").split(",") password = request.forms.get("password") year = request.forms.get("year") cursor = db.users.find({"_id": _id}) if cursor.count() > 0: ret = "Usuario encontrado. Actualizando usuario" db.users.update_one({"_id": _id}, { "$set": { "_id": _id, "address": { "country": country, "zip": zipValue, }, "email": email, "gender": gender, "likes": likes, "password": password, "year": year } }) else: ret = "Usuario no encontrado. Insertando usuario" db.users.insert_one({ "_id": _id, "address": { "country": country, "zip": zipValue, }, "email": email, "gender": gender, "likes": likes, "password": password, "year": year }) c.close() return ret
def getAllAnswersByQuestionId(request, questionID): try: cursor = global_answer_class.data.find({"questionID": questionID}) cursorList = [] for x in range(0, cursor.count(), 1): dico = cursor[x] dico["_id"] = "" cursorList.append(dico) return JsonResponse(cursorList, safe=False) except Exception as e: print(e) return output('False')
def insert_or_update(): client = MongoClient() db = client["giw"] # Getting post variables _id = request.forms.get("_id") country = request.forms.get("country") zipValue = request.forms.get("zip") email = request.forms.get("email") gender = request.forms.get("gender") likes = request.forms.get("likes").split(",") password = request.forms.get("password") year = request.forms.get("year") cursor = db.users.find({"_id": _id}) if cursor.count() > 0: ret = "Usuario encontrado. Actualizando usuario" db.users.update_one({"_id": _id}, { "$set": {"_id": _id, "address": {"country": country, "zip": zipValue, }, "email": email, "gender": gender, "likes":likes, "password":password, "year": year} }) else: ret = "Usuario no encontrado. Insertando usuario" db.users.insert_one({ "_id": _id, "address": {"country": country, "zip": zipValue, }, "email": email, "gender": gender, "likes":likes, "password":password, "year": year} ) c.close() return ret