mirror of
https://github.com/civsocit/olgram.git
synced 2023-07-22 01:29:12 +03:00
120 lines
4.5 KiB
Python
120 lines
4.5 KiB
Python
"""
|
||
Здесь работа с ботами на первом уровне вложенности: список ботов, добавление ботов
|
||
"""
|
||
from aiogram import types, Bot as AioBot
|
||
from aiogram.dispatcher import FSMContext
|
||
from aiogram.utils.exceptions import Unauthorized, TelegramAPIError
|
||
from tortoise.exceptions import IntegrityError
|
||
import re
|
||
from textwrap import dedent
|
||
|
||
from ..utils.router import Router
|
||
from .bot import select_bot
|
||
from olgram.models.models import Bot, User
|
||
from olgram.settings import OlgramSettings
|
||
|
||
router = Router()
|
||
token_pattern = r'[0-9]{8,10}:[a-zA-Z0-9_-]{35}'
|
||
|
||
|
||
@router.message_handler(commands=["mybots"], state="*")
|
||
async def my_bots(message: types.Message, state: FSMContext):
|
||
"""
|
||
Команда /mybots (список ботов)
|
||
"""
|
||
user = await User.get_or_none(telegram_id=message.from_user.id)
|
||
bots = await Bot.filter(owner=user)
|
||
if not bots:
|
||
await message.answer(dedent("""
|
||
У вас нет добавленных ботов.
|
||
|
||
Отправьте команду /addbot, чтобы добавить бот.
|
||
"""))
|
||
return
|
||
|
||
keyboard = types.InlineKeyboardMarkup(row_width=2)
|
||
for bot in bots:
|
||
keyboard.insert(types.InlineKeyboardButton(text="@" + bot.name, callback_data=select_bot.new(bot_id=bot.id)))
|
||
|
||
await message.answer("Ваши боты", reply_markup=keyboard)
|
||
|
||
|
||
@router.message_handler(commands=["addbot"], state="*")
|
||
async def add_bot(message: types.Message, state: FSMContext):
|
||
"""
|
||
Команда /addbot (добавить бота)
|
||
"""
|
||
bot_count = await Bot.filter(owner__telegram_id=message.from_user.id).count()
|
||
if bot_count >= OlgramSettings.max_bots_per_user():
|
||
await message.answer("У вас уже слишком много ботов.")
|
||
return
|
||
|
||
await message.answer(dedent("""
|
||
Чтобы подключить бот, вам нужно выполнить три действия:
|
||
|
||
1. Перейдите в бот @BotFather, нажмите START и отправьте команду /newbot
|
||
2. Введите название бота, а потом username бота.
|
||
3. После создания бота перешлите ответное сообщение в этот бот или скопируйте и пришлите token бота.
|
||
|
||
Важно: не подключайте боты, которые используются в других сервисах (Manybot, Chatfuel, Livegram и других).
|
||
|
||
Подробную инструкцию по созданию бота читайте здесь /help)
|
||
"""))
|
||
await state.set_state("add_bot")
|
||
|
||
|
||
@router.message_handler(state="add_bot", content_types="text", regexp="^[^/].+") # Not command
|
||
async def bot_added(message: types.Message, state: FSMContext):
|
||
"""
|
||
Пользователь добавляет бота и мы ждём от него токен
|
||
"""
|
||
token = re.findall(token_pattern, message.text)
|
||
|
||
async def on_invalid_token():
|
||
await message.answer(dedent("""
|
||
Это не токен бота.
|
||
|
||
Токен выглядит вот так: 123456789:AAAA-abc123_AbcdEFghijKLMnopqrstu12
|
||
"""))
|
||
|
||
async def on_dummy_token():
|
||
await message.answer(dedent("""
|
||
Не удалось запустить этого бота: неверный токен
|
||
"""))
|
||
|
||
async def on_unknown_error():
|
||
await message.answer(dedent("""
|
||
Не удалось запустить этого бота: непредвиденная ошибка
|
||
"""))
|
||
|
||
async def on_duplication_bot():
|
||
await message.answer(dedent("""
|
||
Такой бот уже есть в базе данных
|
||
"""))
|
||
|
||
if not token:
|
||
return await on_invalid_token()
|
||
|
||
token = token[0]
|
||
|
||
try:
|
||
test_bot = AioBot(token)
|
||
test_bot_info = await test_bot.get_me()
|
||
await test_bot.session.close()
|
||
except ValueError:
|
||
return await on_invalid_token()
|
||
except Unauthorized:
|
||
return await on_dummy_token()
|
||
except TelegramAPIError:
|
||
return await on_unknown_error()
|
||
|
||
user, _ = await User.get_or_create(telegram_id=message.from_user.id)
|
||
bot = Bot(token=token, owner=user, name=test_bot_info.username, super_chat_id=message.from_user.id)
|
||
try:
|
||
await bot.save()
|
||
except IntegrityError:
|
||
return await on_duplication_bot()
|
||
|
||
await message.answer("Бот добавлен!")
|
||
await state.reset_state()
|