File size: 902 Bytes
21bc372 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import sqlite3
conn = sqlite3.connect('bot_prefix.db')
cursor = conn.cursor()
cursor.execute('''
CREATE TABLE IF NOT EXISTS prefixes (
user_id INTEGER PRIMARY KEY,
prefix TEXT NOT NULL
)
''')
conn.commit()
conn.close()
async def set_prefix_in_db(user_id: int, prefix: str):
conn = sqlite3.connect('bot_prefix.db')
cursor = conn.cursor()
cursor.execute('''
INSERT INTO prefixes (user_id, prefix)
VALUES (?, ?)
ON CONFLICT(user_id) DO UPDATE SET prefix=excluded.prefix
''', (user_id, prefix))
conn.commit()
conn.close()
async def get_prefix(user_id: int):
conn = sqlite3.connect('bot_prefix.db')
cursor = conn.cursor()
cursor.execute('SELECT prefix FROM prefixes WHERE user_id=?', (user_id,))
result = cursor.fetchone()
conn.close()
if result:
return result[0]
else:
return None |