mafia_api/app.py

from flask import Flask, request, jsonify
import db
import api_functions

app = Flask(__name__)


@app.route('/action', methods=["POST"])
def action():
    # TODO: обработка ошибок + логи
    data = request.json
    true_chat_id = data['chatID']
    killed_user_id = data['KilledUserID']
    healed_user_id = data['HealedUserID']
    checked_user_id = data['CheckedUserID']

    response = {
        "killedUserId": None,
        "isEndGame": None,
        "checkedRole": False
    }

    cursor = db.cursor

    if killed_user_id != healed_user_id:
        cursor.execute(f"Select UserID from chats join userschats uc on uc.chatid = chats.chatid join "
                    f"users on users.userid = uc.userid where truechatid = '{true_chat_id}' "
                    f"and TrueUserID = '{killed_user_id}'")
        generated_user_id = cursor.fetchall()[0]
        cursor.execute(f"UPDATE Users SET isAlive = false WHERE UserID = '{generated_user_id}'")
        response["killedUserId"] = killed_user_id

    cursor.execute(f"SELECT role FROM users WHERE TrueUserID ='{checked_user_id}';")
    role = cursor.fetchall()[0]

    if role == 4:
        response["checkedRole"] = True

    response["isEndGame"] = api_functions.isEnd(cursor, true_chat_id)
    db.close(cursor)

    return jsonify(response), 200


@app.route('/create_chat', methods=["POST"])
def create():
    # TODO: обработка ошибок + логи
    tci = request.json['chatID']
    cursor = db.connection.cursor()
    db.query_ins_chat(tci, cursor)
    db.close(cursor)
    return 'ok', 200


@app.route('/start_game', methods=["POST"])
def start():  # Команда принимает на вход
    tci = request.json['chatID']
    user_list = request.json['userIDs']
    cursor = db.connection.cursor()
    for current_user in user_list:
        db.query_ins_user(current_user, cursor)
    db.close(cursor)
    resp = api_functions.role_distribution(user_list)
    return "", resp


if __name__ == '__main__':
    app.run(host="0.0.0.0", debug=True)