From 8f0250df43e906f32c949f199de0cab62cc26724 Mon Sep 17 00:00:00 2001 From: Jerry Yan <792602257@qq.com> Date: Fri, 8 Aug 2025 13:56:39 +0800 Subject: [PATCH] =?UTF-8?q?=E9=80=9A=E8=BF=87argv=E4=BC=A0skip=5Fif=5Fexis?= =?UTF-8?q?t=E9=BB=98=E8=AE=A4=E5=80=BC?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- util/oss.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/util/oss.py b/util/oss.py index 265e3ed..d3832c6 100644 --- a/util/oss.py +++ b/util/oss.py @@ -1,5 +1,6 @@ import logging import os +import sys import requests from opentelemetry.trace import Status, StatusCode @@ -70,7 +71,7 @@ def upload_to_oss(url, file_path): return False -def download_from_oss(url, file_path, skip_if_exist=False): +def download_from_oss(url, file_path, skip_if_exist=None): """ 使用签名URL下载文件到OSS :param skip_if_exist: 如果存在就不下载了 @@ -82,6 +83,11 @@ def download_from_oss(url, file_path, skip_if_exist=False): with tracer.start_as_current_span("download_from_oss") as span: span.set_attribute("file.url", url) span.set_attribute("file.path", file_path) + + # 如果skip_if_exist为None,则从启动参数中读取 + if skip_if_exist is None: + skip_if_exist = 'skip_if_exist' in sys.argv + if skip_if_exist and os.path.exists(file_path): span.set_attribute("file.exist", True) span.set_attribute("file.size", os.path.getsize(file_path))