From 4aee514ea5246c7a2c89310a19291adc362eb72e Mon Sep 17 00:00:00 2001 From: glowzz <24627181@qq.com> Date: Mon, 3 Mar 2025 17:01:07 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E5=9B=BE=E7=89=87=E6=96=87?= =?UTF-8?q?=E4=BB=B6=E5=90=8D=E5=A4=84=E7=90=86=E9=80=BB=E8=BE=91=EF=BC=8C?= =?UTF-8?q?=E6=94=AF=E6=8C=81=E7=9B=B4=E6=8E=A5=E4=BD=BF=E7=94=A8URL?= =?UTF-8?q?=E6=89=A9=E5=B1=95=E5=90=8D=EF=BC=8C=E5=B9=B6=E6=9B=B4=E6=96=B0?= =?UTF-8?q?Markdown=E6=96=87=E4=BB=B6=E4=BF=9D=E5=AD=98=E8=B7=AF=E5=BE=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dialogue_download_change.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/dialogue_download_change.py b/dialogue_download_change.py index 2ac5281..a08f3e6 100644 --- a/dialogue_download_change.py +++ b/dialogue_download_change.py @@ -34,12 +34,18 @@ def choose_input_file(): def clean_filename(url): try: - # 提取fdId参数 + # 首先检查URL是否有常见的图片扩展名 + image_extensions = ('.jpg', '.jpeg', '.png', '.gif', '.bmp', '.webp') + if url.lower().endswith(image_extensions): + # 如果有扩展名，直接使用URL的最后部分作为文件名 + return os.path.basename(url) + + # 如果没有常见扩展名，则检查fdId fd_id = re.search(r'fdId=([0-9a-f]+)', url) if fd_id: return f"{fd_id.group(1)}.jpg" - # 如果没有fdId，使用时间戳 + # 如果既没有扩展名也没有fdId，使用时间戳 return f"image_{int(time.time())}.jpg" except: return f"image_{int(time.time())}.jpg" @@ -100,9 +106,11 @@ def download_images(): new_content = markdown_content for old_url, new_path in url_mapping.items(): new_content = new_content.replace(old_url, new_path) - + # 构造新的文件名 + file_name, file_ext = os.path.splitext(markdown_file_path) + new_file_path = f"{file_name}_local{file_ext}" # 保存更新后的Markdown文件 - with open(markdown_file_path, 'w', encoding='utf-8') as file: + with open(new_file_path, 'w', encoding='utf-8') as file: file.write(new_content) messagebox.showinfo("完成", f"成功下载 {success_count} 张图片并更新Markdown文件")