From c6726aa0ef820443ff82239fd87918581e4a9237 Mon Sep 17 00:00:00 2001
From: "E. S." <the@rootless.one>
Date: Tue, 9 Jul 2024 19:21:30 +0300
Subject: [PATCH] initial

---
 .gitignore    |  2 ++
 tgschedpub.py | 88 +++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 90 insertions(+)
 create mode 100644 .gitignore
 create mode 100755 tgschedpub.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..9cd5661
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,2 @@
+/.idea
+__pycache__
\ No newline at end of file
diff --git a/tgschedpub.py b/tgschedpub.py
new file mode 100755
index 0000000..a134744
--- /dev/null
+++ b/tgschedpub.py
@@ -0,0 +1,88 @@
+#!/usr/bin/env python3
+import logging
+import yaml
+import os
+import re
+
+from asyncio import sleep
+from argparse import ArgumentParser
+from telethon import TelegramClient
+from telethon.extensions.html import (
+    unparse as entities_to_html,
+    parse as html_to_entities
+)
+from telethon.tl.patched import Message
+from telethon.tl.types import InputPeerChannel, InputMediaWebPage
+from telethon.tl.functions.messages import EditMessageRequest, SendScheduledMessagesRequest
+
+# Set up logging
+logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
+
+CONFIG_PATH = os.path.expanduser('~/.config/tgschedpub/config.yaml')
+DOT_IMAGE_PATTERN = r'\.\<\/a\>$'
+SESSION_FILE = os.path.join(os.path.dirname(CONFIG_PATH), 'tgschedpub.session')
+client: TelegramClient
+
+
+def load_config():
+    with open(CONFIG_PATH, 'r') as file:
+        return yaml.safe_load(file)
+
+
+async def main(mode, channel_id=None):
+    messages = reversed(await client.get_messages(channel_id, limit=100, scheduled=True))
+    channel = await client.get_entity(channel_id)
+    input_peer = InputPeerChannel(channel.id, channel.access_hash)
+
+    for m in messages:
+        if not isinstance(m, Message):
+            continue
+
+        if mode == 'send':
+            await client(SendScheduledMessagesRequest(
+                peer=input_peer,
+                id=[m.id]
+            ))
+            break
+
+        elif mode == 'images':
+            html = entities_to_html(m.message, m.entities)
+            if re.search(DOT_IMAGE_PATTERN, html):
+                html = re.sub(DOT_IMAGE_PATTERN, '&#8204;</a>', html)
+                text, entities = html_to_entities(html)
+                await client(EditMessageRequest(
+                    peer=input_peer,
+                    id=m.id,
+                    media=InputMediaWebPage(
+                        url=m.media.webpage.url,
+                        force_large_media=False,
+                        force_small_media=False
+                    ),
+                    reply_markup=m.reply_markup,
+                    schedule_date=m.date,
+                    invert_media=m.invert_media,
+                    message=text,
+                    entities=entities
+                ))
+                logging.info(f'Message {m.id} seems to be edited')
+
+                await sleep(2)
+            else:
+                logging.error(f'Message {m.id}: could not find link with a dot!')
+
+
+if __name__ == '__main__':
+    parser = ArgumentParser()
+    parser.add_argument('--channel-id', type=str)
+    parser.add_argument('mode', choices=['auth', 'images', 'send'])
+    args = parser.parse_args()
+
+    config = load_config()
+    client = TelegramClient(SESSION_FILE, config['api_id'], config['api_hash'])
+
+    if args.mode == 'auth':
+        client.start(phone=config['phone'], password=config.get('password'))
+        logging.info('Authenticated successfully.')
+    else:
+        client.start()
+        client.loop.run_until_complete(main(args.mode, args.channel_id))