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
|
from home.database.sqlite import SQLiteBase
class BotDatabase(SQLiteBase):
def __init__(self):
super().__init__()
def schema_init(self, version: int) -> None:
if version < 1:
cursor = self.cursor()
cursor.execute("""CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY,
lang TEXT NOT NULL
)""")
self.commit()
def get_user_lang(self, user_id: int, default: str = 'en') -> str:
cursor = self.cursor()
cursor.execute('SELECT lang FROM users WHERE id=?', (user_id,))
row = cursor.fetchone()
if row is None:
cursor.execute('INSERT INTO users (id, lang) VALUES (?, ?)', (user_id, default))
self.commit()
return default
else:
return row[0]
def set_user_lang(self, user_id: int, lang: str) -> None:
cursor = self.cursor()
cursor.execute('UPDATE users SET lang=? WHERE id=?', (lang, user_id))
self.commit()
|
