-
Notifications
You must be signed in to change notification settings - Fork 0
/
sql.py
32 lines (26 loc) · 1.5 KB
/
sql.py
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
import sqlite3
class SQL:
def __init__(self, database_file):
"""Подключаемся к БД и сохраняем курсор соединения"""
self.connection = sqlite3.connect(database_file)
self.cursor = self.connection.cursor()
def get_subscriptions(self, status = True):
"""Получаем всех активных подписчиков бота"""
with self.connection:
return self.cursor.execute("SELECT * FROM `subscriptions` WHERE `status` = ?", (status,)).fetchall()
def subscriber_exists(self, user_id):
"""Проверяем, есть ли уже юзер в базе"""
with self.connection:
result = self.cursor.execute('SELECT * FROM `subscriptions` WHERE `user_id` = ?', (user_id,)).fetchall()
return bool(len(result))
def add_subscriber(self, user_id, status = True):
"""Добавляем нового подписчика"""
with self.connection:
return self.cursor.execute("INSERT INTO `subscriptions` (`user_id`, `status`) VALUES(?,?)", (user_id,status))
def update_subscription(self, user_id, status):
"""Обновляем статус подписки пользователя"""
with self.connection:
return self.cursor.execute("UPDATE `subscriptions` SET `status` = ? WHERE `user_id` = ?", (status, user_id))
def close(self):
"""Закрываем соединение с БД"""
self.connection.close()