from os import sep, remove, getpid from shutil import move from sys import exit from argparse import ArgumentParser from modules.logger import logWrite from modules.scheduler import scheduler from modules.utils import configGet, jsonLoad, jsonSave, killProc, locale # Args ===================================================================================================================================== parser = ArgumentParser( prog="Telegram Poster", description="Bot for posting some of your stuff and also receiving submissions.", ) parser.add_argument("-m", "--move-sent", action="store_true") parser.add_argument("-c", "--cleanup", action="store_true") parser.add_argument("--confirm", action="store_true") parser.add_argument("-i", "--cleanup-index", action="store_true") parser.add_argument("-n", "--norun", action="store_true") args = parser.parse_args() if args.move_sent: for entry in jsonLoad(configGet("index", "locations"))["sent"]: try: move( configGet("queue", "locations") + sep + entry, configGet("sent", "locations") + sep + entry, ) except FileNotFoundError: logWrite( locale( "move_sent_doesnt_exist", "console", locale=configGet("locale") ).format(entry) ) except Exception as exp: logWrite( locale( "move_sent_doesnt_exception", "console", locale=configGet("locale") ).format(entry, exp) ) logWrite(locale("move_sent_completed", "console", locale=configGet("locale"))) if args.cleanup: if args.confirm: index = jsonLoad(configGet("index", "locations")) for entry in index["sent"]: try: try: remove(configGet("queue", "locations") + sep + entry) except FileNotFoundError: pass try: remove(configGet("sent", "locations") + sep + entry) except FileNotFoundError: pass except Exception as exp: logWrite( locale( "cleanup_exception", "console", locale=configGet("locale") ).format(entry, exp) ) jsonSave(index, jsonLoad(configGet("index", "locations"))) logWrite(locale("cleanup_completed", "console", locale=configGet("locale"))) else: logWrite(locale("cleanup_unathorized", "console", locale=configGet("locale"))) if args.cleanup_index: if args.confirm: index = jsonLoad(configGet("index", "locations")) index["sent"] = [] jsonSave(index, jsonLoad(configGet("index", "locations"))) logWrite( locale("cleanup_index_completed", "console", locale=configGet("locale")) ) else: logWrite( locale("cleanup_index_unathorized", "console", locale=configGet("locale")) ) if args.norun: logWrite(locale("passed_norun", "console", locale=configGet("locale_log"))) exit() # =========================================================================================================================================== # Import =================================================================================================================================== try: from modules.app import app from pyrogram.sync import idle except ModuleNotFoundError: print(locale("deps_missing", "console", locale=configGet("locale")), flush=True) exit() # =========================================================================================================================================== pid = getpid() # Work in progress # def check_forwards(app): # try: # index = jsonLoad(configGet("index", "locations")) # channel = app.get_chat(configGet("channel", "posting")) # peer = app.resolve_peer(configGet("channel", "posting")) # print(peer, flush=True) # posts_list = [i for i in range(index["last_id"]-100,index["last_id"])] # last_posts = app.get_messages(configGet("channel", "posting"), message_ids=posts_list) # for post in last_posts: # post_forwards = GetMessagePublicForwards(channel=peer, msg_id=post.id, offset_peer=peer, offset_rate=0, offset_id=0, limit=100) # print(post_forwards, flush=True) # for forward in post_forwards: # print(forward, flush=True) # except Exception as exp: # logWrite("Could not get last posts forwards due to {0} with traceback {1}".format(str(exp), traceback.format_exc()), debug=True) # if configGet("error", "reports"): # app.send_message(configGet("admin"), traceback.format_exc()) # pass # Work in progress # @app.on_message(~ filters.scheduled & filters.command(["forwards"], prefixes="/")) # def cmd_forwards(app, msg): # check_forwards(app) # Imports ================================================================================================================================== from plugins.commands.general import * if configGet("submit", "mode"): from plugins.callbacks.nothing import * from plugins.callbacks.submission import * from plugins.commands.mode_submit import * from plugins.handlers.submission import * if configGet("api_based", "mode"): from modules.api_client import authorize # =========================================================================================================================================== # Work in progress # Handle new forwards # @app.on_raw_update() # def fwd_got(app, update, users, chats): # if isinstance(update, UpdateChannelMessageForwards): # logWrite(f'Forward count increased to {update["forwards"]} on post {update["id"]} in channel {update["channel_id"]}') # logWrite(str(users), debug=True) # logWrite(str(chats), debug=True) # # else: # # logWrite(f"Got raw update of type {type(update)} with contents {update}", debug=True) # async def main(): # await app.start() # logWrite(locale("startup", "console", locale=configGet("locale")).format(str(pid))) # if configGet("startup", "reports"): # await app.send_message(configGet("admin"), locale("startup", "message", locale=configGet("locale")).format(str(pid))) # if configGet("post", "mode"): # scheduler.start() # if configGet("api_based", "mode"): # token = authorize() # if len(get(f'{configGet("address", "posting", "api")}/albums?q={configGet("queue", "posting", "api", "albums")}', headers={"Authorization": f"Bearer {token}"}).json()["results"]) == 0: # post(f'{configGet("address", "posting", "api")}/albums?name={configGet("queue", "posting", "api", "albums")}&title={configGet("queue", "posting", "api", "albums")}', headers={"Authorization": f"Bearer {token}"}) # await idle() # await app.send_message(configGet("admin"), locale("shutdown", "message", locale=configGet("locale")).format(str(pid))) # logWrite(locale("shutdown", "console", locale=configGet("locale")).format(str(pid))) # killProc(pid) # if __name__ == "__main__": # if find_spec("uvloop") is not None: # uvloop.install() # asyncio.run(main()) if __name__ == "__main__": logWrite(locale("startup", "console", locale=configGet("locale")).format(str(pid))) app.start() if configGet("startup", "reports"): app.send_message( app.owner, locale("startup", "message", locale=configGet("locale")).format(str(pid)), ) if configGet("post", "mode"): scheduler.start() # if configGet("api_based", "mode"): # token = await authorize() # if len(get(f'{configGet("address", "posting", "api")}/albums?q={configGet("queue", "posting", "api", "albums")}', headers={"Authorization": f"Bearer {token}"}).json()["results"]) == 0: # post(f'{configGet("address", "posting", "api")}/albums?name={configGet("queue", "posting", "api", "albums")}&title={configGet("queue", "posting", "api", "albums")}', headers={"Authorization": f"Bearer {token}"}) idle() app.send_message( app.owner, locale("shutdown", "message", locale=configGet("locale")).format(str(pid)), ) logWrite(locale("shutdown", "console", locale=configGet("locale")).format(str(pid))) killProc(pid)