mirror of
https://github.com/IgorL123/mafia_api.git
synced 2026-09-24 07:30:15 +00:00
48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
import psycopg2
|
|
|
|
|
|
db_params = {
|
|
'host': '51.250.68.42',
|
|
'database': 'test',
|
|
'user': 'curator',
|
|
'password': '4oumQk'
|
|
}
|
|
|
|
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)
|
|
|
|
|
|
|
|
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()
|
|
|