основа

modules/sqlite_methods.py

@@ -0,0 +1,111 @@
+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