diff --git a/scripts/ksunextbot.py b/scripts/ksunextbot.py index 0cf3499b..7b7245e7 100644 --- a/scripts/ksunextbot.py +++ b/scripts/ksunextbot.py @@ -9,6 +9,7 @@ API_ID = os.environ.get("API_ID") API_HASH = os.environ.get("API_HASH") BOT_TOKEN = os.environ.get("BOT_TOKEN") CHAT_ID = os.environ.get("CHAT_ID") +MESSAGE_THREAD_ID = os.environ.get("MESSAGE_THREAD_ID") COMMIT_URL = os.environ.get("COMMIT_URL") COMMIT_MESSAGE = os.environ.get("COMMIT_MESSAGE") RUN_URL = os.environ.get("RUN_URL") @@ -39,13 +40,15 @@ def get_caption(): def check_environ(): - global CHAT_ID + global CHAT_ID, MESSAGE_THREAD_ID if BOT_TOKEN is None: print("[-] Invalid BOT_TOKEN") exit(1) if CHAT_ID is None: print("[-] Invalid CHAT_ID") exit(1) + else: + CHAT_ID = int(CHAT_ID) if COMMIT_URL is None: print("[-] Invalid COMMIT_URL") exit(1) @@ -61,10 +64,15 @@ def check_environ(): if VERSION is None: print("[-] Invalid VERSION") exit(1) + if MESSAGE_THREAD_ID is None: + print("[-] Invaild MESSAGE_THREAD_ID") + exit(1) + else: + MESSAGE_THREAD_ID = int(MESSAGE_THREAD_ID) async def main(): - print("[+] Uploading to Telegram channel") + print("[+] Uploading to telegram") check_environ() files = sys.argv[1:] print("[+] Files:", files) @@ -86,6 +94,7 @@ async def main(): entity=CHAT_ID, file=files, caption=caption, + reply_to=MESSAGE_THREAD_ID, parse_mode="markdown" ) print("[+] Done!")