
文件管理是我们日常工作中常见的任务之一,Python作为一种功能强大的编程语言,可以帮助我们轻松完成这些任务。今天我们来学习6个实用的Python脚本模板,让自动化文件管理工作变得简单又高效。
1. 获取指定目录下的所有文件首先,我们学习如何获取指定目录下的所有文件。这通常是文件管理的第一步。
import osdef list_files_in_directory(directory): # 使用os.listdir列出指定目录中的所有文件和子目录 files_and_directories = os.listdir(directory) # 筛选出文件(不是子目录) only_files = [file for file in files_anddirectories if os.path.isfile(os.path.join(directory, file))] return only_files# 示例使用my_directory = 'C:/Users/username/Documents'files = list_files_in_directory(my_directory)print("以下是文件列表:")for f in files: print(f)# 输出:将显示指定目录的所有文件名称。这个函数利用了os模块来处理与文件相关的操作。
2. 文件按扩展名分类当我们需要对大量文件进行整理时,按扩展名分类是一个非常好的方法。
def sort_files_by_extension(directory): from collections import defaultdict files_dict = defaultdict(list) # 创建默认字典用于存储不同扩展名的文件 for file_name in os.listdir(directory): if os.path.isfile(os.path.join(directory, file_name)): # 检查是否是文件 _, ext = os.path.splitext(file_name) # 分割文件名和扩展名 if ext: # 如果有扩展名,则加入到相应的键中 files_dict[ext].append(file_name) return dict(files_dict)# 示例使用my_directory = "C:/temp"files_sorted = sort_files_by_extension(my_directory)# 打印分类后的结果for extension, files_list in files_sorted.items(): print(f"文件类型 {extension}:") for fname in files_list: print(fname)这里用到了collections.defaultdict以便于快速生成一个按照文件扩展名归类的字典。
3. 移动文件到特定的文件夹当你已经按照某种规则分类了文件,下一步就是把它们移动到相应的位置上。
import shutildef move_files_based_on_extension(source_dir, destination_base_dir): sorted_files = sort_files_by_extension(source_dir) for ext, filenames in sorted_files.items(): destination_folder = os.path.join(destination_base_dir, ext[1:]) # 忽略点号 ifnot os.path.exists(destination_folder): # 如果不存在则创建 os.makedirs(destination_folder) for filename in filenames: full_path = os.path.join(source_dir, filename) new_path = os.path.join(destination_folder, filename) shutil.move(full_path, new_path) # 移动文件 print(f"已将文件{filename}移动到{new_path}")# 示例使用move_files_based_on_extension('C:/Source', 'C:/Destination')我们用到shutil.move()函数来执行文件移动,并且保证了不会丢失原文件位置信息。
4. 计算单个或多个文件的总大小知道所管理和移动的文件总体量,对于优化磁盘空间非常有用。
def get_total_size_of_files(directory, filenames=None): total_size = 0 if filenames isNone: # 如果没有指定文件列表,则读取全部文件 filenames = os.listdir(directory) for filename in filenames: filepath = os.path.join(directory, filename) if os.path.isfile(filepath): # 只考虑实际存在的文件 total_size += os.path.getsize(filepath) return total_size# 示例size = get_total_size_of_files('C:/SourceFiles')print(f"该目录下文件总大小为: {size/(1024**2)} MB") # 转换为MB这里的代码展示了如何通过遍历和os.path.getsize()函数得到所有文件的总大小。
5. 删除超过一定天数的旧文件有时需要清理长期未被访问的旧文件。
from datetime import datetime, timedeltadef remove_old_files(path_to_scan, days_limit=30): critical_time = datetime.now() - timedelta(days=days_limit) for root, dirs, files in os.walk(path_to_scan): for file in files: file_full_path = os.path.join(root, file) mod_time = datetime.fromtimestamp(os.path.getmtime(file_full_path)) if mod_time < critical_time: # 超过限制日期,删除 print(f"删除文件:{file_full_path}") os.remove(file_full_path)# 使用示例remove_old_files('C:/OldData', 365) # 删除一年以上的文件通过比较文件修改时间与设定的天数限制值,可以确定哪些文件应该被删除。
6. 压缩文件或整个目录为zip格式最终的一个脚本例子是如何将某些重要文件压缩成.zip形式保存。
import zipfiledef compress_directory(dir_to_zip, output_filename='archive'): # 创建压缩包实例 zipf = zipfile.ZipFile(output_filename + '.zip', 'w', zipfile.ZIP_DEFLATED) # 遍历并添加所有文件到zip包内 for root, dirs, files in os.walk(dir_to_zip): for file in files: zipf.write(os.path.join(root, file), os.path.relpath(os.path.join(root, file), os.path.join(dir_to_zip, '..'))) zipf.close()# 使用compress_directory('C:/BackupThis', 'backup_archive')上述代码将指定目录下的所有内容都添加到名为“archive.zip”的压缩文件里。
实战案例:整理家庭相册设想一下你需要整理一个包含数千张照片的大相册:
步骤一: 获取目标目录里的全部图片。步骤二: 根据年份(通过图片元数据提取或者以文件创建时间为标准)分组。步骤三: 新建相应的文件夹,并按组移入相应的位置。步骤四: 压缩每一年的照片为单独的存档压缩文件备份至另一存储装置上。通过组合以上模板可以有效实现上述流程。
本文介绍了从基本列出文件到高级文件处理操作如分类、移动、大小计算、删除、以及压缩等多个Python脚本模版;并且给出了实际应用场景下综合运用多个脚本的情况解析。