diff --git a/app.py b/app.py index 18f1e43..8f0d75e 100644 --- a/app.py +++ b/app.py @@ -1,4 +1,4 @@ -from flask import Flask +from flask import Flask, request app = Flask(__name__) @@ -7,16 +7,16 @@ app = Flask(__name__) def hello_world(): # put application's code here return 'Hello World!' -@app.route('/create_game') +@app.route('/create_game', method = ["POST"]) def create(str truechatid): - + @app.route('/start_game') def start(): # Команда принимает на вход - + @app.route('/action') def action(): diff --git a/db.py b/db.py index ad26ec7..7e500e4 100644 --- a/db.py +++ b/db.py @@ -8,21 +8,40 @@ db_params = { 'password': '4oumQk' } +connection = psycopg2.connect(**db_params) +cursor = connection.cursor() -try: - connection = psycopg2.connect(**db_params) - cursor = connection.cursor() +def query_ins_user(str x): + try: + # Запрос добавления пользователя + cursor.execute(f"""INSERT into public.Users(TrueUserID, role, isAlive) values ({x}, 0)""") + cursor.commit() + except Exception as e: + print("Ошибка:", e) + - # Ваш код выполнения запросов +def query_ins_chat(str x): + try: # Запрос добавления чата + cursor.execute(f"""INSERT into public.Chats(TrueChatID, StateID) values ({x}, 0)""") + cursor.commit() + except Exception as e: + print("Ошибка:", e) + - cursor.execute("SELECT * FROM chats") - rows = cursor.fetchall() - for row in rows: - print(row) +def query_sel(str x): + try: + # Запрос вывода из БД + cursor.execute("SELECT * FROM chats") + rows = cursor.fetchall() + # for row in rows: + # print(row) + + except Exception as e: + print("Ошибка:", e) + +def close(cur = cursor, conn = connection): cursor.close() connection.close() -except Exception as e: - print("Ошибка:", e) \ No newline at end of file