以下是一个Python脚本,用于获取目录/data/ftp/shwfy/和/data/shwfy-web/data/中所有后缀名为.mp3的文件的绝对路径,随机选取其中的1000条,将这些文件拷贝到一个临时目录中,并将它们压缩成一个zip文件,存储在/root/feixi/1000_mp3.zip中:
import os
import random
import shutil
import zipfile
def find_mp3_files(directory):
"""遍历目录,查找所有 .mp3 文件的绝对路径"""
mp3_files = []
for root, dirs, files in os.walk(directory):
for file_name in files:
if file_name.lower().endswith('.mp3'):
mp3_files.append(os.path.join(root, file_name))
return mp3_files
def copy_files_to_temp(files, temp_dir):
"""将指定文件拷贝到临时目录"""
if not os.path.exists(temp_dir):
os.makedirs(temp_dir)
for file in files:
shutil.copy(file, temp_dir)
def zip_files(src_dir, zip_file_path):
"""将指定目录下的文件压缩成zip文件"""
with zipfile.ZipFile(zip_file_path, 'w') as zipf:
for root, dirs, files in os.walk(src_dir):
for file in files:
file_path = os.path.join(root, file)
zipf.write(file_path, os.path.relpath(file_path, src_dir))
def main():
directories = ['/data/ftp/XXX/', '/data/XXXX/data/']
temp_dir = '/root/feixi/temp_mp3'
zip_file_path = '/root/feixi/1000_mp3.zip'
all_mp3_files = []
for directory in directories:
all_mp3_files.extend(find_mp3_files(directory))
# 随机打乱文件列表
random.shuffle(all_mp3_files)
# 只获取前1000条
selected_mp3_files = all_mp3_files[:1000]
# 将文件拷贝到临时目录
copy_files_to_temp(selected_mp3_files, temp_dir)
# 压缩临时目录中的文件
zip_files(temp_dir, zip_file_path)
# 删除临时目录
shutil.rmtree(temp_dir)
if __name__ == '__main__':
main()
这个脚本包含以下步骤:
- 获取所有.mp3文件的绝对路径。
- 随机打乱这些路径,并选取前1000条。
- 将选中的文件拷贝到一个临时目录中。
- 将临时目录中的文件压缩成一个zip文件。
- 删除临时目录。
请确保您有足够的权限来创建目录、拷贝文件和生成压缩文件,并且目标路径存在。如果路径不存在,请手动创建目标路径或在脚本中添加相应的代码以创建这些路径。

被折叠的 条评论
为什么被折叠?



