feat: bot start, database
This commit is contained in:
29
database.py
Normal file
29
database.py
Normal file
@@ -0,0 +1,29 @@
|
||||
import aiosqlite
|
||||
|
||||
class Database:
|
||||
def __init__(self, db_path: str):
|
||||
self.db_path = db_path
|
||||
|
||||
async def create_tables(self):
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
await db.execute("""
|
||||
CREATE TABLE IF NOT EXISTS users (
|
||||
id INTEGER PRIMARY KEY,
|
||||
user_id INTEGER UNIQUE
|
||||
)
|
||||
""")
|
||||
await db.commit()
|
||||
|
||||
async def add_user(self, user_id: int):
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
await db.execute(
|
||||
"INSERT OR IGNORE INTO users (user_id) VALUES (?)",
|
||||
(user_id,)
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
async def get_all(self):
|
||||
async with aiosqlite.connect(self.db_path) as db:
|
||||
async with db.execute("SELECT * FROM users") as cursor:
|
||||
rows = await cursor.fetchall()
|
||||
return rows
|
||||
Reference in New Issue
Block a user