Python文件复制全攻略

想要高效、安全地复制文件？本文为你提供Python复制文件的全方位指南。从基础的`shutil`模块入手，详解`copy2()`和`copyfile()`的区别与应用，并深入探讨如何处理大型文件，避免内存溢出。文章还涵盖文件权限的处理、异常捕获，以及使用`copytree()`复制目录的技巧。除了`shutil`，还介绍了`os`模块的替代方案，但需注意安全风险。最后，我们还将学习如何通过SHA256哈希值验证文件完整性，确保数据安全无误。掌握这些Python文件复制技巧，让你的数据处理更加得心应手！

使用shutil模块可高效复制文件，shutil.copy2()保留元数据，copyfile()仅复制内容；大文件需分块读取避免内存溢出；通过os.stat和chmod处理权限；结合try-except捕获异常；copytree()复制目录并可设置dirs_exist_ok=True允许目标存在；可用os.system调用系统命令但有安全风险；最后通过SHA256哈希值验证文件完整性。

Python复制文件，核心在于利用各种模块提供的函数，实现文件的读取和写入。简单来说，就是打开源文件，读取内容，然后打开目标文件，写入内容。但实际操作中，我们需要考虑效率、错误处理以及不同的复制需求。

shutil模块是首选，它提供了高级的文件操作功能，包括复制。shutil.copy2()函数不仅复制文件内容，还会保留文件的元数据，比如创建时间、修改时间等。如果仅仅需要复制文件内容，shutil.copyfile()就足够了，效率更高。

import shutil

# 复制文件内容和元数据
shutil.copy2('source_file.txt', 'destination_file.txt')

# 仅复制文件内容
shutil.copyfile('source_file.txt', 'destination_file.txt')

如何高效复制大型文件？

对于大型文件，一次性读取整个文件到内存中显然不可行。更好的方法是使用缓冲区读取，分块处理。

def copy_large_file(src, dst, buffer_size=16384): # 16KB buffer
    try:
        with open(src, 'rb') as f_src, open(dst, 'wb') as f_dst:
            while True:
                chunk = f_src.read(buffer_size)
                if not chunk:
                    break
                f_dst.write(chunk)
    except FileNotFoundError:
        print(f"Error: File not found.")
    except Exception as e:
        print(f"Error during file copy: {e}")

copy_large_file('large_file.dat', 'large_file_copy.dat')

这段代码使用with语句确保文件在使用后会被正确关闭。rb和wb分别表示以二进制读取和写入模式打开文件，这对于处理各种类型的文件都是安全的。buffer_size可以根据实际情况调整，更大的缓冲区可能提高效率，但也占用更多内存。

复制文件时如何处理权限问题？

在Linux或macOS等类Unix系统中，文件权限至关重要。shutil.copy2()在复制文件时会尽可能保留原始文件的权限。但如果用户没有足够的权限，复制操作可能会失败。

可以使用os模块来显式地设置文件权限。

import os
import shutil
import stat

def copy_with_permissions(src, dst):
    shutil.copy2(src, dst)
    st = os.stat(src)
    os.chmod(dst, st.st_mode) # 复制权限

# 示例
copy_with_permissions('important_file.txt', 'backup_file.txt')

这里，os.stat()获取源文件的状态信息，包括权限。然后，os.chmod()将相同的权限应用于目标文件。注意，这需要用户具有修改目标文件权限的权限。

如何处理文件复制过程中可能出现的异常？

文件复制过程中可能出现各种异常，比如文件不存在、权限不足、磁盘空间不足等。良好的错误处理是必不可少的。

import shutil

def safe_copy(src, dst):
    try:
        shutil.copy2(src, dst)
        print(f"File copied successfully from {src} to {dst}")
    except FileNotFoundError:
        print(f"Error: Source file {src} not found.")
    except PermissionError:
        print(f"Error: Permission denied to copy {src} to {dst}.")
    except OSError as e:
        print(f"Error: An OS error occurred: {e}")
    except Exception as e:
        print(f"An unexpected error occurred: {e}")

safe_copy('my_file.txt', 'backup/my_file.txt')

这个函数使用了try...except块来捕获可能发生的异常，并打印相应的错误信息。这有助于诊断问题并采取适当的措施。

如何复制目录及其所有内容？

shutil.copytree()函数可以递归地复制整个目录。

import shutil

try:
    shutil.copytree('source_directory', 'destination_directory')
    print("Directory copied successfully!")
except FileExistsError:
    print("Destination directory already exists!")
except Exception as e:
    print(f"An error occurred: {e}")

注意，如果目标目录已经存在，shutil.copytree()会抛出FileExistsError。可以通过dirs_exist_ok=True参数来解决这个问题，允许目标目录存在。

import shutil

try:
    shutil.copytree('source_directory', 'destination_directory', dirs_exist_ok=True)
    print("Directory copied successfully!")
except Exception as e:
    print(f"An error occurred: {e}")

使用dirs_exist_ok=True时，如果目标目录存在，shutil.copytree()会将源目录的内容合并到目标目录中。如果目标目录中已经存在同名文件，它将被覆盖。

除了shutil，还有其他复制文件的方法吗？

是的，os模块也提供了一些文件操作函数，虽然不如shutil方便，但在某些情况下可能更适用。

import os

def copy_file_os(src, dst):
    try:
        os.system(f'cp "{src}" "{dst}"') # Linux/macOS
        # 或者使用Windows命令
        # os.system(f'copy "{src}" "{dst}"')
        print("File copied using os.system")
    except Exception as e:
        print(f"Error during copy: {e}")

copy_file_os('myfile.txt', 'backup/myfile.txt')

os.system()函数允许执行系统命令，比如cp（Linux/macOS）或copy（Windows）。这种方法的缺点是依赖于操作系统，并且可能存在安全风险，因为它允许执行任意系统命令。

如何验证复制的文件是否完整？

复制后验证文件完整性非常重要，尤其是在处理重要数据时。可以使用哈希算法（比如MD5或SHA256）来比较源文件和目标文件的哈希值。

import hashlib

def calculate_hash(filename):
    hasher = hashlib.sha256()
    with open(filename, 'rb') as file:
        while True:
            chunk = file.read(4096) # 4KB chunk
            if not chunk:
                break
            hasher.update(chunk)
    return hasher.hexdigest()

def verify_copy(src, dst):
    src_hash = calculate_hash(src)
    dst_hash = calculate_hash(dst)

    if src_hash == dst_hash:
        print("File integrity verified: Hashes match.")
    else:
        print("File integrity check failed: Hashes do not match.")

verify_copy('original.dat', 'copied.dat')

这段代码计算源文件和目标文件的SHA256哈希值，并比较它们。如果哈希值相同，则说明文件复制成功。

理论要掌握，实操不能落！以上关于《Python文件复制全攻略》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！