This repository has been archived on 2024-08-21. You can view files and clone it, but cannot push or open issues or pull requests.
TelegramPoster/modules/sender.py

127 lines
5.7 KiB
Python
Raw Normal View History

2023-02-17 22:55:38 +02:00
from datetime import datetime
2023-02-14 17:25:56 +02:00
from os import makedirs, path
from shutil import copyfileobj, rmtree
2023-01-10 13:52:44 +02:00
from traceback import format_exc
2023-02-14 17:25:56 +02:00
from uuid import uuid4
2023-02-17 17:46:44 +02:00
from PIL import Image
2023-02-14 17:25:56 +02:00
2023-02-17 17:46:44 +02:00
from classes.poster_client import PosterClient
2023-02-14 17:25:56 +02:00
from requests import get
2023-01-17 15:11:23 +02:00
from modules.api_client import authorize, move_pic, random_pic
2023-02-14 17:25:56 +02:00
from modules.database import col_sent, col_submitted
from modules.logger import logWrite
from modules.utils import configGet, locale
2023-01-10 13:52:44 +02:00
async def send_content(app: PosterClient) -> None:
2023-01-10 13:52:44 +02:00
try:
2023-02-14 17:25:56 +02:00
try:
token = await authorize()
except ValueError:
2023-02-17 22:55:38 +02:00
await app.send_message(app.owner, locale("api_creds_invalid", "message", locale=configGet("locale")))
2023-02-14 17:25:56 +02:00
return
2023-01-17 15:11:23 +02:00
2023-02-14 17:25:56 +02:00
try:
pic = await random_pic()
except KeyError:
logWrite(locale("post_empty", "console", locale=configGet("locale")))
if configGet("error", "reports"):
2023-02-17 22:55:38 +02:00
await app.send_message(app.owner, locale("api_queue_empty", "message", locale=configGet("locale")))
2023-02-14 17:25:56 +02:00
return
except ValueError:
if configGet("error", "reports"):
2023-02-17 22:55:38 +02:00
await app.send_message(app.owner, locale("api_queue_error", "message", locale=configGet("locale")))
2023-02-14 17:25:56 +02:00
return
response = get(f'{configGet("address", "posting", "api")}/photos/{pic[0]}', headers={"Authorization": f"Bearer {token}"}, stream=True)
2023-01-17 15:11:23 +02:00
2023-02-14 17:25:56 +02:00
if response.status_code != 200:
logWrite(locale("post_invalid_pic", "console", locale=configGet("locale")).format(str(response.json())))
if configGet("error", "reports"):
2023-02-17 22:55:38 +02:00
await app.send_message(app.owner, locale("post_invalid_pic", "message", locale=configGet("locale")).format(response.json()))
2023-01-17 15:11:23 +02:00
2023-02-14 17:25:56 +02:00
tmp_dir = str(uuid4())
2023-01-17 15:11:23 +02:00
2023-02-14 17:25:56 +02:00
makedirs(path.join(configGet("tmp", "locations"), tmp_dir), exist_ok=True)
2023-01-17 15:11:23 +02:00
2023-02-14 17:25:56 +02:00
tmp_path = path.join(tmp_dir, pic[1])
2023-01-17 15:11:23 +02:00
2023-02-14 17:25:56 +02:00
with open(path.join(configGet("tmp", "locations"), tmp_path), 'wb') as out_file:
copyfileobj(response.raw, out_file)
2023-01-17 15:11:23 +02:00
2023-02-17 17:46:44 +02:00
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
2023-02-14 17:25:56 +02:00
del response
2023-01-10 13:52:44 +02:00
2023-02-18 00:18:34 +02:00
submitted = col_submitted.find_one({"temp.file": pic[1]})
2023-01-10 13:52:44 +02:00
2023-02-18 00:18:34 +02:00
if submitted is not None and submitted["caption"] is not None:
caption = submitted["caption"].strip()
2023-01-10 13:52:44 +02:00
else:
caption = ""
2023-02-24 22:06:18 +02:00
if submitted is not None and configGet("enabled", "posting", "submitted_caption") and (
2023-02-24 20:48:06 +02:00
(submitted["user"] not in app.admins) or (configGet("ignore_admins", "posting", "submitted_caption") is False)
):
2023-02-24 19:39:33 +02:00
caption = f"{caption}\n\n{configGet('text', 'posting', 'submitted_caption')}\n"
else:
caption = f"{caption}\n\n"
2023-01-10 13:52:44 +02:00
if configGet("enabled", "caption"):
if configGet("link", "caption") != None:
2023-02-24 19:39:33 +02:00
caption = f"{caption}[{configGet('text', 'caption')}]({configGet('link', 'caption')})"
2023-01-10 13:52:44 +02:00
else:
2023-02-24 19:39:33 +02:00
caption = f"{caption}{configGet('text', 'caption')}"
2023-01-10 13:52:44 +02:00
else:
caption = caption
2023-02-14 17:25:56 +02:00
try:
sent = await app.send_photo(configGet("channel", "posting"), path.join(configGet("tmp", "locations"), tmp_path), caption=caption, disable_notification=configGet("silent", "posting"))
except Exception as exp:
logWrite(f"Could not send image {pic[1]} ({pic[0]}) due to {exp}")
if configGet("error", "reports"):
2023-02-17 22:55:38 +02:00
await app.send_message(app.owner, locale("post_exception", "message", locale=configGet("locale")).format(exp, format_exc()))
2023-02-14 17:25:56 +02:00
# rmtree(path.join(configGet("tmp", "locations"), tmp_dir), ignore_errors=True)
return
2023-01-10 13:52:44 +02:00
2023-02-14 17:25:56 +02:00
col_sent.insert_one(
{
2023-02-17 22:55:38 +02:00
"date": datetime.now(),
2023-02-14 17:25:56 +02:00
"image": pic[0],
"filename": pic[1],
"channel": configGet("channel", "posting"),
2023-02-18 00:18:34 +02:00
"caption": None if (submitted is None or submitted["caption"] is None) else submitted["caption"].strip()
2023-02-14 17:25:56 +02:00
}
)
2023-01-10 13:52:44 +02:00
2023-02-14 17:25:56 +02:00
await move_pic(pic[0])
2023-01-10 13:52:44 +02:00
2023-02-14 17:25:56 +02:00
rmtree(path.join(configGet("tmp", "locations"), tmp_dir), ignore_errors=True)
2023-01-10 13:52:44 +02:00
2023-02-14 17:25:56 +02:00
logWrite(locale("post_sent", "console", locale=configGet("locale")).format(pic[0], str(configGet("channel", "posting")), caption.replace("\n", "%n"), str(configGet("silent", "posting"))))
2023-01-10 13:52:44 +02:00
2023-02-14 17:25:56 +02:00
except Exception as exp:
logWrite(locale("post_exception", "console", locale=configGet("locale")).format(str(exp), format_exc()))
if configGet("error", "reports"):
2023-02-17 22:55:38 +02:00
await app.send_message(app.owner, locale("post_exception", "message", locale=configGet("locale")).format(exp, format_exc()))
2023-02-17 17:46:44 +02:00
try:
rmtree(path.join(configGet("tmp", "locations"), tmp_dir), ignore_errors=True)
except:
pass