6.2 KiB
6.2 KiB
123盘AlistURL生成器
简介
123盘AlistURL生成器是一个用于生成文件URL链接的Python脚本。它可以递归地遍历目录及其子目录中的所有文件，根据文件路径、基础URL、子目录和最小文件大小生成URL链接，并将生成的URL链接保存到YAML文件中。生成的YAML文件格式化输出，支持自定义根文件夹名称。
名称
- 程序名称：123盘AlistURL生成器
- 脚本文件名：main.py
功能特点
- 支持递归遍历目录及其子目录中的所有文件
- 可以指定基础URL和子目录，生成的URL链接将包含这些信息
- 可以指定最小文件大小，小于指定大小的文件将被排除
- 生成的URL链接包含文件大小、时间戳和编码后的文件路径
- 支持自然排序，确保文件按照合理的顺序排列
- 输出链接格式化为YAML格式，子文件夹使用缩进表示层级关系
- 支持自定义根文件夹名称
使用方法
-
将脚本文件放置在要生成URL链接的目录中。
-
打开命令行或终端，切换到脚本所在的目录。
运行以下命令：
sh
python main.py [--dir <sub_directory>] [--output <output_file>] [--base-url <base_url>] [--min-size <min_size>] --rf <root_folder>
参数说明
--dir：指定生成文件URL的子目录（可选）。如果不指定，默认为空。--output：指定输出文件名（默认为urls.yaml）。生成的YAML文件将保存到该文件中。--base-url：指定生成文件URL的基础URL（默认为https://link.kite.kim/feng）。生成的URL链接将包含该基础URL。--min-size：指定最小文件大小，以字节为单位（默认为128KB）。小于该大小的文件将被排除。--rf：指定YAML输出的根文件夹名称（必选）。生成的YAML文件将以该名称作为根文件夹。
示例
假设当前目录结构如下：
.
├── dir1
│ ├── file1.txt (200KB)
│ └── file2.txt (50KB)
├── dir2
│ ├── file3.txt (300KB)
│ └── file4.txt (100KB)
└── main.py
运行以下命令：
sh
python main.py --dir "subdir" --min-size 150000 --rf "八尾妖姬"
生成的urls.yaml文件内容将类似于：
八尾妖姬:
dir1:
204800:1624761234:https://link.kite.kim/feng/subdir/dir1/file1.txt
dir2:
307200:1624761234:https://link.kite.kim/feng/subdir/dir2/file3.txt
这个示例中，我们指定了子目录为"subdir"，最小文件大小为150000字节（约150KB），并且指定了根文件夹名称为"八尾妖姬"。生成的URL链接包含了子目录、文件大小、时间戳和编码后的文件路径。小于150KB的文件（file2.txt和file4.txt）被排除在输出结果之外。
代码结构
自然排序函数
pythondef natural_sort_key(s):
"""为了自然排序的辅助函数，将字符串中的数字部分转换成整数"""
return [int(text) if text.isdigit() else text.lower() for text in re.split('([0-9]+)', s)]
生成URL函数
pythondef generate_urls(file_paths, base_url, sub_directory, min_size):
"""根据文件路径、基础URL、子目录和最小文件大小生成URL链接"""
urls = {}
if not base_url.endswith('/'):
base_url += '/'
if sub_directory and not sub_directory.endswith('/'):
sub_directory += '/'
current_timestamp = int(time.time()) # 移到循环外
for path in sorted(file_paths, key=natural_sort_key):
file_size_bytes = os.path.getsize(path)
if file_size_bytes < min_size:
continue
relative_path = os.path.relpath(path, start='.')
encoded_path = urllib.parse.quote(relative_path)
url = f"{base_url}{sub_directory}{encoded_path}"
dir_name = os.path.dirname(relative_path)
if dir_name not in urls:
urls[dir_name] = []
urls[dir_name].append(f"{file_size_bytes}:{current_timestamp}:{url}")
return urls
保存URL函数
pythondef save_urls(urls, output_file, root_folder):
"""将URL链接保存到YAML文件中"""
with open(output_file, 'w', encoding='utf-8') as f:
f.write(f"{root_folder}:\n")
for dir_name, files in urls.items():
if dir_name == '.':
for file in files:
f.write(f" {file}\n")
else:
f.write(f" {dir_name}:\n")
for file in files:
f.write(f" {file}\n")
解析命令行参数
pythondef parse_arguments():
"""解析命令行参数"""
parser = argparse.ArgumentParser(description='Generate URLs from file names.')
parser.add_argument('--dir', type=str, default='', help='Sub-directory for generating file URLs (optional)')
parser.add_argument('--output', type=str, default='urls.yaml', help='Output file name (default: urls.yaml)')
parser.add_argument('--base-url', type=str, default='https://link.kite.kim/feng', help='Base URL for generating file URLs (default: https://link.kite.kim/feng)')
parser.add_argument('--min-size', type=int, default=128*1024, help='Minimum file size in bytes (default: 128KB)')
parser.add_argument('--rf', type=str, required=True, help='Root folder name for YAML output')
return parser.parse_args()
递归列出文件
pythondef list_files_recursive(start_path='.', exclude_files=None):
"""递归列出目录及其子目录中的所有文件，排除指定的文件"""
if exclude_files is None:
exclude_files = set()
file_paths = []
for root, dirs, files in os.walk(start_path):
for file in files:
if file not in exclude_files:
file_paths.append(os.path.join(root, file))
return file_paths
主函数
pythondef main():
args = parse_arguments()
current_script = os.path.basename(__file__)
exclude_files = {current_script} # 排除当前脚本文件
file_paths = list_files_recursive('.', exclude_files)
urls = generate_urls(file_paths, args.base_url, args.dir, args.min_size)
save_urls(urls, args.output, args.rf)
print(f"URL链接已保存到{args.output}")
if __name__ == '__main__':
main()