63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
import os
|
||
import urllib.parse
|
||
import argparse
|
||
import re
|
||
import time
|
||
|
||
def natural_sort_key(s):
|
||
"""为了自然排序的辅助函数，将字符串中的数字部分转换成整数"""
|
||
return [int(text) if text.isdigit() else text.lower() for text in re.split('([0-9]+)', s)]
|
||
|
||
def generate_urls(file_names, base_url, sub_directory):
|
||
"""根据文件名、基础URL和子目录生成URL链接"""
|
||
urls = []
|
||
# 确保base_url和sub_directory以斜线结尾（如果sub_directory不为空）
|
||
if not base_url.endswith('/'):
|
||
base_url += '/'
|
||
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): # 使用自定义排序
|
||
file_path = os.path.join('.', name)
|
||
file_size_bytes = os.path.getsize(file_path) # 获取文件大小（字节），为整数
|
||
# 对文件名进行URL编码
|
||
encoded_name = urllib.parse.quote(name)
|
||
url = f"{base_url}{sub_directory}{encoded_name}"
|
||
urls.append(f"{file_size_bytes}:{current_timestamp}:{url}")
|
||
return urls
|
||
|
||
def save_urls(urls, output_file):
|
||
"""将URL链接保存到文本文件中"""
|
||
with open(output_file, 'w', encoding='utf-8') as f:
|
||
for url in urls:
|
||
f.write(url + '\n')
|
||
|
||
def 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)')
|
||
return parser.parse_args()
|
||
|
||
def main():
|
||
# 解析命令行参数
|
||
args = parse_arguments()
|
||
|
||
# 固定的base_url值
|
||
base_url = 'https://link.kite.kim/feng'
|
||
|
||
# 获取当前目录下的所有文件名，排除此脚本文件和隐藏文件
|
||
current_script = os.path.basename(__file__)
|
||
file_names = [f for f in os.listdir('.') if os.path.isfile(f) and f != current_script and not f.startswith('.')]
|
||
|
||
# 生成URL链接
|
||
urls = generate_urls(file_names, base_url, args.dir)
|
||
|
||
# 保存URL链接到文本文件
|
||
output_file = 'urls.txt'
|
||
save_urls(urls, output_file)
|
||
|
||
print(f"URL链接已保存到{output_file}")
|
||
|
||
if __name__ == '__main__':
|
||
main()
|