TelegramPoster/modules/sender.py

from os import listdir, remove, sep
from random import choice
from shutil import copyfileobj, move
from requests import get
from traceback import format_exc
from pyrogram.client import Client
from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton
from modules.logger import logWrite
from modules.api_client import authorize, move_pic, random_pic
from modules.utils import jsonLoad, jsonSave, configGet, locale

async def send_content(app: Client):

    # Send post to channel
    try:

        index = jsonLoad(configGet("index", "locations"))

        if configGet("api_based", "mode"):

            try:
                pic = random_pic()
            except:
                logWrite(locale("post_empty", "console", locale=configGet("locale")))
                if configGet("error", "reports"):
                    await app.send_message(configGet("admin"), locale("post_empty", "message", locale=configGet("locale")))
                return

            token = authorize()

            response = get(f'{configGet("address", "posting", "api")}/photos/{pic[0]}', headers={"Authorization": f"Bearer {token}"}, stream=True)

            with open(configGet("tmp", "locations")+sep+pic[0]+".jpg", 'wb') as out_file:
                copyfileobj(response.raw, out_file)

            del response

            candidate = configGet("tmp", "locations")+sep+pic[0]+".jpg"
            candidate_file = pic[1]
            ext_type = "photo"

        if not configGet("api_based", "mode"):

            list_queue = listdir(configGet("queue", "locations"))

            for file in list_queue:

                if not file in index["sent"]:

                    ext_match = False

                    for ext in configGet("photo", "posting", "extensions"):
                        if file.endswith(ext):
                            ext_match = True
                            ext_type = "photo"
                            break

                    for ext in configGet("video", "posting", "extensions"):
                        if file.endswith(ext):
                            ext_match = True
                            ext_type = "video"
                            break

                    if not ext_match:
                        list_queue.remove(file)

                else:
                    list_queue.remove(file)

            if len(list_queue) > 0:
                candidate_file = choice(list_queue)
                candidate = configGet("queue", "locations")+sep+candidate_file
            else:
                logWrite(locale("post_empty", "console", locale=configGet("locale")))
                if configGet("error", "reports"):
                    await app.send_message(configGet("admin"), locale("post_empty", "message", locale=configGet("locale")))
                return

        if candidate_file in index["captions"]:
            caption = index["captions"][candidate_file]
        else:
            caption = ""

        if configGet("enabled", "caption"):
            if configGet("link", "caption") != None:
                caption = f"{caption}\n\n[{configGet('text', 'caption')}]({configGet('link', 'caption')})"
            else:
                caption = f"{caption}\n\n{configGet('text', 'caption')}"
        else:
            caption = caption

        if ext_type == "photo":

            if configGet("enabled", "caption"):
                if configGet("link", "caption") != None:
                    sent = await app.send_photo(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting"))
                else:
                    sent = await app.send_photo(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting"))
            else:
                sent = await app.send_photo(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting"))

        elif ext_type == "video":

            if configGet("enabled", "caption"):
                if configGet("link", "caption") != None:
                    sent = await app.send_video(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting"))
                else:
                    sent = await app.send_video(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting"))
            else:
                sent = await app.send_video(configGet("channel", "posting"), candidate, caption=caption, disable_notification=configGet("silent", "posting"))

        else:
            return

        if configGet("api_based", "mode"):
            remove(configGet("tmp", "locations")+sep+pic[0]+".jpg")
            move_pic(pic[0])

        index["sent"].append(candidate_file)
        index["last_id"] = sent.id

        jsonSave(index, configGet("index", "locations"))

        if configGet("move_sent", "posting"):
            move(candidate, configGet("sent", "locations")+sep+candidate_file)

        logWrite(locale("post_sent", "console", locale=configGet("locale")).format(candidate, ext_type, str(configGet("channel", "posting")), caption.replace("\n", "%n"), str(configGet("silent", "posting"))))

        if configGet("sent", "reports"):
            await app.send_message(configGet("admin"), f"Posted `{candidate_file}`", disable_web_page_preview=True, reply_markup=InlineKeyboardMarkup([
                [InlineKeyboardButton(locale("post_view", "button", locale=configGet("locale")), url=sent.link)]
            ]))

    except Exception as exp:
        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()))
        pass


    # Work in progress
    # Check last posts forwards
    # check_forwards(app)