From b80e790ddc3777e7c041bdbbef4dcb29f4810315 Mon Sep 17 00:00:00 2001
From: ada-dmitry <ada.dmitry@gmail.com>
Date: Tue, 22 Aug 2023 16:02:01 +0300
Subject: [PATCH] ready create

---
 app.py |  8 +++++---
 db.py  | 23 +++++++++++++----------
 2 files changed, 18 insertions(+), 13 deletions(-)

diff --git a/app.py b/app.py
index a69f6be..5825b32 100644
--- a/app.py
+++ b/app.py
@@ -1,5 +1,5 @@
 from flask import Flask, request
-# import db
+import db
 
 app = Flask(__name__)
 
@@ -11,8 +11,10 @@ def hello_world():  # put application's code here
 
 @app.route('/create_game', methods=["POST"])
 def create():
-    TCI = request.json
-    db.query_ins_chat(TCI)
+    TCI = request.json['chatID']
+    curs = db.conn()
+    db.query_ins_chat(TCI, curs)
+    db.close(curs)
 
 
 @app.route('/start_game')
diff --git a/db.py b/db.py
index b14e3fa..5651983 100644
--- a/db.py
+++ b/db.py
@@ -8,30 +8,34 @@ db_params = {
     'password': '4oumQk'
 }
 
-connection = psycopg2.connect(**db_params)
-cursor = connection.cursor()
+# connection = psycopg2.connect(**db_params) #запускаем один раз
 
 
-def query_ins_user(x, curs=cursor):
+def conn(conn=connection):
+    cursor = conn.cursor()
+    return cursor
+
+
+def query_ins_user(tui, curs=cursor):
     try:
         # Запрос добавления пользователя
         curs.execute(
-            f"""INSERT into public.Users(TrueUserID, role, isAlive) values ({x}, 0)""")
-        curs.commit()
+            f"""INSERT into public.Users(TrueUserID, role, isAlive) values ({tui}, 0)""")
+        conn.commit()
     except Exception as e:
         print("Ошибка:", e)
 
 
-def query_ins_chat(x, curs=cursor):
+def query_ins_chat(tci, curs=cursor):
     try:  # Запрос добавления чата
         curs.execute(
-            f"""INSERT into public.Chats(TrueChatID, StateID) values ({x}, 0)""")
-        curs.commit()
+            f"""INSERT into public.Chats(TrueChatID, StateID) values ({tci}, 0)""")
+        conn.commit()
     except Exception as e:
         print("Ошибка:", e)
 
 
-def query_sel(x, curs=cursor):
+def query_sel(query, curs=cursor):
     try:
         # Запрос вывода из БД
         curs.execute("SELECT * FROM chats")
@@ -51,4 +55,3 @@ def query_change(x):
 
 def close(cur=cursor, conn=connection):
     cursor.close()
-    connection.close()