replace-bot/handlers/groups/main.py

import logging
import io
import base64

from aiogram import types
from aiogram.dispatcher.filters import ChatTypeFilter

from load import dp, bot, config
from database import set_group_settings, get_group
from parser import get_about_replacements
from keyboards.inline.donate import donate
from database import register


@dp.message_handler(ChatTypeFilter(['group', 'supergroup']), commands=['set'])
async def set_group(message: types.Message):
    if (message.from_user.id not in [admin.user.id for admin in await bot.get_chat_administrators(message.chat.id)]) and (str(message.from_user.id) not in config.admin_users.split(",")):
        await message.answer("Вы не являетесь администратором чата!")
        return
    args = message.text.split()
    if len(args) < 2:
        await message.answer(
            ("Вы не передали имя своей группы!\n"
            "Пример: /set 221")
        )
        return
    
    set_group_settings(message.chat.id, args[1])
    await message.answer("Настройка завершена успешно!")

@dp.message_handler(ChatTypeFilter(['group', 'supergroup']), commands=['start', 'get'])
async def get_replace_on_chat(message: types.Message):
    if config.enable_logging:
        register(
            user_id=message.from_user.id,
            username=message.from_user.username,
            first_name=str(message.from_user.first_name),
            last_name=message.from_user.last_name
        )

    logging.info("User: {user_id} - {username}".format(
        user_id=str(message.from_user.id),
        username=str(message.from_user.username)
    ))
    try:
        data = get_about_replacements()
        if 'image' in data:
            await bot.send_photo(
                message.chat.id,
                io.BytesIO(
                    base64.b64decode(
                        data["data"]['all']
                    )
                ),
                parse_mode="Markdown",
                reply_markup=donate
            )
            return
    except Exception as e:
         logging.error(e)
         await bot.send_message(message.chat.id, "Error")
         return
    group = get_group(message.chat.id)

    if group is not None:
        if group in data['data']:
            await message.answer(
                (
                    "Группа: {group}\n"
                    "Замены {date}\n"
                    "{teacher}\n"
                    "Замены: {replace}\n"
                ).format(
                    group=str(group),
                    replace=data['data'][group],
                    date=data["date"].lower(),
                    teacher=data["another_teacher"]
                )
            )
        else:
            await message.answer("Похоже замен нет")
    else:
        await message.answer("Похоже администратор группы не настроил привязку")