From 3728fb939aafcc61d4727ff400a9199a25ab1f67 Mon Sep 17 00:00:00 2001 From: Zhang Yanpo Date: Sun, 22 Feb 2026 20:08:20 +0800 Subject: [PATCH] refactor: replace MIME reverse-lookup loop with dict Add mime_to_suffix reverse dict in mime.py and use it in render_to_img instead of iterating over mimetypes to find a key by value. Co-Authored-By: Claude Opus 4.6 --- down2.py | 9 ++------- mime.py | 2 ++ 2 files changed, 4 insertions(+), 7 deletions(-) diff --git a/down2.py b/down2.py index d0cbef5..bd3516c 100755 --- a/down2.py +++ b/down2.py @@ -20,7 +20,7 @@ from pylatexenc.latex2text import LatexNodes2Text import k3proc -from .mime import mimetypes +from .mime import mime_to_suffix, mimetypes from .syntax_highlight import code_to_html @@ -354,12 +354,7 @@ def render_to_img( input = ''.format(base_uri) + input m = mimetypes.get(mime) or mime - - suffix = mime - for k, v in mimetypes.items(): - if v == m: - suffix = k - break + suffix = mime_to_suffix.get(m, mime) with tempfile.TemporaryDirectory() as tdir: fn = os.path.join(tdir, "xxx." + suffix) diff --git a/mime.py b/mime.py index ca94e28..b09e86f 100644 --- a/mime.py +++ b/mime.py @@ -968,3 +968,5 @@ "zirz": "application/vnd.zul", "zmm": "application/vnd.handheld-entertainment+xml", } + +mime_to_suffix = {v: k for k, v in mimetypes.items()}