def get_images(userid):
user = User.objects(id=userid).first()
collect = request.args.get("collect")
if collect == 'true':
imgs = Img.objects(Q(user=user) & Q(is_collected=True))
elif collect == 'false':
imgs = Img.objects(user=user)
page = int(request.args.get("page"))
per_page = int(request.args.get("per_page"))
paginated_imgs = imgs.skip((page - 1) * per_page).limit(per_page)
return jsonify({
"message": 'OK',
"data": {
"total_count": imgs.count(),
"page": page,
"per_page": per_page,
"results": paginated_imgs.to_public_json()
}
})
def collectImage(userid, imageId):
print(request.method)
if request.method == 'PUT':
img = Img.objects(id=imageId).first()
img.is_collected = request.json.get('collect')
img.save()
return jsonify({
"message": 'OK',
"data": {
"id": str(img.id),
"collect": img.is_collected
}
})
elif request.method == 'DELETE':
img = Img.objects(id=imageId).first()
img.delete()
return jsonify({
"message": 'OK',
"data": {
"id": str(img.id),
"collect": img.is_collected
}
})
