|
|
|
|
|
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: |
|
|
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) |
|
|
|
