def check_status(): # name = request.json['name'] # address = request.json['address'] # mobile = request.json['mobile'] id = request.json['user_id'] user = User.User.query.get(id) if user: user_schema = User.UserSchema() return user_schema.jsonify(user) return {"message": "User doesn't exist in database"}, 404
def add_user(): # name = request.json['name'] id = request.json['user_id'] # address = request.json['address'] # mobile = request.json['mobile'] exists = User.User.query.get(id) if not exists: new_user = User.User(id=id) db.session.add(new_user) db.session.commit() message = "Successfully created user: {}".format(id) user_schema = User.UserSchema() data = user_schema.dump(new_user) data['message'] = message return data, 201 else: user_schema = User.UserSchema() data = user_schema.dump(exists) data['message'] = "User is already in the database" return data, 200
def update_infection(): # name = request.json['name'] # address = request.json['address'] # mobile = request.json['mobile'] id = request.json['user_id'] status = request.json['status'] is_diseased = True if (status and status.lower() == "infected") else False user = User.User.query.get(id) if not user: return {"message": "User doesn't exist in database"}, 404 user.is_diseased = is_diseased db.session.commit() send_to_queue(id) user_schema = User.UserSchema() data = user_schema.dump(user) data['message'] = "User infection status updated" return data, 200