文件管理是我们日常工作中常见的任务之一，Python作为一种功能强大的编程语言，可以帮助我们轻松完成这些任务。今天我们来学习6个实用的Python脚本模板，让自动化文件管理工作变得简单又高效。
1. 获取指定目录下的所有文件
首先，我们学习如何获取指定目录下的所有文件。这通常是文件管理的第一步。
import os
def 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 shutil
def 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:]) # 忽略点号
if not 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 is None: # 如果没有指定文件列表，则读取全部文件
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, timedelta
def 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 zipfile
def 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脚本模版；并且给出了实际应用场景下综合运用多个脚本的情况解析。
