Refactor queries.

This commit is contained in:
Rykov7
2023-12-24 16:42:37 +03:00
parent fa5a502a33
commit f5e94a3880
+2 -2
View File
@@ -121,7 +121,7 @@ async def set_antispam_report_reminder(message: types.Message):
@bot.message_handler(func=is_white_id, commands=["get_quotes"], chat_types=["supergroup", "group"]) @bot.message_handler(func=is_white_id, commands=["get_quotes"], chat_types=["supergroup", "group"])
async def get_quotes(message: types.Message): async def get_quotes(message: types.Message):
logging.info(f"[{message.chat.id}] {message.from_user.first_name} {message.text}.") logging.info(f"[{message.chat.id}] {message.from_user.first_name} {message.text}.")
quotes = session.query(Quote.text).filter(Quote.chat_id == message.chat.id).all() quotes = session.query(Quote.text).filter_by(chat_id=message.chat.id).all()
if quotes: if quotes:
length = len(session.query(Quote).filter(Quote.chat_id == message.chat.id).all()) length = len(session.query(Quote).filter(Quote.chat_id == message.chat.id).all())
text = f"Количество цитат: {length}\n\Последние добавленные\n\n" text = f"Количество цитат: {length}\n\Последние добавленные\n\n"
@@ -133,7 +133,7 @@ async def get_quotes(message: types.Message):
@bot.message_handler(func=is_white_id, commands=["get_group_info"], chat_types=["supergroup", "group"]) @bot.message_handler(func=is_white_id, commands=["get_group_info"], chat_types=["supergroup", "group"])
async def get_group_info(message: types.Message): async def get_group_info(message: types.Message):
group = session.query(Chat).filter(Chat.chat_id == message.chat.id).first() group = session.query(Chat).filter_by(chat_id=message.chat.id).first()
if group: if group:
await bot.send_message( await bot.send_message(
message.chat.id, message.chat.id,