dev #19
@ -3,9 +3,10 @@ from os import makedirs, path
|
||||
from shutil import copyfileobj, rmtree
|
||||
from traceback import format_exc
|
||||
from uuid import uuid4
|
||||
from PIL import Image
|
||||
|
||||
from bson import ObjectId
|
||||
from pyrogram.client import Client
|
||||
from classes.poster_client import PosterClient
|
||||
from requests import get
|
||||
|
||||
from modules.api_client import authorize, move_pic, random_pic
|
||||
@ -14,7 +15,7 @@ from modules.logger import logWrite
|
||||
from modules.utils import configGet, locale
|
||||
|
||||
|
||||
async def send_content(app: Client):
|
||||
async def send_content(app: PosterClient):
|
||||
|
||||
try:
|
||||
|
||||
@ -52,6 +53,22 @@ async def send_content(app: Client):
|
||||
with open(path.join(configGet("tmp", "locations"), tmp_path), 'wb') as out_file:
|
||||
copyfileobj(response.raw, out_file)
|
||||
|
||||
logWrite(f'Candidate {pic[1]} ({pic[0]}) is {path.getsize(path.join(configGet("tmp", "locations"), tmp_path))} bytes big', debug=True)
|
||||
|
||||
if path.getsize(path.join(configGet("tmp", "locations"), tmp_path)) > 5242880:
|
||||
image = Image.open(path.join(configGet("tmp", "locations"), tmp_path))
|
||||
width, height = image.size
|
||||
image = image.resize((int(width/2), int(height/2)), Image.ANTIALIAS)
|
||||
if tmp_path.lower().endswith(".jpeg") or tmp_path.lower().endswith(".jpg"):
|
||||
image.save(path.join(configGet("tmp", "locations"), tmp_path), "JPEG", optimize=True, quality=50)
|
||||
elif tmp_path.lower().endswith(".png"):
|
||||
image.save(path.join(configGet("tmp", "locations"), tmp_path), "PNG", optimize=True, compress_level=8)
|
||||
image.close()
|
||||
|
||||
if path.getsize(path.join(configGet("tmp", "locations"), tmp_path)) > 5242880:
|
||||
rmtree(path.join(configGet("tmp", "locations"), tmp_dir), ignore_errors=True)
|
||||
raise BytesWarning
|
||||
|
||||
del response
|
||||
|
||||
submitted_caption = col_submitted.find_one( {"image": ObjectId(pic[0])} )
|
||||
@ -98,9 +115,13 @@ async def send_content(app: Client):
|
||||
logWrite(locale("post_exception", "console", locale=configGet("locale")).format(str(exp), format_exc()))
|
||||
if configGet("error", "reports"):
|
||||
await app.send_message(configGet("admin"), locale("post_exception", "message", locale=configGet("locale")).format(exp, format_exc()))
|
||||
try:
|
||||
rmtree(path.join(configGet("tmp", "locations"), tmp_dir), ignore_errors=True)
|
||||
except:
|
||||
pass
|
||||
|
||||
|
||||
# async def send_content_old(app: Client):
|
||||
# async def send_content_old(app: PosterClient):
|
||||
|
||||
# # Send post to channel
|
||||
# try:
|
||||
|
Reference in New Issue
Block a user