def get_images(userid): user = User.objects(id=userid).first() collect = request.args.get("collect") if collect == 'true': imgs = Image.objects(Q(user=user) & Q(isCollect=True)) elif collect == 'false': imgs = Image.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) img = Image.objects(id=imageId).first() if request.method == 'PUT': img.isCollect = request.json.get('collect') img.save() elif request.method == 'DELETE': img.delete() return jsonify({ "message": 'OK', "data": { "id": str(img.id), "collect": img.isCollect } })