mirror of
https://github.com/EDeev/circlechek.git
synced 2026-10-07 20:49:51 +03:00
Egor Deev
e74873588a
- папка обработки была одна на чат: после первой же ошибки она оставалась, и все следующие кружочки в этом чате падали на os.mkdir — теперь своя временная папка на каждую обработку и уборка в любом случае; - кнопки фона знают свой кружочек: если прислать несколько подряд, обработается нужный; повторное нажатие отвечает подсказкой вместо ошибки; - покадровая обработка шла прямо в обработчике и блокировала бота для всех — вынесена в отдельный поток; - кружочек без звука падал на .audio.write_audiofile; - градиент по тёмным и светлым кадрам «переворачивался» (−10 в uint8 = 246): значения ограничены 0…255; средний цвет кадра считается NumPy, а не циклом по пикселям; - декоратор bug_report вызывал sys.exit() внутри обработчика — убран, ошибки логируются, пользователь получает сообщение; - видео, которое не удалось превратить в кружок, больше не остаётся на диске; - токен — из переменной BOT_TOKEN; ссылка «АВТОР» вела на чужой теперь профиль IGlek — заменена на EDeev; - requirements.txt: moviepy>=1.0.3 не подходил к коду под moviepy 2 — версии закреплены по работающему окружению.
133 lines
8 KiB
Python
133 lines
8 KiB
Python
import asyncio
|
||
import logging
|
||
import os
|
||
import shutil
|
||
import tempfile
|
||
|
||
from aiogram import F, Router
|
||
from aiogram.exceptions import TelegramBadRequest
|
||
from aiogram.types import Message, CallbackQuery, ContentType, InlineKeyboardButton, InlineKeyboardMarkup, FSInputFile
|
||
from aiogram.filters import Command
|
||
|
||
from scripts import Movie, process_frames
|
||
from config import WARN
|
||
|
||
router = Router()
|
||
|
||
|
||
# СТАРТОВАЯ КОМАНДА
|
||
@router.message(Command("start", "help"))
|
||
async def helps(msg: Message) -> None:
|
||
buttons = [[InlineKeyboardButton(text="ФУНКЦИИ", callback_data="fun"),
|
||
InlineKeyboardButton(text="АВТОР", callback_data="auth")]]
|
||
keyboard = InlineKeyboardMarkup(inline_keyboard=buttons)
|
||
|
||
await msg.answer(text="<b>Кружочичек</b> — бот для обработки видео и кружочков в Телеграме. Для начала работы "
|
||
"вам достаточно скинуть квадратное видео не дольше минуты в чат, чтобы получить кружок. "
|
||
"Или скинуть кружок, выбрать формат обработки углов и получить готовый видеоролик! "
|
||
"<b>НО</b> у Телеграмма бывают кружочки с некорректными метаданными, из-за чего его "
|
||
"обработка становится невозможной, или в итоге видео будет с браком, поэтому "
|
||
"<b>обязательно проверяйте</b> полученный результат!", reply_markup=keyboard)
|
||
|
||
|
||
@router.callback_query(F.data == "auth")
|
||
async def author(call: CallbackQuery) -> None:
|
||
await call.message.answer('<b>| АВТОР |</b>\n\n<b>>></b> Этот телеграм бот, крайне прост и примитивен. В его '
|
||
'распоряжении есть всего лишь две функции, а именно: превращение квадратных '
|
||
'видеороликов в кружочки и скачивание кружочков с последующей обработкой краёв в '
|
||
'двух предложенных вариантах.'
|
||
|
||
'\n\nЯ же пишу подобные небольшие проекты, о которых вы можете узнать '
|
||
'больше на моём <a href="https://github.com/EDeev">GitHub</a>.')
|
||
|
||
|
||
@router.callback_query(F.data == "fun")
|
||
async def function(call: CallbackQuery) -> None:
|
||
await call.message.answer('<b>| ФУНКЦИИ |</b>\n\n<b>1.</b> Обработка кружочка с градиентным или размытым фоном на '
|
||
'выбор по бокам\n<b>2.</b> Получение из квадратного видео длиною не больше минуты кружочек')
|
||
|
||
|
||
# ОБРАБОТЧИК ВИДЕО
|
||
@router.message(F.content_type == ContentType.VIDEO)
|
||
async def video_to_circle(msg: Message) -> None:
|
||
video = f"../data/circles/{msg.chat.id}_{msg.message_id}.mp4"
|
||
|
||
await msg.reply("<b>Началась обработка видео!</b> Оно должно быть квадратным и не дольше одной минуты, в ином "
|
||
"случае бот в ответ вернёт вам изначальное видео, а не кружочек!")
|
||
try:
|
||
await msg.bot.download(file=msg.video.file_id, destination=video)
|
||
await msg.answer_video_note(video_note=FSInputFile(video))
|
||
except TelegramBadRequest as err:
|
||
logging.warning("Не удалось сделать кружочек: %s", err)
|
||
await msg.answer("<b>Не получилось сделать кружочек.</b> Видео должно быть квадратным, не дольше минуты "
|
||
"и не больше 20 МБ.")
|
||
finally:
|
||
if os.path.exists(video):
|
||
os.remove(video)
|
||
|
||
|
||
# ОБРАБОТЧИК КРУЖОЧКОВ
|
||
@router.message(F.content_type == ContentType.VIDEO_NOTE)
|
||
async def video_note(msg: Message) -> None:
|
||
# номер сообщения в кнопках: если прислать несколько кружочков подряд, каждый обработается свой
|
||
buttons = [[InlineKeyboardButton(text="Градиент", callback_data=f"grad:{msg.message_id}"),
|
||
InlineKeyboardButton(text="Блюр", callback_data=f"blur:{msg.message_id}")]]
|
||
keyboard = InlineKeyboardMarkup(inline_keyboard=buttons)
|
||
|
||
msg_answer = await msg.reply("Кружочек загружается...")
|
||
await msg.bot.download(file=msg.video_note.file_id,
|
||
destination=f"../data/video_notes/{msg.chat.id}_{msg.message_id}.mp4")
|
||
await msg_answer.edit_text(text=WARN + "Какой тип фона в углах вы выберите?", reply_markup=keyboard)
|
||
|
||
|
||
@router.callback_query(F.data.regexp(r"^(grad|blur):\d+$"))
|
||
async def work_part(call: CallbackQuery) -> None:
|
||
mode, note_id = call.data.split(":")
|
||
video_name = f"{call.message.chat.id}_{note_id}"
|
||
video_file = video_name + ".mp4"
|
||
source = "../data/video_notes/" + video_file
|
||
|
||
if not os.path.exists(source):
|
||
# кнопку нажали повторно или кружочек уже обработан
|
||
await call.answer("Этот кружочек уже обработан. Пришлите его ещё раз.", show_alert=True)
|
||
return
|
||
|
||
msg = await call.message.edit_text(WARN + "<b>Начало обработки!</b>")
|
||
|
||
# отдельная папка на каждую обработку и уборка в любом случае — иначе после первой же ошибки
|
||
# повторная обработка в этом чате падала на уже существующей папке
|
||
path = tempfile.mkdtemp(prefix=video_name + "-", dir="../data/videos")
|
||
try:
|
||
path_video = path + "/" + video_file
|
||
os.replace(source, path_video)
|
||
|
||
path_frames = path + "/frames"
|
||
os.mkdir(path_frames)
|
||
|
||
path_background = path + "/background"
|
||
os.mkdir(path_background)
|
||
|
||
# тяжёлая обработка — в отдельном потоке, чтобы бот не замирал для остальных пользователей
|
||
msg = await msg.edit_text(WARN + "<b>Этап:</b> 1 - Обработка видео.")
|
||
video = Movie(path_video, video_name, path)
|
||
if not await asyncio.to_thread(video.split_into_frames):
|
||
await msg.edit_text("<b>Возникла ошибка!</b> Кружочек невозможно обработать!")
|
||
return
|
||
|
||
msg = await msg.edit_text(WARN + "<b>Этап:</b> 2 - Обработка кадров.")
|
||
frames = await asyncio.to_thread(process_frames, path_frames, path_background, mode)
|
||
|
||
msg = await msg.edit_text(WARN + "<b>Этап:</b> 3 - Объединение кадров.")
|
||
result = await asyncio.to_thread(video.unity_into_video, frames)
|
||
|
||
msg_final = await msg.edit_text(WARN + "<b>Готово!</b> Видео отправляется...")
|
||
|
||
reply_to = call.message.reply_to_message
|
||
await msg.answer_video(video=FSInputFile(result), caption=WARN,
|
||
reply_to_message_id=reply_to.message_id if reply_to else None)
|
||
await msg_final.delete()
|
||
except Exception:
|
||
logging.exception("Ошибка обработки кружочка")
|
||
await msg.edit_text("<b>Возникла ошибка!</b> Кружочек невозможно обработать!")
|
||
finally:
|
||
shutil.rmtree(path, ignore_errors=True)
|