You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
4.5 KiB
98 lines
4.5 KiB
from telegram import ForceReply, Update
|
|
from telegram.ext import Application, CommandHandler, ContextTypes, MessageHandler, filters, ConversationHandler
|
|
import os, pathlib, asyncio
|
|
from time import time
|
|
|
|
async def add_watermark(input_file, watermark, transparency, out_ext):
|
|
#print(f'ffmpeg -i {input_file} -i {watermark} -filter_complex "[1]lut=a=val*0.08[a];[0][a]overlay=0:0" -c:v libx264 {input_file}.mp4')
|
|
resize_filter = "[logo][0]scale2ref=w=oh*mdar:h=ih*1.0[logo][0];"
|
|
insert_filter = "[0][logo]overlay=(W-w)/2:(H-h)/2:format=auto,format=yuv420p"
|
|
#insert_filter = "[1][logo]overlay=5:H-h-5"
|
|
filter = f"[1]format=rgba,colorchannelmixer=aa={transparency}[logo];{resize_filter}{insert_filter}"
|
|
cmd = f'ffmpeg -i {input_file} -i {watermark} -filter_complex "{filter}" {"-c:v libx264 -c:a copy" if out_ext == "mp4" else ""} {input_file}.{out_ext}'
|
|
print(cmd)
|
|
process = await asyncio.create_subprocess_shell(cmd)
|
|
await process.wait()
|
|
if os.path.exists(f"{input_file}.{out_ext}"):
|
|
return f"{input_file}.{out_ext}"
|
|
return ""
|
|
|
|
class Extension:
|
|
cmds = ["/watermark"]
|
|
wait_content = {}
|
|
content_directory = os.getenv("CONTENT_DIR") if os.getenv("CONTENT_DIR", None) else "./content"
|
|
png_path = os.getenv("WATERMARK")
|
|
default_transparency=20.0
|
|
watermark_users = []
|
|
|
|
def __init__(self, core) -> None:
|
|
self.watermark_users = core.admins
|
|
print(f"Content save path: {self.content_directory}")
|
|
print(f"Watermark users: {' '.join([str(i) for i in self.watermark_users])}")
|
|
|
|
async def __call__(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
if update.message.text and update.message.text.split()[0] in self.cmds:
|
|
return await self.watermark(update, context)
|
|
return await self.content_handler(update, context)
|
|
|
|
async def watermark(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
if not update.message.from_user.id in self.watermark_users:
|
|
return await update.message.reply_text("Это не для таких как ты сделано, друг...")
|
|
|
|
if update.message.from_user.id in self.wait_content:
|
|
del self.wait_content[update.message.from_user.id]
|
|
return await update.message.reply_text("Понял понял, не бей, отменяем думанье...")
|
|
else:
|
|
try:
|
|
self.wait_content[update.message.from_user.id] = float(update.message.text.split()[1])
|
|
except:
|
|
self.wait_content[update.message.from_user.id] = self.default_transparency
|
|
return await update.message.reply_text(f"Отлично, жду от тебя ОДНУ фотку/видео/переслал\nНепрозрачность: {self.wait_content[update.message.from_user.id]}%")
|
|
|
|
async def content_handler(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
if not update.message.from_user.id in self.wait_content:
|
|
return
|
|
if update.message.video:
|
|
return await self.video_handler(update, context)
|
|
elif update.message.photo:
|
|
return await self.photo_handler(update, context)
|
|
|
|
#Что ниже над будет засунуть в одно, ахуй
|
|
async def photo_handler(self, update:Update, context: ContextTypes.DEFAULT_TYPE):
|
|
if update.message.from_user.id in self.wait_content:
|
|
transparency = self.wait_content[update.message.from_user.id] / 100
|
|
del self.wait_content[update.message.from_user.id]
|
|
else:
|
|
return
|
|
|
|
_id = update.message.photo[-1].file_id
|
|
photo = await context.bot.get_file(_id)
|
|
downloaded_photo = await photo.download_to_drive(f"{self.content_directory}/{time()}")
|
|
res = await add_watermark(downloaded_photo.absolute(), self.png_path, transparency, "jpg")
|
|
if res:
|
|
await update.message.reply_photo(photo=res)
|
|
os.remove(res)
|
|
else:
|
|
await update.message.reply_text("Ошибка при создании фотки")
|
|
if os.path.exists(downloaded_photo.absolute()):
|
|
os.remove(downloaded_photo.absolute())
|
|
return
|
|
|
|
async def video_handler(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
if update.message.from_user.id in self.wait_content:
|
|
transparency = self.wait_content[update.message.from_user.id] / 100
|
|
del self.wait_content[update.message.from_user.id]
|
|
else:
|
|
return
|
|
|
|
video = await update.message.video.get_file()
|
|
downloaded_video = await video.download_to_drive(f"{self.content_directory}/{time()}")
|
|
res = await add_watermark(downloaded_video.absolute(), self.png_path, transparency, "mp4")
|
|
if res:
|
|
await update.message.reply_video(video=res)
|
|
os.remove(res)
|
|
else:
|
|
await update.message.reply_text("Ошибка при создании видео")
|
|
if os.path.exists(downloaded_video.absolute()):
|
|
os.remove(downloaded_video.absolute())
|
|
return
|