文件管理是我们日常工作中常见的任务之一，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:])  # 忽略点号
        
        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, 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脚本模版；并且给出了实际应用场景下综合运用多个脚本的情况解析。