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)