32 lines
896 B
Python
32 lines
896 B
Python
import database
|
|
|
|
|
|
class TgDatabase(database.BasicDatabase):
|
|
def __init__(self):
|
|
super().__init__('tg.db')
|
|
|
|
self.cursor.execute("""
|
|
CREATE TABLE IF NOT EXISTS chats (
|
|
"id" INTEGER,
|
|
"active" INTEGER NOT NULL DEFAULT 0,
|
|
"rules" TEXT,
|
|
"greeting_join" TEXT,
|
|
"ai_prompt" TEXT,
|
|
PRIMARY KEY("id"))
|
|
""")
|
|
|
|
self.cursor.execute("""
|
|
CREATE TABLE IF NOT EXISTS users (
|
|
"chat_id" INTEGER,
|
|
"user_id" INTEGER,
|
|
"last_message" INTEGER NOT NULL DEFAULT 0,
|
|
"messages_today" INTEGER NOT NULL DEFAULT 0,
|
|
"messages_month" INTEGER NOT NULL DEFAULT 0,
|
|
"warnings" INTEGER NOT NULL DEFAULT 0,
|
|
PRIMARY KEY("chat_id","user_id"))
|
|
""")
|
|
|
|
self.conn.commit()
|
|
|
|
|
|
DB = TgDatabase()
|