vkbot.py

import os

from dotenv import load_dotenv
from fastapi import FastAPI, Request
import uvicorn

from loguru import logger
from vkbottle.bot import Bot, Message

from consts import ADMINS, GROUP_ID_COEFFICIENT
from db_interface import (add_group, groups_ids,
                          delete_group, ids_by_course, init_database)

app = FastAPI()

load_dotenv()

bot = Bot(os.getenv('VKTOKEN', 'NoToken'))

init_database()


async def broadcast(courses: str, text=None, attachment=None):
    for course in courses:
        for group in ids_by_course(int(course)):
            try:
                await bot.api.messages.send(
                    peer_id=(GROUP_ID_COEFFICIENT + group),
                    message=text,
                    attachment=attachment,
                    random_id=0
                )
            except Exception as exception:
                logger.warning(exception)
                delete_group(group)


@bot.on.chat_message(text='Рассылка: <courses>, Текст <text>')
async def sharing_text(message: Message, courses: str, text: str):
    if message.from_id not in ADMINS:
        return
    await broadcast(courses, text=text)


@bot.on.chat_message(text='Рассылка: <courses>, Пост')
async def share_publication(message: Message, courses: str):
    if message.from_id not in ADMINS:
        return
    attachment = message.get_wall_attachment()[0]
    await broadcast(
        courses,
        attachment=[f"wall{attachment.owner_id}_{attachment.id}"]
    )


@bot.on.chat_message(text='Рассылка: <courses>, Сообщение')
async def share_message(message: Message, courses: str):
    if message.from_id not in ADMINS:
        return

    if message.fwd_messages:
        await broadcast(courses, text=message.fwd_messages[0].text)
    else:
        await message.answer('Ошибка: нет пересланного сообщения')


@bot.on.chat_message(text='Добавить <course>')
async def test(message: Message, course: str):
    if course == 'admin':
        course = -1
    elif course.isnumeric() and 1 <= int(course) <= 5:
        course = int(course)
    else:
        await message.answer("Не верно введен курс!")
        return

    group_id = message.peer_id - GROUP_ID_COEFFICIENT

    if group_id in groups_ids():
        await message.answer('Ваша беседа уже есть в списке')
        return

    add_group(group_id, course)

    await message.answer('Ваша беседа успешно добавлена!')
    await message.answer(
        'Добро пожаловать в беседу!\n\n'
        'Этот чат создан специально для '
        f'старост {course} курса факультета ИКСС. '
        'Здесь будет собрана только важная информация, '
        'которую вы обязаны знать и/или распространить!\n\n'
        'Сейчас вам необходимо ознакомиться с правилами чата.\n'
        '🟧Писать здесь могут только:\n'
        '👉🏼 Староста\n'
        '👉🏼 Зам. старосты\n'
        '👉🏼 Бот\n'
        'Для остальных участников данный чат доступен '
        'только для просмотра информации.\n'
        '🟧Запрещено писать сообщения без предварительного '
        'согласования их со старостой курса, указанным в пункте выше.\n\n'
        '🟧 Полезные ресурсы:\n'
        '✅Бот в Телеграм: https://t.me/BonchGUTBot\n'
        '✅Сайт Бонча: https://www.sut.ru\n'
        '✅ГУТ.Навигатор: https://nav.sut.ru/?cab=k2-117\n'
        '✅Студгородок: https://vk.com/campusut\n'
        '✅Факультет ИКСС: https://vk.com/iksssut\n'
        '✅Группа СПбГУТ: https://vk.com/sutru\n'
        '✅Студсовет: https://vk.com/studsovet.bonch\n'
        '✅InGUT: https://vk.com/ingut\n'
        '✅Подслушано Бонч: https://vk.com/overhear_bonch\n'
        '✅Bonch Media: https://vk.com/bonch.media\n'
        '✅Первокурсники СПбГУТ: https://vk.com/onegut\n\n'
        'По вопросам и предложениям писать @pavel.cmake(разработчику)'
    )


@bot.on.chat_message(text='Помощь')
async def user_help(message: Message):
    if message.from_id in ADMINS:
        await message.answer(
            'Команды:\n\n'
            'Добавить <course> - Добавляет беседу в БД, '
            'флаг admin значит что в беседу не будут приходить новости\n\n'
            'Рассылка: <courses>, Сообщение - '
            'Рассылает пересланное сообщение\n\n'
            'Рассылка: <courses>, Пост - Рассылает пересланный пост\n\n'
            'Рассылка: <courses>, Текст <text> - Рассылает набранный текст'
        )


@app.post("/callback")
async def callback(request: Request):
    data = await request.json()
    print(data)
    await bot.process_event([data])


if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=80)