glowz/markdown_solve
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

优化图片文件名处理，修复URL处理逻辑以支持去除引号和空格

Browse Source
This commit is contained in:
glowzz 2025-03-06 14:07:40 +08:00
parent 4aee514ea5
commit 94271a2406
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
dialogue_download_change.py
View File

@ -33,6 +33,8 @@ def choose_input_file():
input_file_entry.insert(0, file_path) input_file_entry.insert(0, file_path)
def clean_filename(url): def clean_filename(url):
#print("origin: "+url)
try: try:
# 首先检查URL是否有常见的图片扩展名 # 首先检查URL是否有常见的图片扩展名
image_extensions = ('.jpg', '.jpeg', '.png', '.gif', '.bmp', '.webp') image_extensions = ('.jpg', '.jpeg', '.png', '.gif', '.bmp', '.webp')
@ -45,6 +47,10 @@ def clean_filename(url):
if fd_id: if fd_id:
return f"{fd_id.group(1)}.jpg" return f"{fd_id.group(1)}.jpg"
# 如果既没有扩展名也没有fdId，使用时间戳 # 如果既没有扩展名也没有fdId，使用时间戳
return f"image_{int(time.time())}.jpg" return f"image_{int(time.time())}.jpg"
except: except:
@ -79,7 +85,10 @@ def download_images():
if not url.startswith(('http://', 'https://')): if not url.startswith(('http://', 'https://')):
if 'www.soujianzhu.cn' not in url: if 'www.soujianzhu.cn' not in url:
url = 'https://www.soujianzhu.cn' + url url = 'https://www.soujianzhu.cn' + url
if '"' in url:
url = url.split(' ')[0]
url=url.strip()
try: try:
response = requests.get(url, headers=headers, timeout=30) response = requests.get(url, headers=headers, timeout=30)
if response.status_code == 200: if response.status_code == 200: