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