From 85dd93ffe63f174da28b90ebef75f53d8e0151a4 Mon Sep 17 00:00:00 2001 From: glowzz <24627181@qq.com> Date: Fri, 11 Apr 2025 09:56:01 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=9C=AC=E5=9C=B0/=E5=9C=A8?= =?UTF-8?q?=E7=BA=BF=E5=9B=BE=E5=BA=8A=E5=88=87=E6=8D=A2=E5=8A=9F=E8=83=BD?= =?UTF-8?q?=EF=BC=8C=E4=BC=98=E5=8C=96=E4=B8=8A=E4=BC=A0=E5=9B=BE=E7=89=87?= =?UTF-8?q?=E9=80=BB=E8=BE=91=E4=BB=A5=E6=94=AF=E6=8C=81=E9=80=89=E6=8B=A9?= =?UTF-8?q?=E6=9C=8D=E5=8A=A1=E5=99=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dialogue_download_change.py | 36 ++++++++++++++++++++++++++++++------ 1 file changed, 30 insertions(+), 6 deletions(-) diff --git a/dialogue_download_change.py b/dialogue_download_change.py index dd9f438..329815c 100644 --- a/dialogue_download_change.py +++ b/dialogue_download_change.py @@ -11,11 +11,14 @@ import shutil # 配置信息 local_image_folder = 'downloaded_images' -#lsky_pro_url = 'http://192.168.107.248:18089/api/v1' +lsky_pro_url_local = 'http://192.168.107.248:18089/api/v1' lsky_pro_url = 'https://image.lqsjy.cn/api/v1' upload_endpoint = '/upload' token = '1|QJP2YEr9GIN52VBgmm5hCqV5DwBSvJLUKjnwcKB8' +# 添加全局变量控制使用本地/在线图床 +use_local_server = True + # 确保本地图片文件夹存在 #os.makedirs(local_image_folder, exist_ok=True) def get_image_folder(markdown_path): @@ -155,7 +158,12 @@ def upload_image_to_lsky_pro(image_path, retries=3, timeout=30): 'Authorization': f'Bearer {token}', 'Accept': 'application/json' } - response = requests.post(f'{lsky_pro_url}{upload_endpoint}', files=files, headers=headers, timeout=timeout) + # 根据复选框状态选择URL + current_url = lsky_pro_url_local if use_local_server else lsky_pro_url + response = requests.post(f'{current_url}{upload_endpoint}', + files=files, + headers=headers, + timeout=timeout) if response.status_code == 200: try: data = response.json() @@ -193,7 +201,10 @@ def process_image_file(markdown_file_path): # 上传到图床 - new_url = upload_image_to_lsky_pro(url) + if ":" in url: + new_url = upload_image_to_lsky_pro(url) + else: + new_url = upload_image_to_lsky_pro(os.path.dirname(markdown_file_path)+"//"+url) if new_url: url_mapping[url] = new_url print(f"处理成功: {url} -> {new_url}") @@ -379,6 +390,19 @@ def select_files(): root = tk.Tk() root.title("Markdown图片处理工具") +def toggle_server(): + global use_local_server + use_local_server = use_local_var.get() + print(f"使用{'本地' if use_local_server else '在线'}服务器") + +# 添加复选框 +use_local_var = tk.BooleanVar(value=True) # 默认使用本地服务器 +use_local_checkbox = tk.Checkbutton(root, + text="使用本地服务器", + variable=use_local_var, + command=toggle_server) +use_local_checkbox.grid(row=5, column=0, columnspan=3, pady=5) + # 文件路径显示 tk.Label(root, text="Input File:").grid(row=0, column=0, padx=10, pady=10) input_file_entry = tk.Entry(root, width=50) @@ -389,9 +413,9 @@ tk.Button(root, text="Browse", command=choose_input_file).grid(row=0, column=2, download_button = tk.Button(root, text="下载图片", command=download_images) download_button.grid(row=1, column=0, columnspan=3, pady=5) -# 上传图片按钮 -upload_button = tk.Button(root, text="上传图片", command=upload_images) -upload_button.grid(row=2, column=0, columnspan=3, pady=5) +# # 上传图片按钮 +# upload_button = tk.Button(root, text="上传图片", command=upload_images) +# upload_button.grid(row=2, column=0, columnspan=3, pady=5) # 上传图片按钮_batch_files # upload_button = tk.Button(root, text="上传图片_批量", command=upload_images_batch_files)