from datetime import datetime from os import getpid, path from sys import exit from time import time from traceback import format_exc from modules.api_client import authorize from modules.cli import * from modules.http_client import http_session from modules.logger import logWrite from modules.scheduler import scheduler from modules.utils import configGet, jsonLoad, jsonSave, locale # Import =================================================================================================================================== try: from dateutil.relativedelta import relativedelta from pyrogram.errors import bad_request_400 from pyrogram.sync import idle from modules.app import app except ModuleNotFoundError: print(locale("deps_missing", "console", locale=configGet("locale")), flush=True) exit() # =========================================================================================================================================== pid = getpid() version = 0.1 # 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) from plugins.callbacks.shutdown import * # 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("post", "mode"): from plugins.commands.photos 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()) async def main(): logWrite(locale("startup", "console").format(str(pid))) await app.start() if configGet("startup", "reports"): try: if path.exists(path.join(configGet("cache", "locations"), "shutdown_time")): downtime = relativedelta( datetime.now(), datetime.fromtimestamp( jsonLoad( path.join(configGet("cache", "locations"), "shutdown_time") )["timestamp"] ), ) if downtime.days >= 1: await app.send_message( configGet("owner"), locale( "startup_downtime_days", "message", ).format(pid, downtime.days), ) elif downtime.hours >= 1: await app.send_message( configGet("owner"), locale( "startup_downtime_hours", "message", ).format(pid, downtime.hours), ) else: await app.send_message( configGet("owner"), locale( "startup_downtime_minutes", "message", locale=configGet("locale"), ).format(pid, downtime.minutes), ) else: await app.send_message( configGet("owner"), locale("startup", "message").format(pid), ) except bad_request_400.PeerIdInvalid: logWrite( f"Could not send startup message to bot owner. Perhaps user has not started the bot yet." ) if configGet("update", "reports"): try: check_update = await http_session.get( "https://git.end-play.xyz/api/v1/repos/profitroll/TelegramPoster/releases?page=1&limit=1" ) response = await check_update.json() if len(response) == 0: raise ValueError("No bot releases on git found.") if float(response[0]["tag_name"].replace("v.", "")) > version: logWrite( f'New version {response[0]["tag_name"].replace("v.", "")} found (current {version})' ) await app.send_message( configGet("owner"), locale( "update_available", "message", ).format( response[0]["tag_name"], response[0]["html_url"], response[0]["body"], ), ) else: logWrite(f"No updates found, bot is up to date.") except bad_request_400.PeerIdInvalid: logWrite( f"Could not send startup message to bot owner. Perhaps user has not started the bot yet." ) except Exception as exp: logWrite(f"Update check failed due to {exp}: {format_exc()}") if configGet("post", "mode"): scheduler.start() try: token = await authorize() if ( len( ( await ( await http_session.get( f'{configGet("address", "posting", "api")}/albums?q={configGet("queue", "posting", "api", "albums")}', headers={"Authorization": f"Bearer {token}"}, ) ).json() )["results"] ) == 0 ): logWrite("Media album does not exist on API server. Trying to create it...") try: await http_session.post( f'{configGet("address", "posting", "api")}/albums?name={configGet("queue", "posting", "api", "albums")}&title={configGet("queue", "posting", "api", "albums")}', headers={"Authorization": f"Bearer {token}"}, ) logWrite( "Created media album on API server."}" ) except Exception as exp: logWrite( f"Could not create media album on API server due to {exp}: {format_exc()}" ) except Exception as exp: logWrite(f"Could not check if API album exists due to {exp}: {format_exc()}") await idle() try: await app.send_message( configGet("owner"), locale("shutdown", "message").format(pid), ) except bad_request_400.PeerIdInvalid: logWrite( f"Could not send shutdown message to bot owner. Perhaps user has not started the bot yet." ) makedirs(configGet("cache", "locations"), exist_ok=True) jsonSave( {"timestamp": time()}, path.join(configGet("cache", "locations"), "shutdown_time"), ) logWrite(locale("shutdown", "console").format(str(pid))) scheduler.shutdown() if __name__ == "__main__": loop = asyncio.get_event_loop() loop.run_until_complete(main())