some fix
This commit is contained in:
11
db.py
11
db.py
@@ -233,9 +233,14 @@ class BotDB:
|
||||
except sqlite3.Error as error:
|
||||
print(error)
|
||||
|
||||
def get_blacklist_users_by_filters(self):
|
||||
"""Возвращает список пользователей в черном списке по фильтру"""
|
||||
return None
|
||||
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."""
|
||||
|
||||
8
main.py
8
main.py
@@ -57,6 +57,11 @@ class TelegramHelperBot:
|
||||
# Router for user
|
||||
@self.bot.message_handler(func=lambda message: True, chat_types=['private'])
|
||||
def handle_message(message):
|
||||
if BotDB.check_user_in_blacklist(message.from_user.id):
|
||||
attribute = BotDB.get_blacklist_users_by_id(message.from_user.id)
|
||||
self.bot.send_message(message.chat.id,
|
||||
f'<b>Ты заблокирован\nПричина блокировки:</b> {attribute[2]}\n<b>Дата разблокировки:</b> {attribute[3]}', parse_mode='HTML')
|
||||
return
|
||||
if self.state == State.START:
|
||||
if message.text == '/start':
|
||||
self.start_message(message)
|
||||
@@ -88,13 +93,14 @@ class TelegramHelperBot:
|
||||
self.bot.send_message(message.chat.id,
|
||||
"Не понимаю где ты находишься. Нажми /state, и я расскажу что ты можешь "
|
||||
"сделать")
|
||||
|
||||
if self.state == State.SUGGEST:
|
||||
self.bot.register_next_step_handler(message, self.send_to_suggest)
|
||||
self.state = State.START
|
||||
if message.text == '/start':
|
||||
self.state = State.START
|
||||
self.start_message(message)
|
||||
elif self.state == State.PRE_CHAT:
|
||||
if self.state == State.PRE_CHAT:
|
||||
self.bot.register_next_step_handler(message, self.resend_message_in_group_for_message)
|
||||
self.state = State.START
|
||||
if message.text == '/start':
|
||||
|
||||
Reference in New Issue
Block a user