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
