From c9010de0505cc104c11d86ee307a940b7bdacd00 Mon Sep 17 00:00:00 2001 From: nyaasuki Date: Wed, 21 Feb 2024 13:48:01 +0800 Subject: [PATCH] fix bug --- urls/main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/urls/main.py b/urls/main.py index b7b75e0..e4fd9d9 100644 --- a/urls/main.py +++ b/urls/main.py @@ -11,10 +11,10 @@ def natural_sort_key(s): def generate_urls(file_names, base_url, sub_directory): """根据文件名、基础URL和子目录生成URL链接""" urls = [] - # 确保base_url和sub_directory以斜线结尾 + # 确保base_url和sub_directory以斜线结尾（如果sub_directory不为空） if not base_url.endswith('/'): base_url += '/' - if not sub_directory.endswith('/'): + if sub_directory and not sub_directory.endswith('/'): sub_directory += '/' current_timestamp = int(time.time()) # 获取当前时间的时间戳，转换为整数 for name in sorted(file_names, key=natural_sort_key): # 使用自定义排序 @@ -35,7 +35,7 @@ def save_urls(urls, output_file): def parse_arguments(): """解析命令行参数""" parser = argparse.ArgumentParser(description='Generate URLs from file names.') - parser.add_argument('--dir', type=str, help='Sub-directory for generating file URLs', required=True) + parser.add_argument('--dir', type=str, default='', help='Sub-directory for generating file URLs (optional)') return parser.parse_args() def main():