Vlad
28 lines
906 B
Python
28 lines
906 B
Python
import aiosqlite
|
|
from datetime import datetime
|
|
|
|
DATABASE_PATH = 'database.db'
|
|
|
|
async def init_db():
|
|
async with aiosqlite.connect(DATABASE_PATH) as db:
|
|
await db.execute('''
|
|
CREATE TABLE IF NOT EXISTS connections (
|
|
tg_id INTEGER NOT NULL,
|
|
client_id TEXT NOT NULL,
|
|
email TEXT NOT NULL,
|
|
expiry_time INTEGER NOT NULL,
|
|
PRIMARY KEY (tg_id, client_id)
|
|
)
|
|
''')
|
|
await db.commit()
|
|
|
|
async def add_connection(tg_id: int, client_id: str, email: str, expiry_time: int):
|
|
async with aiosqlite.connect(DATABASE_PATH) as db:
|
|
# Добавляем новый ключ
|
|
await db.execute('''
|
|
INSERT INTO connections (tg_id, client_id, email, expiry_time)
|
|
VALUES (?, ?, ?, ?)
|
|
''', (tg_id, client_id, email, expiry_time))
|
|
await db.commit()
|
|
|