mirror of
https://github.com/EDeev/circlechek.git
synced 2026-06-15 11:01:00 +03:00
122 lines
6.4 KiB
Python
122 lines
6.4 KiB
Python
from aiogram import types, F, Router
|
||
from aiogram.types import Message, CallbackQuery, ContentType, InlineKeyboardButton, InlineKeyboardMarkup, FSInputFile
|
||
from aiogram.filters import Command
|
||
|
||
import shutil
|
||
|
||
from scripts import *
|
||
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, row_width=2)
|
||
|
||
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/IGlek">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 = "../data/circles/" + str(msg.chat.id) + ".mp4"
|
||
|
||
await msg.reply("<b>Началась обработка видео!</b> Оно должно быть квадратным и не дольше одной минуты, в ином "
|
||
"случае бот в ответ вернёт вам изначальное видео, а не кружочек!")
|
||
await msg.bot.download(file=msg.video.file_id, destination=video)
|
||
await msg.answer_video_note(video_note=FSInputFile(video))
|
||
os.remove(video)
|
||
|
||
|
||
# ОБРАБОТЧИК КРУЖОЧКОВ
|
||
@router.message(F.content_type == ContentType.VIDEO_NOTE)
|
||
async def video_note(msg: Message) -> None:
|
||
buttons = [[InlineKeyboardButton(text="Градиент", callback_data="grad"),
|
||
InlineKeyboardButton(text="Блюр", callback_data="blur")]]
|
||
keyboard = types.InlineKeyboardMarkup(inline_keyboard=buttons, row_width=2)
|
||
|
||
msg_answer = await msg.reply("Кружочек загружается...")
|
||
await msg.bot.download(file=msg.video_note.file_id, destination="../data/video_notes/" + str(msg.chat.id) + ".mp4")
|
||
await msg_answer.edit_text(text=WARN + "Какой тип фона в углах вы выберите?", reply_markup=keyboard)
|
||
|
||
|
||
@router.callback_query(lambda call: call.data == "grad" or call.data == "blur")
|
||
async def work_part(call: CallbackQuery) -> None:
|
||
msg = await call.message.edit_text(WARN + "<b>Начало обработки!</b>")
|
||
|
||
video_name = str(msg.chat.id)
|
||
video_file = video_name + ".mp4"
|
||
|
||
path = f"../data/videos/{video_name}"
|
||
os.mkdir(path)
|
||
|
||
path_video = path + "/" + video_file
|
||
os.replace("../data/video_notes/" + video_file, path_video)
|
||
|
||
path_frames = path + f"/frames"
|
||
os.mkdir(path_frames)
|
||
|
||
path_background = path + f"/background"
|
||
os.mkdir(path_background)
|
||
|
||
msg = await msg.edit_text(WARN + "<b>Этап:</b> 1 - Обработка видео.")
|
||
video = Movie(path_video, video_name, path)
|
||
procces = video.split_into_frames()
|
||
|
||
if not procces:
|
||
await msg.edit_text("<b>Возникла ошибка!</b> Кружочек невозможно обработать!")
|
||
shutil.rmtree(path)
|
||
return
|
||
|
||
msg = await msg.edit_text(WARN + "<b>Этап:</b> 2 - Обработка кадров.")
|
||
frames = list(sorted(os.listdir(path_frames)))
|
||
|
||
for frame in frames:
|
||
img = Frame(path_frames + "/" + frame, path_background + "/" + f'{frame[:-5]}-g.jpeg')
|
||
width, height = img.size()
|
||
|
||
if call.data == "blur":
|
||
img.blur(width, height)
|
||
else:
|
||
r, g, b = img.medium_color(); nearly = 10
|
||
img.gradient(width, height, (r - nearly, g - nearly, b - nearly),
|
||
(r + nearly, g + nearly, b + nearly), (True, False, False))
|
||
|
||
img.unity_image()
|
||
|
||
msg = await msg.edit_text(WARN + "<b>Этап:</b> 3 - Объединение кадров.")
|
||
video.unity_into_video(frames)
|
||
|
||
msg_final = await msg.edit_text(WARN + "<b>Готово!</b> Видео отправляется...")
|
||
|
||
await msg.answer_video(video=FSInputFile(path + "/acs-" + video_file), caption=WARN,
|
||
reply_to_message_id=call.message.reply_to_message.message_id)
|
||
await msg_final.delete()
|
||
|
||
shutil.rmtree(path)
|