Merge pull request #160 from YooooEX/telegram-polish

fix(tgapp): polish Telegram MarkdownV2 rendering and startup helpers
This commit is contained in:
LJQ
2026-04-25 21:59:49 +08:00
committed by GitHub

View File

@@ -1,4 +1,4 @@
import os, sys, re, threading, asyncio, queue as Q, socket, time, random import os, sys, re, threading, asyncio, queue as Q, time, random
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
_TEMP_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'temp') _TEMP_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'temp')
from agentmain import GeneraticAgent from agentmain import GeneraticAgent
@@ -16,10 +16,12 @@ from chatapp_common import (
HELP_TEXT, HELP_TEXT,
TELEGRAM_MENU_COMMANDS, TELEGRAM_MENU_COMMANDS,
clean_reply, clean_reply,
ensure_single_instance,
extract_files, extract_files,
format_restore, format_restore,
redirect_log,
require_runtime,
split_text, split_text,
strip_files,
) )
from continue_cmd import handle_frontend_command, reset_conversation from continue_cmd import handle_frontend_command, reset_conversation
from llmcore import mykeys from llmcore import mykeys
@@ -38,6 +40,8 @@ _MD_TOKEN_RE = re.compile(
r"|\[([^\]]+)\]\(([^)\n]+)\)" r"|\[([^\]]+)\]\(([^)\n]+)\)"
r"|`([^`\n]+)`" r"|`([^`\n]+)`"
r"|\*\*([^\n]+?)\*\*" r"|\*\*([^\n]+?)\*\*"
r"|__([^\n]+?)__"
r"|~~([^\n]+?)~~"
r"|(?<!\*)\*(?!\*)([^\n]+?)(?<!\*)\*(?!\*)", r"|(?<!\*)\*(?!\*)([^\n]+?)(?<!\*)\*(?!\*)",
re.DOTALL, re.DOTALL,
) )
@@ -61,6 +65,11 @@ def _resolve_files(paths):
return files return files
def _render_file_markers(text):
def repl(match):
return os.path.basename(match.group(1))
return re.sub(r"\[FILE:([^\]]+)\]", repl, text or "").strip()
def _escape_pre(text): def _escape_pre(text):
return escape_markdown(text or "", version=2, entity_type="pre") return escape_markdown(text or "", version=2, entity_type="pre")
@@ -90,7 +99,11 @@ def _to_markdown_v2(text):
elif match.group(7) is not None: elif match.group(7) is not None:
parts.append(f"*{escape_markdown(match.group(7), version=2)}*") parts.append(f"*{escape_markdown(match.group(7), version=2)}*")
elif match.group(8) is not None: elif match.group(8) is not None:
parts.append(f"_{escape_markdown(match.group(8), version=2)}_") parts.append(f"*{escape_markdown(match.group(8), version=2)}*")
elif match.group(9) is not None:
parts.append(f"~{escape_markdown(match.group(9), version=2)}~")
elif match.group(10) is not None:
parts.append(f"_{escape_markdown(match.group(10), version=2)}_")
pos = match.end() pos = match.end()
parts.append(escape_markdown(text[pos:], version=2)) parts.append(escape_markdown(text[pos:], version=2))
return "".join(parts) return "".join(parts)
@@ -144,7 +157,7 @@ class _TelegramStreamSession:
async def _refresh(self, done, send_files): async def _refresh(self, done, send_files):
cleaned = clean_reply(self.raw_text) if self.raw_text.strip() else "" cleaned = clean_reply(self.raw_text) if self.raw_text.strip() else ""
self.files = _resolve_files(extract_files(cleaned)) self.files = _resolve_files(extract_files(cleaned))
body = strip_files(cleaned) body = _render_file_markers(cleaned)
if done and not body and self.files: if done and not body and self.files:
body = "已生成附件" body = "已生成附件"
elif done and not body: elif done and not body:
@@ -359,17 +372,12 @@ async def handle_command(update, ctx):
return await update.message.reply_text(HELP_TEXT) return await update.message.reply_text(HELP_TEXT)
if __name__ == '__main__': if __name__ == '__main__':
try: # Single instance lock using socket _LOCK_SOCK = ensure_single_instance(19527, "Telegram")
_lock_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM); _lock_sock.bind(('127.0.0.1', 19527))
except OSError:
print('[Telegram] Another instance is already running, skiping...')
sys.exit(1)
if not ALLOWED: if not ALLOWED:
print('[Telegram] ERROR: tg_allowed_users in mykey.py is empty or missing. Set it to avoid unauthorized access.') print('[Telegram] ERROR: tg_allowed_users in mykey.py is empty or missing. Set it to avoid unauthorized access.')
sys.exit(1) sys.exit(1)
_logf = open(os.path.join(os.path.dirname(os.path.dirname(__file__)), 'temp', 'tgapp.log'), 'a', encoding='utf-8', buffering=1) require_runtime(agent, "Telegram", tg_bot_token=mykeys.get("tg_bot_token"))
sys.stdout = sys.stderr = _logf redirect_log(__file__, "tgapp.log", "Telegram", ALLOWED)
print('[NEW] New process starting, the above are history infos ...')
threading.Thread(target=agent.run, daemon=True).start() threading.Thread(target=agent.run, daemon=True).start()
proxy = mykeys.get('proxy') proxy = mykeys.get('proxy')
if proxy: if proxy: