Stan-Ansible/app/commands/service.py

""" Service commands for Admin purposes. """

import logging
from telebot import logger
from datetime import datetime as dt
from ..models import Chat, Quote, session
from ..config import bot, ADMIN_ID, types
from .. import reminder
from .. import report
from ..filters import is_admin, is_white
from .get import me, my_ip


@bot.message_handler(commands=["me"])
async def command_me(message):
    """Send info about user and chat id [Service]."""
    await bot.send_message(message.chat.id, me(message))


@bot.message_handler(func=is_admin, commands=["ip"])
async def get_ip(message):
    await bot.send_message(message.chat.id, my_ip())


@bot.message_handler(commands=["remind"])
async def remind_manually(message):
    """Remind holidays manually."""
    args = message.text.split()
    if len(args) > 1:
        try:
            today = dt.strptime(args[1], "%m-%d-%Y")
        except ValueError as ve:
            await bot.send_message(message.chat.id, f"Не удалось разобрать дату!\n{ve}")
        else:
            await reminder.remind(message.chat.id, today)
    else:
        await bot.send_message(
            message.chat.id,
            f"<b>Формат даты: MM-DD-YYYY</b>\n\n"
            f"Примеры:\n"
            f"/remind 09-12-2024\n"
            f"/remind 09-13-2022",
        )


@bot.message_handler(func=is_admin, commands=["jobs"])
async def list_jobs(message):
    """List all the jobs in schedule."""
    await bot.send_message(ADMIN_ID, reminder.print_get_jobs())


@bot.message_handler(func=is_admin, commands=["stats"])
async def send_stats(message: types.Message):
    if len(message.text.split()) == 1:
        rep = report.create_report_text(message.chat.id)
        if rep:
            await bot.send_message(message.chat.id, rep)
    else:
        await bot.send_message(message.chat.id, report.create_report_text(message.text.split()[-1]))


@bot.message_handler(func=is_admin, commands=["reset_stats"], chat_types=["supergroup", "group"])
async def send_stats(message: types.Message):
    logging.warning("reset_stats")
    report.reset_report_stats(message.chat.id)
    await bot.send_message(message.chat.id, report.reset_report_stats(message.chat.id))


@bot.message_handler(func=is_white, commands=["enable_stan"], chat_types=["supergroup", "group"])
async def enable_stan(message: types.Message):
    """Add group to database."""

    logging.info(
        f"[{message.chat.title}] [{message.from_user.id}] {message.from_user.username}: {message.text}"
    )
    if not session.query(Chat.id).filter(Chat.chat_id == message.chat.id).first():
        session.add(
            Chat(
                chat_id=message.chat.id,
                title=message.chat.title,
                antispam=1,
                report=0,
                reminder=1,
            )
        )
        session.commit()
        await bot.send_message(
            message.chat.id,
            f"""Группа "{message.chat.title} добавлена в БД.
/get_group_info - узнать текущие настройки""",
        )
    else:
        await bot.send_message(message.chat.id, "Отказ. Группа уже включена")


@bot.message_handler(func=is_admin, commands=["disable_stan"], chat_types=["supergroup", "group"])
async def disable_stan(message: types.Message):
    logging.info(f"[{message.chat.title}] [{message.from_user.id}] {message.from_user.username}: {message.text}")
    if session.query(Chat.id).filter(Chat.chat_id == message.chat.id).first():
        chat = session.query(Chat).filter_by(chat_id=message.chat.id).first()
        session.delete(chat)
        session.commit()
        await bot.send_message(
            message.chat.id,
            f"Группа {message.chat.title} и все связанные с ней цитаты удалёны!",
        )
    else:
        await bot.send_message(message.chat.id, "Отказ. Этой группы нет в БД.")


@bot.message_handler(
    func=is_white,
    commands=["set_antispam_report_reminder", "set_rules"],
    chat_types=["supergroup", "group"],
)
async def set_antispam_report_reminder(message: types.Message):
    args = message.text.split()
    if len(args) == 4:
        try:
            antispam = int(args[1])
            rep = int(args[2])
            rem = int(args[3])
        except ValueError:
            logging.info(f"[ERROR] Неверные аргументы {message.text}")
        else:
            session.query(Chat).filter_by(chat_id=message.chat.id).update(
                {"antispam": antispam, "report": rep, "reminder": rem}
            )
            session.commit()
            await bot.send_message(message.chat.id, f"""Настройки обновлены. Проверить: /get_group_info""")


@bot.message_handler(
    func=is_white, commands=["get_quotes"], chat_types=["supergroup", "group"]
)
async def get_quotes(message: types.Message):
    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()
    if quotes:
        text = f"Количество цитат: {len(session.query(Quote).filter(Quote.chat_id == message.chat.id).all())}\n\
Последние добавленные\n\n"
        text += "\n".join(f"· {quote[0]}" for quote in quotes[-10:])
        await bot.send_message(message.chat.id, f"{text}")
    else:
        await bot.send_message(
            message.chat.id, f"Цитаты отсутствуют. Подробнее: /get_group_info"
        )


@bot.message_handler(func=is_white, commands=["get_group_info"], chat_types=["supergroup", "group"])
async def get_group_info(message: types.Message):
    group = session.query(Chat).filter(Chat.chat_id == message.chat.id).first()
    if group:
        await bot.send_message(
            message.chat.id,
            f"""Группа: {group.title}
ID группы: {group.chat_id} 

Количество цитат:  {len(session.query(Quote).filter(Quote.chat_id == message.chat.id).all())}
Последние добавленные: /get_quotes

Текущие настройки:
  Антиспам: {group.antispam}
  Ежедневные отчёты: {group.report}
  Праздники: {group.reminder}""",
        )
    else:
        await bot.send_message(message.chat.id, f"Группа не включена. Включить: /enable_stan")


@bot.message_handler(func=is_white, commands=["set_logging_level"], chat_types=["supergroup", "group"])
async def set_logging_level(message: types.Message):
    args = message.text.split()
    if args[-1] == '10':
        logger.setLevel(10)  # DEBUG
        logging.info("[LEVEL] Установлен уровень DEBUG")
    else:
        logger.setLevel(20)  # INFO
        logging.info("[LEVEL] Установлен уровень INFO")