112 lines
3.5 KiB
Python
112 lines
3.5 KiB
Python
import sqlite3
|
|
|
|
|
|
def is_message_user_hatelisted(user_id: int) -> bool:
|
|
with sqlite3.connect("database/db.db") as conn:
|
|
c = conn.cursor()
|
|
c.execute("SELECT count(user_id) FROM message_users WHERE user_id = %d" % user_id)
|
|
result = c.fetchone()[0]
|
|
c.close()
|
|
conn.close()
|
|
return result > 0
|
|
|
|
|
|
def insert_message_hatelist(user_id: int):
|
|
with sqlite3.connect("database/db.db") as conn:
|
|
c = conn.cursor()
|
|
c.execute("INSERT INTO message_users(user_id) VALUES (%d)" % user_id)
|
|
conn.commit()
|
|
print(f"Пользователь {user_id} добавлен в хейтлист")
|
|
c.close()
|
|
conn.close()
|
|
|
|
|
|
def delete_message_hatelist(user_id: int):
|
|
with sqlite3.connect("database/db.db") as conn:
|
|
c = conn.cursor()
|
|
c.execute("DELETE FROM message_users WHERE user_id=%d" % user_id)
|
|
conn.commit()
|
|
print(f"Пользователь {user_id} удален из хейтлиста")
|
|
c.close()
|
|
conn.close()
|
|
|
|
|
|
def get_all_message_users():
|
|
with sqlite3.connect("database/db.db") as conn:
|
|
c = conn.cursor()
|
|
c.execute("SELECT * FROM message_users")
|
|
result = c.fetchall()
|
|
c.close()
|
|
conn.close()
|
|
return result
|
|
|
|
|
|
def is_photo_user_hatelisted(user_id: int) -> bool:
|
|
with sqlite3.connect("database/db.db") as conn:
|
|
c = conn.cursor()
|
|
c.execute("SELECT count(user_id) FROM photo_users WHERE user_id = %d" % user_id)
|
|
result = c.fetchone()[0]
|
|
c.close()
|
|
conn.close()
|
|
return result > 0
|
|
|
|
|
|
def insert_photo_hatelist(user_id: int):
|
|
with sqlite3.connect("database/db.db") as conn:
|
|
c = conn.cursor()
|
|
c.execute("INSERT INTO photo_users(user_id) VALUES (%d)" % user_id)
|
|
conn.commit()
|
|
print(f"Пользователь {user_id} добавлен в хейтлист")
|
|
c.close()
|
|
conn.close()
|
|
|
|
|
|
def delete_photo_hatelist(user_id: int):
|
|
with sqlite3.connect("database/db.db") as conn:
|
|
c = conn.cursor()
|
|
c.execute("DELETE FROM photo_users WHERE user_id=%d" % user_id)
|
|
conn.commit()
|
|
print(f"Пользователь {user_id} удален из хейтлиста")
|
|
c.close()
|
|
conn.close()
|
|
|
|
|
|
def get_all_photo_users():
|
|
with sqlite3.connect("database/db.db") as conn:
|
|
c = conn.cursor()
|
|
c.execute("SELECT * FROM photo_users")
|
|
result = c.fetchall()
|
|
c.close()
|
|
conn.close()
|
|
return result
|
|
|
|
|
|
def set_message_cooldown(message_cooldown: int, type_id: int):
|
|
with sqlite3.connect("database/db.db") as conn:
|
|
c = conn.cursor()
|
|
c.execute("UPDATE settings SET message_cooldown = %d WHERE id = %d" % (message_cooldown, type_id))
|
|
conn.commit()
|
|
print(f"Задержка текста изменена на {message_cooldown} секунд")
|
|
c.close()
|
|
conn.close()
|
|
|
|
|
|
def set_photo_cooldown(photo_cooldown: int, type_id: int):
|
|
with sqlite3.connect("database/db.db") as conn:
|
|
c = conn.cursor()
|
|
c.execute("UPDATE settings SET photo_cooldown = %d WHERE id = %d" % (photo_cooldown, type_id))
|
|
conn.commit()
|
|
print(f"Задержка фото изменена на {photo_cooldown} секунд")
|
|
c.close()
|
|
conn.close()
|
|
|
|
|
|
def get_settings(type_id: int):
|
|
with sqlite3.connect("database/db.db") as conn:
|
|
c = conn.cursor()
|
|
c.execute("SELECT * FROM settings WHERE id = %d" % type_id)
|
|
result = c.fetchone()
|
|
c.close()
|
|
conn.close()
|
|
return result
|