File size: 3,362 Bytes
afce118 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 |
# Copyright (c) 2023 EDM115
import os
from PIL import Image
from config import Config
from unzipper import LOGGER
from unzipper.modules.bot_data import Buttons, Messages
from pyrogram.errors import FloodWait
from asyncio import sleep
async def silent_del(user_id):
try:
thumb_location = Config.THUMB_LOCATION + "/" + str(user_id) + ".jpg"
os.remove(thumb_location)
except:
pass
async def add_thumb(_, message):
try:
user_id = str(message.from_user.id)
if message.reply_to_message is not None:
reply_message = message.reply_to_message
if reply_message.media_group_id is not None: # album sent
LOGGER.info(Messages.ALBUM.format(user_id))
await message.reply(Messages.ALBUM_NOPE)
return
thumb_location = Config.THUMB_LOCATION + "/" + user_id + ".jpg"
pre_thumb = Config.THUMB_LOCATION + "/not_resized_" + user_id + ".jpg"
final_thumb = Config.THUMB_LOCATION + "/waiting_" + user_id + ".jpg"
if os.path.exists(thumb_location) and os.path.isfile(thumb_location):
await message.reply(
text=Messages.EXISTING_THUMB, reply_markup=Buttons.THUMB_REPLACEMENT
)
else:
await message.reply(
text=Messages.SAVING_THUMB, reply_markup=Buttons.THUMB_SAVE
)
LOGGER.info(Messages.DL_THUMB.format(user_id))
file = await _.download_media(message=reply_message)
await _.send_document(
chat_id=Config.LOGS_CHANNEL,
document=file,
file_name=file.split("/")[-1],
caption=Messages.EXT_CAPTION.format(file),
)
os.rename(file, pre_thumb)
size = 320, 320
try:
with Image.open(pre_thumb) as previous:
previous.thumbnail(size, Image.Resampling.LANCZOS)
previous.save(final_thumb, "JPEG")
LOGGER.info(Messages.THUMB_SAVED)
except:
LOGGER.info(Messages.THUMB_FAILED)
try:
os.remove(pre_thumb)
except:
pass
try:
os.remove(final_thumb)
except:
pass
await message.reply(Messages.THUMB_ERROR)
else:
await _.send_message(
chat_id=message.chat.id,
text=Messages.PLS_REPLY,
reply_to_message_id=message.id,
)
except FloodWait as f:
await sleep(f.value)
return await add_thumb(_, message)
async def del_thumb(message):
try:
uid = message.from_user.id
thumb_location = Config.THUMB_LOCATION + "/" + str(uid) + ".jpg"
if not os.path.exists(thumb_location):
await message.reply(text=Messages.NO_THUMB)
else:
await message.reply(text=Messages.DEL_CONFIRM_THUMB, reply_markup=Buttons.THUMB_DEL)
except FloodWait as f:
await sleep(f.value)
return await del_thumb(message)
async def thumb_exists(chat_id):
thumb_location = Config.THUMB_LOCATION + "/" + str(chat_id) + ".jpg"
return os.path.exists(thumb_location)
|