add logging to db
This commit is contained in:
836
db.py
836
db.py
@@ -24,65 +24,85 @@ class BotDB:
|
|||||||
logger.info(f'Подключен к базе данных: {db_file_path}')
|
logger.info(f'Подключен к базе данных: {db_file_path}')
|
||||||
|
|
||||||
def create_table(self, sql_script):
|
def create_table(self, sql_script):
|
||||||
"""Создает таблицу в базе."""
|
"""
|
||||||
|
Создает таблицу в базе. Используется в миграциях
|
||||||
|
|
||||||
|
Args:
|
||||||
|
sql_script: DDL скрипт таблицы
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
None
|
||||||
|
"""
|
||||||
try:
|
try:
|
||||||
cursor = self.conn.cursor()
|
cursor = self.conn.cursor()
|
||||||
cursor.execute(sql_script)
|
cursor.execute(sql_script)
|
||||||
|
logger.info(f'Таблица создана: {sql_script}')
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"Ошибка при создании таблицы: {e}")
|
logger.error(f'Ошибка при создании таблицы. Данные: {sql_script} Ошибка: {e}')
|
||||||
|
raise
|
||||||
|
|
||||||
def get_current_version(self):
|
def get_current_version(self):
|
||||||
"""Получает текущую версию миграций из таблицы migrations."""
|
"""
|
||||||
|
Возвращает текущую последнюю версию миграции
|
||||||
|
|
||||||
|
Args:
|
||||||
|
None
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
int: Версия последней миграции.
|
||||||
|
"""
|
||||||
|
logger.info(f'Попытка получения версии миграции')
|
||||||
try:
|
try:
|
||||||
cursor = self.conn.cursor()
|
cursor = self.conn.cursor()
|
||||||
cursor.execute("SELECT version FROM migrations ORDER BY version DESC LIMIT 1")
|
cursor.execute("SELECT version FROM migrations ORDER BY version DESC LIMIT 1")
|
||||||
version = cursor.fetchone()[0]
|
version = cursor.fetchone()[0]
|
||||||
|
logger.info(f'Получена текущая версия миграции: {version}')
|
||||||
return version
|
return version
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"Ошибка при получении версии: {e}")
|
logger.error(f'Ошибка при получении текущей версии миграции: {e}')
|
||||||
return 0
|
raise
|
||||||
|
|
||||||
def update_version(self, new_version, script_name):
|
def update_version(self, new_version: str, script_name: str):
|
||||||
"""Обновляет версию миграций в таблице migrations."""
|
"""
|
||||||
logger.info(f'Попытка обновления версии: {new_version}, скрипт: {script_name}')
|
Обновляет версию миграций в таблице migrations.
|
||||||
|
|
||||||
|
Добавляет новую запись в таблицу migrations с указанной версией,
|
||||||
|
именем скрипта и текущей датой и временем.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
new_version (int): Новая версия миграции
|
||||||
|
script_name (str): Имя скрипта миграции
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
None
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3. IntegrityError: Если возникает ошибка целостности при вставке
|
||||||
|
данных в таблицу migrations.
|
||||||
|
Exception: Если возникает любая другая ошибка при обновлении версии.
|
||||||
|
"""
|
||||||
|
logger.info(f'Попытка обновления версии: {new_version}, название скрипта: {script_name}')
|
||||||
try:
|
try:
|
||||||
current_date = datetime.now()
|
today = datetime.now().strftime("%d-%m-%Y %H:%M:%S")
|
||||||
today = current_date.strftime("%d-%m-%Y %H:%M:%S")
|
|
||||||
cursor = self.conn.cursor()
|
cursor = self.conn.cursor()
|
||||||
cursor.execute(
|
cursor.execute(
|
||||||
"INSERT INTO migrations (version, script_name, created_at) VALUES(?, ?, ?)",
|
"INSERT INTO migrations (version, script_name, created_at) VALUES(?, ?, ?)",
|
||||||
(new_version, script_name, today),
|
(new_version, script_name, today),
|
||||||
)
|
)
|
||||||
self.conn.commit()
|
self.conn.commit()
|
||||||
logger.info(f"Версия обновлена: {new_version}, скрипт: {script_name}")
|
logger.info(f"Версия обновлена: {new_version}, название скрипта: {script_name}")
|
||||||
except sqlite3.IntegrityError as e:
|
except sqlite3.IntegrityError as e:
|
||||||
logger.error(f"Ошибка при обновлении версии: {e}")
|
logger.error(f"Ошибка при обновлении версии: {e}")
|
||||||
|
raise
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"Ошибка при обновлении версии: {e}")
|
logger.error(f"Ошибка при обновлении версии: {e}")
|
||||||
|
raise
|
||||||
# TODO: Deprecated, удалить
|
|
||||||
def get_message_from_db(self, type: str, username):
|
|
||||||
"""Функция для запроса к базе данных и получения сообщений для бота. В аргументы передаются:
|
|
||||||
type - тип получаемой обратной связи, строковое значение, сохраненное в БД
|
|
||||||
username - имя пользователя
|
|
||||||
"""
|
|
||||||
# Подключаемся к базе
|
|
||||||
try:
|
|
||||||
cursor = self.conn.cursor()
|
|
||||||
cursor.execute(f"SELECT * FROM messages WHERE type=?", (type,))
|
|
||||||
# Забираем данные из таблицы, преобразуем в строку, заменяем поле username на имя пользователя,
|
|
||||||
# и вместо амберсанда подставляем перенос строки
|
|
||||||
if type == 'connect_with_admin' or type == 'del_message' or type == 'suggest_news' or type == 'start_message':
|
|
||||||
response_from_database = str(cursor.fetchone()[1]).replace('username', username).replace('&', '\n')
|
|
||||||
else:
|
|
||||||
response_from_database = str(cursor.fetchone()[1]).replace('&', '\n')
|
|
||||||
return response_from_database
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
# TODO: Deprecated. Остался только в voice боте, удалить и оттуда
|
# TODO: Deprecated. Остался только в voice боте, удалить и оттуда
|
||||||
def get_error_message_from_db(self, id: int):
|
def get_error_message_from_db(self, id: int):
|
||||||
"""Функция для запроса к базе данных и получения сообщений ошибки. В аргументы передаются:
|
"""
|
||||||
|
@deprecated
|
||||||
|
Функция для запроса к базе данных и получения сообщений ошибки. В аргументы передаются:
|
||||||
id - идентификатор ошибки
|
id - идентификатор ошибки
|
||||||
"""
|
"""
|
||||||
# Подключаемся к базе
|
# Подключаемся к базе
|
||||||
@@ -94,66 +114,578 @@ class BotDB:
|
|||||||
except sqlite3.Error as error:
|
except sqlite3.Error as error:
|
||||||
print(error)
|
print(error)
|
||||||
|
|
||||||
def add_new_user_in_db(self, user_id, first_name, full_name, username, is_bot, language_code, date_added,
|
def add_new_user_in_db(self, user_id: int, first_name: str, full_name: str, username: str, is_bot: bool, language_code: str, date_added: str,
|
||||||
date_changed):
|
date_changed: str):
|
||||||
"""Добавляем юзера в базу"""
|
"""
|
||||||
|
Добавляет нового пользователя в базу данных.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (int): Идентификатор пользователя в Telegram.
|
||||||
|
first_name (str): Имя пользователя.
|
||||||
|
full_name (str): Полное имя пользователя.
|
||||||
|
username (str): Username пользователя в Telegram.
|
||||||
|
is_bot (bool): Флаг, указывающий, является ли пользователь ботом.
|
||||||
|
language_code (str): Код языка пользователя.
|
||||||
|
date_added (str): Дата добавления пользователя в базу.
|
||||||
|
date_changed (str): Дата последнего изменения данных пользователя.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
None: Если запись успешно добавлена в базу.
|
||||||
|
Exception: Если произошла ошибка при добавлении записи.
|
||||||
|
"""
|
||||||
|
logger.info(f"Попытка добавить пользователя в базу данных: user_id={user_id}, first_name={first_name}")
|
||||||
try:
|
try:
|
||||||
self.cursor.execute("INSERT INTO 'our_users' ('user_id', 'first_name', 'full_name', 'username', 'is_bot', "
|
self.cursor.execute("INSERT INTO 'our_users' ('user_id', 'first_name', 'full_name', 'username', 'is_bot', "
|
||||||
"'language_code', 'date_added', 'date_changed') VALUES (?, ?, ?, ?, ?, ?, ?, ?)",
|
"'language_code', 'date_added', 'date_changed') VALUES (?, ?, ?, ?, ?, ?, ?, ?)",
|
||||||
(user_id, first_name, full_name,
|
(user_id, first_name, full_name,
|
||||||
username, is_bot, language_code, date_added, date_changed))
|
username, is_bot, language_code, date_added, date_changed))
|
||||||
return self.conn.commit()
|
self.conn.commit()
|
||||||
|
logger.info(f"Новый пользователь добавлен в базу: user_id={user_id}, first_name={first_name}")
|
||||||
|
return None
|
||||||
except sqlite3.Error as error:
|
except sqlite3.Error as error:
|
||||||
print(error)
|
logger.error(f"Ошибка при добавлении пользователя в базу: {error}. "
|
||||||
|
f"Данные пользователя: user_id={user_id}, first_name={first_name}")
|
||||||
|
raise
|
||||||
|
|
||||||
def user_exists(self, user_id):
|
def user_exists(self, user_id: int):
|
||||||
"""Проверяем, есть ли юзер в базе"""
|
"""
|
||||||
try:
|
Проверяет, существует ли пользователь в базе данных.
|
||||||
result = self.cursor.execute("SELECT `id` FROM `our_users` WHERE `user_id` = ?", (user_id,))
|
|
||||||
return bool(len(result.fetchall()))
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
def get_user_id(self, user_id):
|
Args:
|
||||||
"""Достаем id юзера в базе по его user_id"""
|
user_id (int): Идентификатор пользователя.
|
||||||
try:
|
|
||||||
result = self.cursor.execute("SELECT `id` FROM `our_users` WHERE `user_id` = ?", (user_id,))
|
|
||||||
return result.fetchone()[0]
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
def get_username(self, user_id):
|
Returns:
|
||||||
"""Достаем id юзера в базе по его user_id"""
|
bool: True, если пользователь найден, False - иначе.
|
||||||
|
"""
|
||||||
|
logger.info(f"Попытка проверки существования пользователя: user_id={user_id}")
|
||||||
try:
|
try:
|
||||||
result = self.cursor.execute("SELECT `username` FROM `our_users` WHERE `user_id` = ?", (user_id,))
|
self.cursor.execute("SELECT id FROM our_users WHERE user_id = ?", (user_id,))
|
||||||
return result.fetchone()[0]
|
result = self.cursor.fetchall()
|
||||||
|
logger.info(f"Проверка существования пользователя: user_id={user_id}, результат={result}")
|
||||||
|
return bool(len(result))
|
||||||
except sqlite3.Error as error:
|
except sqlite3.Error as error:
|
||||||
print(error)
|
logger.error(f"Ошибка при проверке существования пользователя: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
def get_user_id(self, user_id: int):
|
||||||
|
"""
|
||||||
|
@deprecated
|
||||||
|
Возвращает ID пользователя в базе данных по его user_id.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (int): Идентификатор пользователя в Telegram.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
int: ID пользователя в базе данных.
|
||||||
|
None: Если пользователь не найден.
|
||||||
|
"""
|
||||||
|
logger.info(f"Попытка получения ID пользователя в базе данных для user_id={user_id}")
|
||||||
|
try:
|
||||||
|
self.cursor.execute("SELECT id FROM our_users WHERE user_id = ?", (user_id,))
|
||||||
|
result = self.cursor.fetchone()
|
||||||
|
if result:
|
||||||
|
user_id_db = result[0]
|
||||||
|
logger.info(f"ID пользователя в базе найден: user_id={user_id}, id_db={user_id_db}")
|
||||||
|
return user_id_db
|
||||||
|
else:
|
||||||
|
logger.info(f"Пользователь с user_id={user_id} не найден в базе данных.")
|
||||||
|
return 0
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка при получении ID пользователя из базы данных: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
def get_username(self, user_id: int):
|
||||||
|
"""
|
||||||
|
Возвращает username пользователя из базы данных по его user_id в Telegram.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (int): Идентификатор пользователя в Telegram.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
str: Username пользователя.
|
||||||
|
None: Если пользователь не найден.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3.Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
self.cursor.execute("SELECT username FROM our_users WHERE user_id = ?", (user_id,))
|
||||||
|
result = self.cursor.fetchone()
|
||||||
|
if result:
|
||||||
|
username = result[0]
|
||||||
|
logger.info(f"Username пользователя найден: user_id={user_id}, username={username}")
|
||||||
|
return username
|
||||||
|
else:
|
||||||
|
logger.info(f"Пользователь с user_id={user_id} не найден в базе данных.")
|
||||||
|
return ""
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка при получении username из базы данных: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
def get_all_user_id(self):
|
def get_all_user_id(self):
|
||||||
"""Достаем все айдишники юзеров из БД и преобразуем их в список"""
|
"""
|
||||||
|
Возвращает список всех user_id из базы данных.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
list: Список user_id.
|
||||||
|
[]: Если в базе данных нет пользователей.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3. Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Попытка получения всех user_id")
|
||||||
try:
|
try:
|
||||||
result = self.cursor.execute("SELECT `user_id` FROM `our_users`", )
|
self.cursor.execute("SELECT user_id FROM our_users")
|
||||||
fetch_all = result.fetchall()
|
fetch_all = self.cursor.fetchall()
|
||||||
list_of_users = []
|
list_of_users = [user_id[0] for user_id in fetch_all]
|
||||||
for i in fetch_all:
|
logger.info(f"Получен список всех user_id: {list_of_users}")
|
||||||
list_of_users.append(i[0])
|
|
||||||
return list_of_users
|
return list_of_users
|
||||||
except sqlite3.Error as error:
|
except sqlite3.Error as error:
|
||||||
print(error)
|
logger.error(f"Ошибка при получении списка user_id из базы данных: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
def get_user_first_name(self, user_id):
|
def get_user_first_name(self, user_id: int):
|
||||||
|
"""
|
||||||
|
Возвращает имя пользователя из базы данных по его user_id в Telegram.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (int): Идентификатор пользователя в Telegram.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
str: Имя пользователя.
|
||||||
|
None: Если пользователь не найден.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3.Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Попытка получения имени пользователя по user_id={user_id}")
|
||||||
try:
|
try:
|
||||||
result = self.cursor.execute("SELECT `first_name` FROM `our_users` WHERE `user_id` = ?", (user_id,))
|
self.cursor.execute("SELECT first_name FROM our_users WHERE user_id = ?", (user_id,))
|
||||||
return result.fetchone()[0]
|
result = self.cursor.fetchone()
|
||||||
|
if result:
|
||||||
|
first_name = result[0]
|
||||||
|
logger.info(f"Имя пользователя найдено: user_id={user_id}, first_name={first_name}")
|
||||||
|
return first_name
|
||||||
|
else:
|
||||||
|
logger.info(f"Пользователь с user_id={user_id} не найден в базе данных.")
|
||||||
|
return None
|
||||||
except sqlite3.Error as error:
|
except sqlite3.Error as error:
|
||||||
print(error)
|
logger.error(f"Ошибка при получении имени пользователя из базы данных: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
def change_name(self, user_id):
|
def change_name(self, user_id):
|
||||||
#TODO: реализовать функцию изменения имени пользователя по которому к нему обращается бот. ОБновляем поля first_name, date_changed
|
#TODO: реализовать функцию изменения имени пользователя по которому к нему обращается бот. Обновляем поля first_name, date_changed
|
||||||
#result = self.cursor.execute("UPDATE 'our_users' SET (?) WHERE user_id = (?)", (new_user_name), )
|
#result = self.cursor.execute("UPDATE 'our_users' SET (?) WHERE user_id = (?)", (new_user_name), )
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
def get_info_about_stickers(self, user_id: int):
|
||||||
|
"""
|
||||||
|
Проверяет, получил ли пользователь стикеры.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (int): Идентификатор пользователя в Telegram.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool: True, если пользователь получил стикеры, False - иначе.
|
||||||
|
None: Если пользователь не найден.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3.Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Попытка проверки получил ли пользователь с user_id={user_id} стикеры.")
|
||||||
|
try:
|
||||||
|
self.cursor.execute("SELECT has_stickers FROM our_users WHERE user_id = ?", (user_id,))
|
||||||
|
result = self.cursor.fetchone()
|
||||||
|
if result:
|
||||||
|
has_stickers = result[0] == 1
|
||||||
|
logger.info(f"Проверено получение стикеров пользователем: user_id={user_id}, has_stickers={has_stickers}")
|
||||||
|
return has_stickers
|
||||||
|
else:
|
||||||
|
logger.info(f"Пользователь с user_id={user_id} не найден в базе данных.")
|
||||||
|
return None
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка при получении информации о получении стикеров: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
def update_info_about_stickers(self, user_id):
|
||||||
|
"""
|
||||||
|
Обновляет информацию о получении стикеров пользователем.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (int): Идентификатор пользователя в Telegram.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
None: Если обновление успешно выполнено.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3. Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции update_info_about_stickers. Параметры: user_id={user_id}")
|
||||||
|
try:
|
||||||
|
self.cursor.execute("UPDATE our_users SET has_stickers = 1 WHERE user_id = ?", (user_id,))
|
||||||
|
self.conn.commit()
|
||||||
|
logger.info(f"Информация о получении стикеров обновлена: user_id={user_id}")
|
||||||
|
return None
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка при обновлении информации о получении стикеров: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
def get_users_blacklist(self):
|
||||||
|
"""
|
||||||
|
Возвращает список пользователей в черном списке.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
dict: Словарь, где ключ - user_id, значение - username.
|
||||||
|
{}: Если в черном списке нет пользователей.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3. Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции get_users_blacklist")
|
||||||
|
try:
|
||||||
|
self.cursor.execute("SELECT user_id, user_name FROM blacklist")
|
||||||
|
fetch_all = self.cursor.fetchall()
|
||||||
|
list_of_users = {user_id: username for user_id, username in fetch_all}
|
||||||
|
logger.info(f"Получен список пользователей в черном списке")
|
||||||
|
return list_of_users
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка при получении списка пользователей в черном списке: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
def get_users_for_unblock_today(self, date_to_unban: str):
|
||||||
|
"""
|
||||||
|
Возвращает список пользователей, у которых истекает срок блокировки сегодня.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
date_to_unban (str): Дата разблокировки.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
dict: Словарь, где ключ - user_id, значение - username.
|
||||||
|
{}: Если сегодня нет пользователей, у которых истекает срок блокировки.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3. Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции get_users_for_unblock_today: date_to_unban={date_to_unban}")
|
||||||
|
try:
|
||||||
|
result = self.cursor.execute("SELECT user_id, user_name "
|
||||||
|
"FROM blacklist WHERE date_to_unban = ?", (date_to_unban,))
|
||||||
|
fetch_all = result.fetchall()
|
||||||
|
list_of_users = {user_id: username for user_id, username in fetch_all}
|
||||||
|
logger.info(f"Получен список пользователей для разблокировки сегодня: {list_of_users}")
|
||||||
|
return list_of_users
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка при получении списка пользователей для разблокировки: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
def get_blacklist_users_by_id(self, user_id: int):
|
||||||
|
"""
|
||||||
|
Возвращает информацию о пользователе в черном списке по user_id.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (int): Идентификатор пользователя в Telegram.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
tuple: Кортеж (user_id, user_name, message_for_user, date_to_unban).
|
||||||
|
None: Если пользователь не найден в черном списке.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3.Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции get_blacklist_users_by_id: user_id={user_id}")
|
||||||
|
try:
|
||||||
|
result = self.cursor.execute("SELECT user_id, user_name, message_for_user, date_to_unban "
|
||||||
|
"FROM blacklist WHERE user_id = ?", (user_id, ))
|
||||||
|
return self.cursor.fetchone()
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка при получении информации о пользователе в черном списке: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
|
||||||
|
def check_user_in_blacklist(self, user_id: int):
|
||||||
|
"""
|
||||||
|
Проверяет, существует ли запись с данным user_id в blacklist.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (int): Идентификатор пользователя в Telegram.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool: True, если пользователь найден в черном списке, False - иначе.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3.Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции check_user_in_blacklist: user_id={user_id}")
|
||||||
|
try:
|
||||||
|
self.cursor.execute("SELECT 1 FROM blacklist WHERE user_id = ?", (user_id,))
|
||||||
|
result = self.cursor.fetchone()
|
||||||
|
return bool(result)
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка при проверке пользователя в черном списке. user_id: {user_id} : {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
def set_user_blacklist(self, user_id: int, user_name=None, message_for_user=None, date_to_unban=None):
|
||||||
|
"""
|
||||||
|
Добавляет пользователя в черный список.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (int): Идентификатор пользователя в Telegram.
|
||||||
|
user_name (str, optional): Username пользователя. Defaults to None.
|
||||||
|
message_for_user (str, optional): Сообщение для пользователя. Defaults to None.
|
||||||
|
date_to_unban (datetime, optional): Дата разблокировки. Defaults to None.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
None: Если добавление в черный список успешно выполнено.
|
||||||
|
sqlite3.Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции set_user_blacklist: user_id={user_id}, user_name={user_name},"
|
||||||
|
f" message_for_user={message_for_user}, date_to_unban={date_to_unban}")
|
||||||
|
try:
|
||||||
|
result = self.cursor.execute("INSERT INTO 'blacklist' ('user_id', 'user_name',"
|
||||||
|
" 'message_for_user', 'date_to_unban') VALUES (?, ?, ?, ?)",
|
||||||
|
(user_id, user_name, message_for_user, date_to_unban,))
|
||||||
|
self.conn.commit()
|
||||||
|
logger.info(f"Пользователь добавлен в черный список: user_id={user_id}")
|
||||||
|
return None
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка при добавлении пользователя в черный список: {error}")
|
||||||
|
return error
|
||||||
|
|
||||||
|
def delete_user_blacklist(self, user_id: int):
|
||||||
|
"""
|
||||||
|
Удаляет пользователя из черного списка.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (int): Идентификатор пользователя в Telegram.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool: True, если удаление прошло успешно, False - в случае ошибки.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
None: Ошибки обрабатываются в блоке except, возвращая False.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции delete_user_blacklist: user_id={user_id}")
|
||||||
|
try:
|
||||||
|
self.cursor.execute("DELETE FROM blacklist WHERE user_id = ?", (user_id,))
|
||||||
|
self.conn.commit()
|
||||||
|
logger.info(f"Пользователь с идентификатором {user_id} успешно удален из черного списка.")
|
||||||
|
return True
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка удаления пользователя с идентификатором {user_id} "
|
||||||
|
f"из таблицы blacklist. Ошибка: {str(error)}")
|
||||||
|
return False
|
||||||
|
|
||||||
|
def add_new_message_in_db(self, message_text: str, user_id: int, message_id: int, date: str):
|
||||||
|
"""
|
||||||
|
Добавляет новое сообщение пользователя в базу данных.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
message_text (str): Текст сообщения.
|
||||||
|
user_id (int): Идентификатор пользователя в Telegram.
|
||||||
|
message_id (int): Идентификатор сообщения в Telegram.
|
||||||
|
date (str): Дата отправки сообщения.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
None: Если добавление прошло успешно.
|
||||||
|
sqlite3.Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3.Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции add_new_message_in_db: user_id={user_id}, message_id={message_id}, date={date}")
|
||||||
|
try:
|
||||||
|
self.cursor.execute(
|
||||||
|
"INSERT INTO user_messages (message_text, user_id, message_id, date) "
|
||||||
|
"VALUES (?, ?, ?, ?)",
|
||||||
|
(message_text, user_id, message_id, date))
|
||||||
|
self.conn.commit()
|
||||||
|
logger.info(f"Новое сообщение добавлено в базу данных: message_id={message_id}")
|
||||||
|
return None
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка добавления сообщения в базу данных: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
def update_date_for_user(self, date: str, user_id: int):
|
||||||
|
"""
|
||||||
|
Обновляет дату последнего изменения данных пользователя в базе.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
date (str): Новая дата изменения.
|
||||||
|
user_id (int): Идентификатор пользователя в Telegram.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
None: Если обновление прошло успешно.
|
||||||
|
sqlite3. Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции update_date_for_user: user_id={user_id}, date={date}")
|
||||||
|
try:
|
||||||
|
self.cursor.execute("UPDATE our_users SET date_changed = ? WHERE user_id = ?",
|
||||||
|
(date, user_id,))
|
||||||
|
self.conn.commit()
|
||||||
|
logger.info(f"Дата изменения обновлена для пользователя: user_id={user_id}")
|
||||||
|
return None
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка обновления даты изменения для пользователя: {error}")
|
||||||
|
return error
|
||||||
|
|
||||||
|
def is_admin(self, user_id: int):
|
||||||
|
"""
|
||||||
|
Проверяет, является ли пользователь администратором.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id: ID пользователя Telegram.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
True, если пользователь администратор, иначе False.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
None: В случае ошибки возвращается None
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
self.cursor.execute("SELECT 1 FROM admins WHERE user_id = ?", (user_id,))
|
||||||
|
result = self.cursor.fetchone()
|
||||||
|
return bool(result)
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка добавления сообщения в базу данных: {error}")
|
||||||
|
return None
|
||||||
|
|
||||||
|
def add_admin(self, user_id: int, role: str):
|
||||||
|
"""
|
||||||
|
Добавляет пользователя в список администраторов.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (int): ID пользователя Telegram.
|
||||||
|
role (str): Роль пользователя. Доступные варианты:
|
||||||
|
1. creator - создатель
|
||||||
|
2. admin - обычная роль
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
None: Если добавление прошло успешно.
|
||||||
|
sqlite3.Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции add_admin: user_id={user_id}, role={role}")
|
||||||
|
try:
|
||||||
|
self.cursor.execute("INSERT INTO admins (user_id, role) VALUES (?, ?)", (user_id, role))
|
||||||
|
self.conn.commit()
|
||||||
|
logger.info(f"Пользователь с user_id={user_id} добавлен в список администраторов с ролью {role}.")
|
||||||
|
return None
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка добавления пользователя в список администраторов: {error}")
|
||||||
|
return error
|
||||||
|
|
||||||
|
def remove_admin(self, user_id: int):
|
||||||
|
"""
|
||||||
|
Удаляет пользователя из списка администраторов.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (int): ID пользователя Telegram.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
None: Если удаление прошло успешно.
|
||||||
|
sqlite3.Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции remove_admin: user_id={user_id}")
|
||||||
|
try:
|
||||||
|
self.cursor.execute("DELETE FROM admins WHERE user_id = ?", (user_id,))
|
||||||
|
self.conn.commit()
|
||||||
|
logger.info(f"Пользователь с user_id={user_id} удален из списка администраторов.")
|
||||||
|
return None
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка удаления пользователя из списка администраторов: {error}")
|
||||||
|
return error
|
||||||
|
|
||||||
|
def get_user_by_message_id(self, message_id: int):
|
||||||
|
"""
|
||||||
|
Возвращает идентификатор пользователя по идентификатору сообщения.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
message_id (int): Идентификатор сообщения в Telegram.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
int: Идентификатор пользователя.
|
||||||
|
None: Если пользователь не найден.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3.Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции get_user_by_message_id: message_id={message_id}")
|
||||||
|
try:
|
||||||
|
result = self.cursor.execute("SELECT user_id FROM user_messages WHERE message_id = ?", (message_id,))
|
||||||
|
user = result.fetchone()[0]
|
||||||
|
logger.info(f"Пользователь успешно получен user_id={user} по message_id={message_id}")
|
||||||
|
return user
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка получения user_id по message_id: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
def get_last_users_from_db(self):
|
||||||
|
"""
|
||||||
|
Возвращает список идентификаторов последних 100 пользователей, обращавшихся в бот.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
list: Список кортежей (full_name, user_id) последних 100 пользователей.
|
||||||
|
[]: Если в базе данных нет пользователей.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3. Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info("Запуск функции get_last_users_from_db")
|
||||||
|
try:
|
||||||
|
result = self.cursor.execute("SELECT full_name, user_id FROM our_users ORDER BY date_changed DESC LIMIT 100")
|
||||||
|
users = result.fetchall()
|
||||||
|
logger.info(f"Получен список последних 100 пользователей: {users}")
|
||||||
|
return users
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка получения списка последних пользователей: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
def get_banned_users_from_db(self):
|
||||||
|
"""
|
||||||
|
Возвращает список идентификаторов пользователей в черном списке бота.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
list: Список кортежей (user_name, user_id, message_for_user, date_to_unban) пользователей в черном списке.
|
||||||
|
[]: Если в черном списке нет пользователей.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3.Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info("Запуск функции get_banned_users_from_db")
|
||||||
|
try:
|
||||||
|
result = self.cursor.execute("SELECT user_name, user_id, message_for_user, date_to_unban FROM blacklist")
|
||||||
|
users = result.fetchall()
|
||||||
|
logger.info(f"Получен список пользователей в черном списке: {users}")
|
||||||
|
return users
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка получения списка пользователей в черном списке: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
|
def get_banned_users_from_db_with_limits(self, offset: int, limit: int):
|
||||||
|
"""
|
||||||
|
Возвращает список идентификаторов пользователей в черном списке бота с учетом смещения и ограничения.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
offset (int): Смещение для выборки
|
||||||
|
limit (int): Ограничение количества возвращаемых записей.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
list: Список кортежей (user_name, user_id, message_for_user, date_to_unban) пользователей в черном списке.
|
||||||
|
[]: Если в черном списке нет пользователей или количество записей с учетом смещения и ограничения равно 0.
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
sqlite3. Error: Если произошла ошибка при выполнении запроса.
|
||||||
|
"""
|
||||||
|
logger.info(f"Запуск функции get_banned_users_from_db_with_limits: offset={offset}, limit={limit}")
|
||||||
|
try:
|
||||||
|
result = self.cursor.execute("SELECT user_name, user_id, message_for_user, date_to_unban "
|
||||||
|
"FROM blacklist LIMIT ?, ?", (offset, limit,))
|
||||||
|
users = result.fetchall()
|
||||||
|
logger.info(f"Получен список пользователей в черном списке (offset={offset}, limit={limit}): {users}")
|
||||||
|
return users
|
||||||
|
except sqlite3.Error as error:
|
||||||
|
logger.error(f"Ошибка получения списка пользователей в черном списке: {error}")
|
||||||
|
raise
|
||||||
|
|
||||||
def add_audio_record(self, file_name, author_id, date_added, listen_count, file_id):
|
def add_audio_record(self, file_name, author_id, date_added, listen_count, file_id):
|
||||||
"""Добавляет информацию о войсе юзера в БД"""
|
"""Добавляет информацию о войсе юзера в БД"""
|
||||||
try:
|
try:
|
||||||
@@ -233,174 +765,6 @@ class BotDB:
|
|||||||
except sqlite3.Error as error:
|
except sqlite3.Error as error:
|
||||||
print(error)
|
print(error)
|
||||||
|
|
||||||
def get_info_about_stickers(self, user_id):
|
|
||||||
"""Получает данные о получении стикеров пользователем"""
|
|
||||||
try:
|
|
||||||
result = self.cursor.execute("SELECT `has_stickers` FROM `our_users` WHERE `user_id` = ?", (user_id,))
|
|
||||||
return result.fetchone()[0] == 1
|
|
||||||
#return result.fetchone()[0]
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
def update_info_about_stickers(self, user_id):
|
|
||||||
"""Обновляет данные о получении стикеров пользователем"""
|
|
||||||
try:
|
|
||||||
result = self.cursor.execute("UPDATE `our_users` SET `has_stickers` = 1 WHERE `user_id` = ?", (user_id,))
|
|
||||||
return self.conn.commit()
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
def get_users_blacklist(self):
|
|
||||||
"""Возвращает список пользователей в черном списке"""
|
|
||||||
try:
|
|
||||||
result = self.cursor.execute("SELECT user_id, user_name FROM `blacklist`")
|
|
||||||
fetch_all = result.fetchall()
|
|
||||||
list_of_users = {}
|
|
||||||
for i in fetch_all:
|
|
||||||
list_of_users[i[0]] = i[1]
|
|
||||||
return list_of_users
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
def get_users_for_unblock_today(self, date_to_unban):
|
|
||||||
"""Возвращает пользователей у которых истекает срок блокировки сегодня"""
|
|
||||||
try:
|
|
||||||
result = self.cursor.execute("SELECT user_id, user_name FROM `blacklist` WHERE date_to_unban = ?", (date_to_unban,))
|
|
||||||
fetch_all = result.fetchall()
|
|
||||||
list_of_users = {}
|
|
||||||
for i in fetch_all:
|
|
||||||
list_of_users[i[0]] = i[1]
|
|
||||||
return list_of_users
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
def get_blacklist_users_by_id(self, user_id):
|
|
||||||
"""Возвращает список пользователей в черном списке по user_id"""
|
|
||||||
try:
|
|
||||||
result = self.cursor.execute("SELECT user_id, user_name, message_for_user, date_to_unban FROM `blacklist` WHERE user_id = ?", (user_id, ))
|
|
||||||
return self.cursor.fetchone()
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
|
|
||||||
def check_user_in_blacklist(self, user_id):
|
|
||||||
"""Проверяет, существует ли запись с данным user_id в blacklist."""
|
|
||||||
self.cursor.execute("SELECT 1 FROM blacklist WHERE user_id = ?", (user_id,))
|
|
||||||
result = self.cursor.fetchone()
|
|
||||||
return bool(result)
|
|
||||||
|
|
||||||
def set_user_blacklist(self, user_id, user_name=None, message_for_user=None, date_to_unban=None):
|
|
||||||
"""Добавляет пользователя в черный список"""
|
|
||||||
try:
|
|
||||||
result = self.cursor.execute("INSERT INTO 'blacklist' ('user_id', 'user_name',"
|
|
||||||
" 'message_for_user', 'date_to_unban') VALUES (?, ?, ?, ?)",
|
|
||||||
(user_id, user_name, message_for_user, date_to_unban,))
|
|
||||||
return self.conn.commit()
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
return error
|
|
||||||
|
|
||||||
def delete_user_blacklist(self, user_id):
|
|
||||||
"""Удаляет пользователя из черного списка"""
|
|
||||||
try:
|
|
||||||
#TODO: Функция всегда возвращает true, даже если такого id нет в таблице
|
|
||||||
self.cursor.execute("DELETE FROM blacklist WHERE user_id = ?", (user_id,))
|
|
||||||
self.conn.commit()
|
|
||||||
logger.info(f"Пользователь с идентификатором {user_id} успешно удален.")
|
|
||||||
return True
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
logger.error(f"Ошибка удаления пользователя с идентификатором {user_id} из таблицы blacklist. Ошибка: {str(error)}")
|
|
||||||
return False
|
|
||||||
|
|
||||||
def add_new_message_in_db(self, message_text, user_id, message_id, date):
|
|
||||||
"""Добавляем сообщение юзера в базу"""
|
|
||||||
try:
|
|
||||||
|
|
||||||
self.cursor.execute(
|
|
||||||
"INSERT INTO `user_messages` (message_text, user_id, message_id, date) "
|
|
||||||
"VALUES (?, ?, ?, ?)",
|
|
||||||
(message_text, message_id, user_id, date))
|
|
||||||
return self.conn.commit()
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
def update_date_for_user(self, date, user_id: int):
|
|
||||||
try:
|
|
||||||
result = self.cursor.execute("UPDATE `our_users` SET `date_changed` = ? WHERE `user_id` = ?",
|
|
||||||
(date, user_id,))
|
|
||||||
return self.conn.commit()
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
def is_admin(self, user_id):
|
|
||||||
"""
|
|
||||||
Проверяет, является ли пользователь администратором.
|
|
||||||
Args:
|
|
||||||
user_id: ID пользователя Telegram.
|
|
||||||
Returns:
|
|
||||||
True, если пользователь администратор, иначе False.
|
|
||||||
"""
|
|
||||||
self.cursor.execute("SELECT 1 FROM admins WHERE user_id = ?", (user_id,))
|
|
||||||
result = self.cursor.fetchone()
|
|
||||||
return bool(result)
|
|
||||||
|
|
||||||
def add_admin(self, user_id, role):
|
|
||||||
"""
|
|
||||||
Добавляет пользователя в список администраторов.
|
|
||||||
Args:
|
|
||||||
user_id: ID пользователя Telegram.
|
|
||||||
role: Роль пользователя.
|
|
||||||
Доступные варианты:
|
|
||||||
1. creator - создатель
|
|
||||||
2. admin - обычная роль
|
|
||||||
"""
|
|
||||||
self.cursor.execute("INSERT INTO admins (user_id, role) VALUES (?, ?)", (user_id, role))
|
|
||||||
return self.conn.commit()
|
|
||||||
|
|
||||||
def remove_admin(self, user_id):
|
|
||||||
"""
|
|
||||||
Удаляет пользователя из списка администраторов.
|
|
||||||
|
|
||||||
Args:
|
|
||||||
user_id: ID пользователя Telegram.
|
|
||||||
"""
|
|
||||||
self.cursor.execute("DELETE FROM admins WHERE user_id = ?", (user_id,))
|
|
||||||
return self.conn.commit()
|
|
||||||
|
|
||||||
def get_user_by_message_id(self, message_id):
|
|
||||||
"""Возвращает идентификатор пользователя по идентификатору сообщения"""
|
|
||||||
try:
|
|
||||||
result = self.cursor.execute("SELECT user_id FROM `user_messages` WHERE message_id = ?", (message_id,))
|
|
||||||
return result.fetchone()[0]
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
def get_last_users_from_db(self):
|
|
||||||
"""Возвращает список идентификаторов последних 10 пользователей обращавшихся в бот"""
|
|
||||||
try:
|
|
||||||
result = self.cursor.execute("SELECT full_name, user_id FROM our_users ORDER BY date_changed DESC")
|
|
||||||
users = result.fetchall()
|
|
||||||
return users
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
def get_banned_users_from_db(self):
|
|
||||||
"""Возвращает список идентификаторов последних 10 пользователей обращавшихся в бот"""
|
|
||||||
try:
|
|
||||||
result = self.cursor.execute("SELECT user_name, user_id, message_for_user, date_to_unban FROM blacklist", )
|
|
||||||
users = result.fetchall()
|
|
||||||
return users
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
def get_banned_users_from_db_with_limits(self, offset: int, limit: int):
|
|
||||||
"""Возвращает список идентификаторов последних 10 пользователей обращавшихся в бот"""
|
|
||||||
try:
|
|
||||||
result = self.cursor.execute("SELECT user_name, user_id, message_for_user, date_to_unban FROM blacklist LIMIT ?, ?", (offset, limit,) )
|
|
||||||
users = result.fetchall()
|
|
||||||
return users
|
|
||||||
except sqlite3.Error as error:
|
|
||||||
print(error)
|
|
||||||
|
|
||||||
def close(self):
|
def close(self):
|
||||||
"""Закрываем соединение с БД"""
|
"""Закрываем соединение с БД"""
|
||||||
self.conn.close()
|
self.conn.close()
|
||||||
|
|||||||
6
main.py
6
main.py
@@ -635,7 +635,7 @@ if __name__ == "__main__":
|
|||||||
# Запускаем бота
|
# Запускаем бота
|
||||||
bot.start()
|
bot.start()
|
||||||
|
|
||||||
scheduler = BackgroundScheduler()
|
#scheduler = BackgroundScheduler()
|
||||||
scheduler.add_job(bot.unban_notifier(), 'cron', hour=0, minute=0)
|
#scheduler.add_job(bot.unban_notifier(), 'cron', hour=0, minute=0)
|
||||||
scheduler.start()
|
#scheduler.start()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user