AutoShop-Djimbo/tgbot/utils/misc_functions.py

# - *- coding: utf- 8 - *-
import asyncio
import json
from datetime import datetime
from typing import Union

from aiogram import Bot
from aiogram.types import FSInputFile, CallbackQuery, Message

from tgbot.data.config import BOT_DATABASE_EXPORT, BOT_STATUS_NOTIFICATION, BOT_VERSION, PATH_DATABASE, get_admins, \
    get_text_desc
from tgbot.database import Userx, Settingsx
from tgbot.utils.const_functions import get_unix, get_date, ded, send_admins
from tgbot.utils.misc.bot_logging import bot_logger
from tgbot.utils.misc.bot_models import ARS
from tgbot.utils.text_functions import get_statistics


# Автоматическая очистка ежедневной статистики после 00:00:15
async def update_profit_day(bot: Bot):
    await send_admins(bot, await get_statistics())

    await Settingsx().update(misc_profit_day=get_unix())


# Автоматическая очистка еженедельной статистики в понедельник 00:00:10
async def update_profit_week():
    await Settingsx().update(misc_profit_week=get_unix())


# Автоматическое обновление счётчика каждый месяц первого числа в 00:00:05
async def update_profit_month():
    await Settingsx().update(misc_profit_month=get_unix())


# Автонастройка UNIX времени в БД
async def autosettings_unix():
    now_day = datetime.now().day
    now_week = datetime.now().weekday()
    now_month = datetime.now().month
    now_year = datetime.now().year

    unix_day = int(datetime.strptime(f"{now_day}.{now_month}.{now_year} 0:0:0", "%d.%m.%Y %H:%M:%S").timestamp())
    unix_week = unix_day - (now_week * 86400)
    unix_month = int(datetime.strptime(f"1.{now_month}.{now_year} 0:0:0", "%d.%m.%Y %H:%M:%S").timestamp())

    await Settingsx().update(
        misc_profit_day=unix_day,
        misc_profit_week=unix_week,
        misc_profit_month=unix_month,
    )


# Проверка на перенесение БД из старого бота в нового или указание токена нового бота
async def check_bot_username(bot: Bot):
    get_login = await Settingsx().get()
    get_bot = await bot.get_me()

    if get_bot.username != get_login.misc_bot:
        await Settingsx().update(misc_bot=get_bot.username)


# Уведомление и проверка обновления при запуске бота
async def startup_notify(bot: Bot, arSession: ARS):
    if len(get_admins()) >= 1 and BOT_STATUS_NOTIFICATION:
        await send_admins(
            bot,
            ded(f"""
                <b>✅ Бот был успешно запущен</b>
                ➖➖➖➖➖➖➖➖➖➖
                {get_text_desc()}
                ➖➖➖➖➖➖➖➖➖➖
                <code>❗ Данное сообщение видят только администраторы бота.</code>
            """),
        )

        await check_update(bot, arSession)


# Автобэкапы БД для админов
async def autobackup_admin(bot: Bot):
    if not BOT_DATABASE_EXPORT:
        return

    for admin in get_admins():
        try:
            await bot.send_document(
                admin,
                FSInputFile(PATH_DATABASE),
                caption=f"<b>📦 #BACKUP | <code>{get_date(full=False)}</code></b>",
                disable_notification=True,
            )
        except Exception:
            bot_logger.warning("Не удалось отправить автобэкап админу %s", admin, exc_info=True)


# Проверка наличия обновлений бота
async def check_update(bot: Bot, arSession: ARS):
    session = await arSession.get_session()

    try:
        response = await session.get(
            "https://djimbo.dev/autoshop_update.json",
            headers={"Accept-Encoding": "gzip, deflate"},
        )

        response_data = json.loads((await response.read()).decode())

        if float(response_data['version']) > float(BOT_VERSION):
            await send_admins(
                bot,
                ded(f"""
                    <b>❇️ Вышло обновление: <a href='{response_data['download']}'>Скачать</a></b>
                    ➖➖➖➖➖➖➖➖➖➖
                    {response_data['text']}
                    ➖➖➖➖➖➖➖➖➖➖
                    <code>❗ Данное сообщение видят только администраторы бота.</code>
                """),
            )
    except Exception:
        bot_logger.warning("Не удалось проверить обновления", exc_info=True)


# Расссылка админам о критических ошибках и обновлениях
async def check_mail(bot: Bot, arSession: ARS):
    session = await arSession.get_session()

    try:
        response = await session.get(
            "https://djimbo.dev/autoshop_mail.json",
            headers={"Accept-Encoding": "gzip, deflate"},
        )
        response_data = json.loads((await response.read()).decode())

        if response_data['status']:
            await send_admins(
                bot,
                ded(f"""
                    {response_data['text']}
                    ➖➖➖➖➖➖➖➖➖➖
                    <code>❗ Данное сообщение видят только администраторы бота.</code>
                """),
            )
    except Exception:
        bot_logger.warning("Не удалось проверить сервисные уведомления", exc_info=True)


# Вставка кастомных тэгов юзера в текст
async def insert_tags(user_id: Union[int, str], text: str) -> str:
    get_user = await Userx().get_required(user_id=user_id)

    if "{user_id}" in text:
        text = text.replace("{user_id}", f"<b>{get_user.user_id}</b>")
    if "{username}" in text:
        text = text.replace("{username}", f"<b>{get_user.user_login}</b>")
    if "{firstname}" in text:
        text = text.replace("{firstname}", f"<b>{get_user.user_name}</b>")

    return text


# Отправка рассылки
async def functions_mail_make(bot: Bot, message: Message, call: CallbackQuery):
    users_receive, users_block, users_count = 0, 0, 0

    get_users = await Userx().get_all()
    get_time = get_unix()

    for user in get_users:
        try:
            await bot.copy_message(
                chat_id=user.user_id,
                from_chat_id=message.from_user.id,
                message_id=message.message_id,
            )
            users_receive += 1
        except Exception:
            users_block += 1
            bot_logger.debug("Пользователь %s не получил рассылку", user.user_id, exc_info=True)

        users_count += 1

        if users_count % 10 == 0:
            await call.message.edit_text(f"<b>📢 Рассылка началась... ({users_count}/{len(get_users)})</b>")

        await asyncio.sleep(0.07)

    await call.message.edit_text(
        ded(f"""
            <b>📢 Рассылка была завершена за <code>{get_unix() - get_time}сек</code></b>
            ➖➖➖➖➖➖➖➖➖➖
            👤 Всего пользователей: <code>{len(get_users)}</code>
            ✅ Пользователей получило сообщение: <code>{users_receive}</code>
            ❌ Пользователей не получило сообщение: <code>{users_block}</code>
        """)
    )