import logging from io import BytesIO from olgram.router import dp from datetime import datetime, timedelta, timezone from aiogram import types, Bot as AioBot from olgram.models.models import Bot, User, DefaultAnswer, BotStartMessage, BotSecondMessage from aiogram.dispatcher import FSMContext from aiogram.utils.callback_data import CallbackData from textwrap import dedent from olgram.utils.mix import edit_or_create, button_text_limit, wrap, send_stored_message from olgram.commands import bot_actions from locales.locale import _ import typing as ty menu_callback = CallbackData('menu', 'level', 'bot_id', 'operation', 'chat') empty = "0" async def send_bots_menu(chat_id: int, user_id: int, call=None): """ Отправить пользователю список ботов :return: """ if call: await call.answer() user = await User.get_or_none(telegram_id=user_id) bots = await Bot.filter(owner=user) if not bots: await AioBot.get_current().send_message(chat_id, dedent(_(""" У вас нет добавленных ботов. Отправьте команду /addbot, чтобы добавить бот. """))) return keyboard = types.InlineKeyboardMarkup(row_width=2) for bot in bots: keyboard.insert( types.InlineKeyboardButton(text="@" + bot.name, callback_data=menu_callback.new(level=1, bot_id=bot.id, operation=empty, chat=empty)) ) text = _("Ваши боты") if call: await edit_or_create(call, text, keyboard) else: await AioBot.get_current().send_message(chat_id, text, reply_markup=keyboard) async def send_chats_menu(bot: Bot, call: types.CallbackQuery): await call.answer() keyboard = types.InlineKeyboardMarkup(row_width=2) chats = await bot.group_chats.all() for chat in chats: keyboard.insert( types.InlineKeyboardButton(text=button_text_limit(chat.name), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="chat", chat=chat.id)) ) if chats: keyboard.insert( types.InlineKeyboardButton(text=_("Личные сообщения"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="chat", chat="personal")) ) keyboard.insert( types.InlineKeyboardButton(text=_("❗️ Выйти из всех чатов"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="chat", chat="leave")) ) keyboard.insert( types.InlineKeyboardButton(text=_("<< Назад"), callback_data=menu_callback.new(level=1, bot_id=bot.id, operation=empty, chat=empty)) ) if not chats: text = dedent(_(""" Этот бот не добавлен в чаты, поэтому все сообщения будут приходить вам в бот. Чтобы подключить чат — добавьте бот @{0} в чат, откройте это меню ещё раз и выберите добавленный чат. Если ваш бот состоял в групповом чате до того, как его добавили в Olgram - удалите бота из чата и добавьте снова. """)).format(bot.name) else: text = dedent(_(""" В этом разделе вы можете привязать бота @{0} к чату. Выберите чат, куда бот будет пересылать сообщения. """)).format(bot.name) await edit_or_create(call, text, keyboard) async def send_bot_menu(bot: Bot, call: types.CallbackQuery): await call.answer() keyboard = types.InlineKeyboardMarkup(row_width=2) keyboard.insert( types.InlineKeyboardButton(text=_("Текст"), callback_data=menu_callback.new(level=2, bot_id=bot.id, operation="text", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Чат"), callback_data=menu_callback.new(level=2, bot_id=bot.id, operation="chat", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Удалить бот"), callback_data=menu_callback.new(level=2, bot_id=bot.id, operation="delete", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Статистика"), callback_data=menu_callback.new(level=2, bot_id=bot.id, operation="stat", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("<< Назад"), callback_data=menu_callback.new(level=0, bot_id=empty, operation=empty, chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Опции"), callback_data=menu_callback.new(level=2, bot_id=bot.id, operation="settings", chat=empty)) ) if bot.enable_mailing: keyboard.insert( types.InlineKeyboardButton(text=_("Рассылка"), callback_data=menu_callback.new(level=2, bot_id=bot.id, operation="go_mailing", chat=empty)) ) await edit_or_create(call, dedent(_(""" Управление ботом @{0}. Если у вас возникли вопросы по настройке бота, то посмотрите нашу справку /help или напишите нам @civsocit_feedback_bot """)).format(bot.name), reply_markup=keyboard) async def send_bot_delete_menu(bot: Bot, call: types.CallbackQuery): await call.answer() keyboard = types.InlineKeyboardMarkup(row_width=2) keyboard.insert( types.InlineKeyboardButton(text=_("Да, удалить бот"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="delete_yes", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("<< Назад"), callback_data=menu_callback.new(level=1, bot_id=bot.id, operation=empty, chat=empty)) ) await edit_or_create(call, dedent(_(""" Вы уверены, что хотите удалить бота @{0}? """)).format(bot.name), reply_markup=keyboard) async def send_bot_settings_menu(bot: Bot, call: types.CallbackQuery): await call.answer() keyboard = types.InlineKeyboardMarkup(row_width=2) keyboard.insert( types.InlineKeyboardButton(text=_("Потоки сообщений"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="threads", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Данные пользователя"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="additional_info", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Антифлуд"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="antiflood", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Автоответчик всегда"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="always_second_message", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Рассылка"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="mailing", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Прерывать поток"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="thread_interrupt", chat=empty)) ) is_promo = await bot.is_promo() if is_promo: keyboard.insert( types.InlineKeyboardButton(text=_("Olgram подпись"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="olgram_text", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("<< Назад"), callback_data=menu_callback.new(level=1, bot_id=bot.id, operation=empty, chat=empty)) ) thread_turn = _("включены") if bot.enable_threads else _("выключены") info_turn = _("включены") if bot.enable_additional_info else _("выключены") antiflood_turn = _("включен") if bot.enable_antiflood else _("выключен") enable_always_second_message = _("включён") if bot.enable_always_second_message else _("выключен") thread_interrupt = _("да") if bot.enable_thread_interrupt else _("нет") mailing_turn = _("включена") if bot.enable_mailing else _("выключена") text = dedent(_(""" Потоки сообщений: {0} Данные пользователя: {1} Антифлуд: {2} Автоответчик всегда: {3} Прерывать поток: {4} Рассылка: {5} """)).format(thread_turn, info_turn, antiflood_turn, enable_always_second_message, thread_interrupt, mailing_turn) if is_promo: olgram_turn = _("включена") if bot.enable_olgram_text else _("выключена") text += _("Olgram подпись: {0}").format(olgram_turn) await edit_or_create(call, text, reply_markup=keyboard, parse_mode="HTML") languages = { "en": "English 🇺🇸", "ru": "Русский 🇷🇺", "uk": "Український 🇺🇦", "tr": "Türkçe 🇹🇷", "hy": "հայերեն 🇦🇲", "ka": "ქართული ენა 🇬🇪" } async def send_bot_text_menu(bot: Bot, call: ty.Optional[types.CallbackQuery] = None, chat_id: ty.Optional[int] = None, state=None): if call: await call.answer() async with state.proxy() as proxy: lang = proxy.get("lang", "none") prepared_languages = {ln.locale: ln.text for ln in await bot.start_texts} keyboard = types.InlineKeyboardMarkup(row_width=2) keyboard.insert( types.InlineKeyboardButton(text=_("<< Завершить редактирование"), callback_data=menu_callback.new(level=1, bot_id=bot.id, operation=empty, chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Автоответчик"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="next_text", chat=empty)) ) keyboard.row( types.InlineKeyboardButton(text=_("Сбросить текст"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="reset_text", chat=empty)) ) keyboard.add( types.InlineKeyboardButton(text=("🟢 " if lang == "none" else "") + _("[все языки]"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="slang_none", chat=empty)) ) for code, name in languages.items(): prefix = "" if code == lang: prefix = "🟢 " elif code in prepared_languages: prefix = "✔️ " keyboard.insert( types.InlineKeyboardButton(text=prefix + name, callback_data=menu_callback.new(level=3, bot_id=bot.id, operation=f"slang_{code}", chat=empty)) ) text = dedent(_(""" Сейчас вы редактируете текст, который отправляется после того, как пользователь отправит вашему боту @{0} команду /start Текущий текст{2}: 
{1}
Отправьте сообщение, чтобы изменить текст. """)) text = text.format(bot.name, prepared_languages.get(lang, bot.start_text), _(" (для языка {0})").format(languages[lang]) if lang != "none" else "") if call: await edit_or_create(call, text, keyboard, parse_mode="HTML") else: await AioBot.get_current().send_message(chat_id, text, reply_markup=keyboard, parse_mode="HTML") async def send_bot_mailing_menu(bot: Bot, call: ty.Optional[types.CallbackQuery] = None, chat_id: ty.Optional[int] = None): if call: await call.answer() keyboard = types.InlineKeyboardMarkup(row_width=1) keyboard.insert( types.InlineKeyboardButton(text=_("<< Отменить рассылку"), callback_data=menu_callback.new(level=1, bot_id=bot.id, operation=empty, chat=empty)) ) text = dedent(_(""" Напишите сообщение, которое нужно разослать всем подписчикам вашего бота @{0}. У сообщения будет до {1} получателей. Учтите, что 1. Рассылается только одно сообщение за раз (в т.ч. только одна картинка) 2. Когда рассылка запущена, её нельзя отменить """)) text = text.format(bot.name, len(await bot.mailing_users)) if call: await edit_or_create(call, text, keyboard, parse_mode="HTML") else: await AioBot.get_current().send_message(chat_id, text, reply_markup=keyboard, parse_mode="HTML") @dp.message_handler(state="wait_mailing_text", content_types=[types.ContentType.TEXT, types.ContentType.LOCATION, types.ContentType.DOCUMENT, types.ContentType.PHOTO, types.ContentType.AUDIO, types.ContentType.VIDEO]) # TODO: not command async def mailing_text_received(message: types.Message, state: FSMContext): async with state.proxy() as proxy: bot_id = proxy["bot_id"] proxy["mailing_content_type"] = message.content_type buffer = BytesIO() if message.content_type == types.ContentType.TEXT: proxy["mailing_text"] = message.html_text elif message.content_type == types.ContentType.LOCATION: proxy["mailing_location"] = message.location elif message.content_type in (types.ContentType.PHOTO, types.ContentType.DOCUMENT, types.ContentType.AUDIO, types.ContentType.VIDEO): proxy["mailing_caption"] = message.caption if message.content_type == types.ContentType.PHOTO: obj = message.photo[-1] elif message.content_type == types.ContentType.DOCUMENT: obj = message.document elif message.content_type == types.ContentType.AUDIO: obj = message.audio elif message.content_type == types.ContentType.VIDEO: obj = message.video if obj.file_size and obj.file_size > 4194304: return await message.answer(_("Слишком большой файл (4 Мб максимум)")) try: await obj.download(buffer, timeout=5) except Exception as err: logging.error("Error downloading file") logging.error(err, exc_info=True) return await message.answer(_("Не удалось загрузить файл (слишком большой размер?)")) proxy["mailing_data"] = buffer.getvalue() proxy["mailing_file_name"] = getattr(obj, "file_name", None) _message_id = await send_stored_message(proxy, AioBot.get_current(), message.chat.id) keyboard = types.InlineKeyboardMarkup(row_width=1) keyboard.insert( types.InlineKeyboardButton(text=_("<< Отменить рассылку"), callback_data=menu_callback.new(level=1, bot_id=bot_id, operation=empty, chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Да, начать рассылку"), callback_data=menu_callback.new(level=3, bot_id=bot_id, operation="go_go_mailing", chat=empty)) ) await AioBot.get_current().send_message(message.chat.id, reply_to_message_id=_message_id, text="Вы уверены, что хотите разослать это сообщение всем пользователям?", reply_markup=keyboard) async def send_bot_statistic_menu(bot: Bot, call: ty.Optional[types.CallbackQuery] = None, chat_id: ty.Optional[int] = None): if call: await call.answer() keyboard = types.InlineKeyboardMarkup(row_width=2) keyboard.insert( types.InlineKeyboardButton(text=_("<< Назад"), callback_data=menu_callback.new(level=1, bot_id=bot.id, operation=empty, chat=empty)) ) text = dedent(_(""" Статистика по боту @{0} Входящих сообщений: {1} Ответных сообщений: {2} Шаблоны ответов: {3} Забанено пользователей: {4} """)).format(bot.name, bot.incoming_messages_count, bot.outgoing_messages_count, len(await bot.answers), len(await bot.banned_users)) if call: await edit_or_create(call, text, keyboard, parse_mode="HTML") else: await AioBot.get_current().send_message(chat_id, text, reply_markup=keyboard, parse_mode="HTML") async def send_bot_second_text_menu(bot: Bot, call: ty.Optional[types.CallbackQuery] = None, chat_id: ty.Optional[int] = None, state=None): if call: await call.answer() async with state.proxy() as proxy: lang = proxy.get("lang", "none") prepared_languages = {ln.locale: ln.text for ln in await bot.second_texts} keyboard = types.InlineKeyboardMarkup(row_width=2) keyboard.insert( types.InlineKeyboardButton(text=_("<< Завершить редактирование"), callback_data=menu_callback.new(level=1, bot_id=bot.id, operation=empty, chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Предыдущий текст"), callback_data=menu_callback.new(level=2, bot_id=bot.id, operation="text", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Шаблоны ответов..."), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="templates", chat=empty)) ) keyboard.insert( types.InlineKeyboardButton(text=_("Сбросить текст"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="reset_second_text", chat=empty)) ) keyboard.add( types.InlineKeyboardButton(text=("🟢 " if lang == "none" else "") + _("[все языки]"), callback_data=menu_callback.new(level=3, bot_id=bot.id, operation="alang_none", chat=empty)) ) for code, name in languages.items(): prefix = "" if code == lang: prefix = "🟢 " elif code in prepared_languages: prefix = "✔️ " keyboard.insert( types.InlineKeyboardButton(text=prefix + name, callback_data=menu_callback.new(level=3, bot_id=bot.id, operation=f"alang_{code}", chat=empty)) ) text = dedent(_(""" Сейчас вы редактируете текст автоответчика. Это сообщение отправляется в ответ на все входящие сообщения @{0} \ автоматически. По умолчанию оно отключено. Текущий текст{2}: 
{1}
Отправьте сообщение, чтобы изменить текст. """)) text = text.format(bot.name, prepared_languages.get(lang, bot.second_text or _("отключено")), _(" (для языка {0})").format(languages[lang]) if lang != "none" else "") if call: await edit_or_create(call, text, keyboard, parse_mode="HTML") else: await AioBot.get_current().send_message(chat_id, text, reply_markup=keyboard, parse_mode="HTML") async def send_bot_templates_menu(bot: Bot, call: ty.Optional[types.CallbackQuery] = None, chat_id: ty.Optional[int] = None): if call: await call.answer() keyboard = types.InlineKeyboardMarkup(row_width=2) keyboard.insert( types.InlineKeyboardButton(text=_("<< Завершить редактирование"), callback_data=menu_callback.new(level=1, bot_id=bot.id, operation=empty, chat=empty)) ) text = dedent(_(""" Сейчас вы редактируете шаблоны ответов для @{0}. Текущие шаблоны: 
    {1}
Отправьте какую-нибудь фразу (например: "Ваш заказ готов, ожидайте!"), чтобы добавить её в шаблон. Чтобы удалить шаблон из списка, отправьте его номер в списке (например, 4) """)) templates = await bot.answers total_text_len = sum(len(t.text) for t in templates) + len(text) # примерная длина текста max_len = 1000 if total_text_len > 4000: max_len = 100 templates_text = "\n".join(f"{n}. {wrap(template.text, max_len)}" for n, template in enumerate(templates)) if not templates_text: templates_text = _("(нет шаблонов)") text = text.format(bot.name, templates_text) if call: await edit_or_create(call, text, keyboard, parse_mode="HTML") else: await AioBot.get_current().send_message(chat_id, text, reply_markup=keyboard, parse_mode="HTML") @dp.message_handler(state="wait_start_text", content_types="text", regexp="^[^/].+") # Not command async def start_text_received(message: types.Message, state: FSMContext): async with state.proxy() as proxy: bot_id = proxy.get("bot_id") lang = proxy.get("lang", "none") bot = await Bot.get_or_none(pk=bot_id) if lang == "none": bot.start_text = message.html_text await bot.save(update_fields=["start_text"]) else: obj, created = await BotStartMessage.get_or_create(bot=bot, locale=lang, defaults={"text": message.html_text}) if not created: obj.text = message.html_text await obj.save(update_fields=["text"]) await send_bot_text_menu(bot, chat_id=message.chat.id, state=state) @dp.message_handler(state="wait_second_text", content_types="text", regexp="^[^/].+") # Not command async def second_text_received(message: types.Message, state: FSMContext): async with state.proxy() as proxy: bot_id = proxy.get("bot_id") lang = proxy.get("lang", "none") bot = await Bot.get_or_none(pk=bot_id) if lang == "none": bot.second_text = message.html_text await bot.save(update_fields=["second_text"]) else: obj, created = await BotSecondMessage.get_or_create(bot=bot, locale=lang, defaults={"text": message.html_text}) if not created: obj.text = message.html_text await obj.save(update_fields=["text"]) if not bot.second_text: bot.second_text = message.html_text await bot.save(update_fields=["second_text"]) await send_bot_second_text_menu(bot, chat_id=message.chat.id, state=state) @dp.message_handler(state="wait_template", content_types="text", regexp="^[^/](.+)?") # Not command async def template_received(message: types.Message, state: FSMContext): async with state.proxy() as proxy: bot_id = proxy.get("bot_id") bot = await Bot.get_or_none(pk=bot_id) if message.text.isdigit(): # Delete template number = int(message.text) templates = await bot.answers if not templates: await message.answer(_("У вас нет шаблонов, чтобы их удалять")) if number < 0 or number >= len(templates): await message.answer(_("Неправильное число. Чтобы удалить шаблон, введите число от 0 до {0}").format( len(templates))) return await templates[number].delete() else: # Add template total_templates = len(await bot.answers) if total_templates > 30: await message.answer(_("У вашего бота уже слишком много шаблонов")) else: answers = await bot.answers.filter(text=message.text) if answers: await message.answer(_("Такой текст уже есть в списке шаблонов")) else: template = DefaultAnswer(text=message.text, bot=bot) await template.save() await send_bot_templates_menu(bot, chat_id=message.chat.id) @dp.callback_query_handler(menu_callback.filter(), state="*") async def callback(call: types.CallbackQuery, callback_data: dict, state: FSMContext): level = callback_data.get("level") if level == "0": return await send_bots_menu(call.message.chat.id, call.from_user.id, call) bot_id = callback_data.get("bot_id") bot = await Bot.get_or_none(id=bot_id) if not bot or (await bot.owner).telegram_id != call.from_user.id: await call.answer(_("У вас нет прав на этого бота"), show_alert=True) return if level == "1": await state.reset_state() return await send_bot_menu(bot, call) operation = callback_data.get("operation") if level == "2": await state.reset_state() if operation == "chat": return await send_chats_menu(bot, call) if operation == "delete": return await send_bot_delete_menu(bot, call) if operation == "stat": return await send_bot_statistic_menu(bot, call) if operation == "settings": return await send_bot_settings_menu(bot, call) if operation == "go_mailing": if bot.last_mailing_at and bot.last_mailing_at >= datetime.now(tz=timezone.utc) - timedelta(minutes=5): return await call.answer(_("Рассылка была совсем недавно, подождите немного"), show_alert=True) if not await bot.mailing_users: return await call.answer(_("Нет пользователей для рассылки")) await state.set_state("wait_mailing_text") async with state.proxy() as proxy: proxy["bot_id"] = bot.id return await send_bot_mailing_menu(bot, call) if operation == "text": await state.set_state("wait_start_text") async with state.proxy() as proxy: proxy["bot_id"] = bot.id return await send_bot_text_menu(bot, call, state=state) if level == "3": if operation == "delete_yes": return await bot_actions.delete_bot(bot, call) if operation == "mailing": await bot_actions.mailing(bot, call) return await send_bot_settings_menu(bot, call) if operation == "go_go_mailing": if (await state.get_state()) == "wait_mailing_text": async with state.proxy() as proxy: mailing_data = dict(proxy) await state.reset_state() if bot.last_mailing_at and bot.last_mailing_at >= datetime.now(tz=timezone.utc) - timedelta(minutes=5): return await call.answer(_("Рассылка была совсем недавно, подождите немного"), show_alert=True) if not await bot.mailing_users: return await call.answer(_("Нет пользователей для рассылки")) await call.answer(_("Рассылка запущена")) count = await bot_actions.go_mailing(bot, mailing_data) return await call.message.answer(_("Рассылка завершена, отправлено {0} сообщений").format(count)) else: return await call.answer(_("Устарело, создайте новую рассылку")) if operation == "chat": return await bot_actions.select_chat(bot, call, callback_data.get("chat")) if operation == "threads": await bot_actions.threads(bot, call) return await send_bot_settings_menu(bot, call) if operation == "antiflood": await bot_actions.antiflood(bot, call) return await send_bot_settings_menu(bot, call) if operation == "additional_info": await bot_actions.additional_info(bot, call) return await send_bot_settings_menu(bot, call) if operation == "always_second_message": await bot_actions.always_second_message(bot, call) return await send_bot_settings_menu(bot, call) if operation == "mailing": await bot_actions.mailing(bot, call) return await send_bot_settings_menu(bot, call) if operation == "thread_interrupt": await bot_actions.thread_interrupt(bot, call) return await send_bot_settings_menu(bot, call) if operation == "olgram_text": await bot_actions.olgram_text(bot, call) return await send_bot_settings_menu(bot, call) if operation == "reset_text": await bot_actions.reset_bot_text(bot, call, state) return await send_bot_text_menu(bot, call, state=state) if operation.startswith("slang_"): async with state.proxy() as proxy: lang = operation.replace("slang_", "") if lang == "none" or lang in languages: proxy["lang"] = lang return await send_bot_text_menu(bot, call, state=state) if operation == "next_text": await state.set_state("wait_second_text") async with state.proxy() as proxy: proxy["bot_id"] = bot.id return await send_bot_second_text_menu(bot, call, state=state) if operation.startswith("alang_"): async with state.proxy() as proxy: lang = operation.replace("alang_", "") if lang == "none" or lang in languages: proxy["lang"] = lang return await send_bot_second_text_menu(bot, call, state=state) if operation == "reset_second_text": await bot_actions.reset_bot_second_text(bot, call, state) return await send_bot_second_text_menu(bot, call, state=state) if operation == "templates": await state.set_state("wait_template") async with state.proxy() as proxy: proxy["bot_id"] = bot.id return await send_bot_templates_menu(bot, call)