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
