python 如何复制一个文件夹

使用Python复制一个文件夹的方法有很多，包括使用shutil模块、os模块、以及第三方库如 distutils等。最常用和简单的方法是使用shutil模块。 其中，shutil.copytree()函数是最常用的，可以递归地复制整个文件夹及其包含的所有文件和子文件夹。下面详细介绍如何使用shutil模块复制一个文件夹。

一、shutil模块

shutil模块是Python标准库中的一部分，提供了许多高级文件操作功能，包括复制文件和文件夹。我们主要使用shutil.copytree()函数来复制文件夹。

1. 使用shutil.copytree()复制文件夹

shutil.copytree()函数可以递归地复制整个文件夹，包括其所有内容。它的基本用法如下：

import shutil
source = 'path/to/source/folder'
destination = 'path/to/destination/folder'
shutil.copytree(source, destination)

在这个示例中，source是你想要复制的源文件夹的路径，destination是目标文件夹的路径。如果目标文件夹不存在，shutil.copytree()会自动创建它。

2. 处理已存在的目标文件夹

如果目标文件夹已经存在，shutil.copytree()会抛出一个FileExistsError异常。为了避免这个问题，你可以在复制之前先删除目标文件夹，或者使用一个不同的目标路径。

import shutil
import os
source = 'path/to/source/folder'
destination = 'path/to/destination/folder'
if os.path.exists(destination):
    shutil.rmtree(destination)
shutil.copytree(source, destination)

在这个示例中，我们在复制之前检查目标文件夹是否存在，如果存在则删除它。

3. 复制特定文件类型

有时候你可能只想复制特定类型的文件，而不是整个文件夹。在这种情况下，你可以使用shutil.copy2()函数和os.walk()函数来实现。

import shutil
import os
source = 'path/to/source/folder'
destination = 'path/to/destination/folder'
file_types = ('.txt', '.jpg')
if not os.path.exists(destination):
    os.makedirs(destination)
for root, dirs, files in os.walk(source):
    for file in files:
        if file.endswith(file_types):
            source_file = os.path.join(root, file)
            destination_file = os.path.join(destination, os.path.relpath(root, source), file)
            destination_dir = os.path.dirname(destination_file)
            if not os.path.exists(destination_dir):
                os.makedirs(destination_dir)
            shutil.copy2(source_file, destination_file)

在这个示例中，我们使用os.walk()遍历源文件夹中的所有文件，检查每个文件的扩展名是否在我们指定的file_types元组中。如果是，我们使用shutil.copy2()函数将它复制到目标文件夹。

二、os模块

尽管shutil模块是复制文件夹的首选方法，有时候你可能想使用os模块来实现更多的控制和定制。os模块提供了低级别的文件操作功能，可以用来手动复制文件和文件夹。

1. 使用os模块复制文件夹

你可以使用os.makedirs()函数创建目标文件夹，并使用shutil.copy2()函数复制每个文件。

import os
import shutil
source = 'path/to/source/folder'
destination = 'path/to/destination/folder'
for root, dirs, files in os.walk(source):
    for dir in dirs:
        os.makedirs(os.path.join(destination, os.path.relpath(os.path.join(root, dir), source)), exist_ok=True)
    for file in files:
        shutil.copy2(os.path.join(root, file), os.path.join(destination, os.path.relpath(root, source), file))

在这个示例中，我们使用os.walk()遍历源文件夹中的所有文件和子文件夹，并手动创建目标文件夹的结构。

三、使用第三方库

虽然shutil模块和os模块已经足够强大，有时候你可能需要使用第三方库来实现更高级的功能。distutils模块是一个这样的库，尽管它在复制文件夹方面并不常用，但你仍然可以使用它来实现复制文件夹的功能。

1. 使用distutils.dir_util.copy_tree()复制文件夹

distutils.dir_util.copy_tree()函数可以递归地复制整个文件夹，包括其所有内容。它的基本用法如下：

from distutils.dir_util import copy_tree
source = 'path/to/source/folder'
destination = 'path/to/destination/folder'
copy_tree(source, destination)

在这个示例中，source是你想要复制的源文件夹的路径，destination是目标文件夹的路径。如果目标文件夹不存在，copy_tree()会自动创建它。

四、处理文件权限

在复制文件夹时，有时候你可能会遇到文件权限问题。例如，如果源文件夹中的某些文件是只读的，shutil.copytree()函数可能会抛出一个PermissionError异常。为了避免这个问题，你可以在复制之前手动修改文件权限，或者在复制之后恢复原始权限。

1. 修改文件权限

你可以使用os.chmod()函数修改文件权限，以确保在复制过程中不会遇到权限问题。

import os
import shutil
import stat
source = 'path/to/source/folder'
destination = 'path/to/destination/folder'
def make_writable(path):
    os.chmod(path, stat.S_IWUSR)
for root, dirs, files in os.walk(source):
    for file in files:
        make_writable(os.path.join(root, file))
shutil.copytree(source, destination)

在这个示例中，我们定义了一个make_writable()函数，用来将文件权限修改为可写。然后，我们在复制之前遍历源文件夹中的所有文件，并将它们的权限修改为可写。

2. 恢复文件权限

如果你需要在复制之后恢复原始权限，你可以在复制之前记录文件的权限，并在复制之后恢复它们。

import os
import shutil
import stat
source = 'path/to/source/folder'
destination = 'path/to/destination/folder'
def get_permissions(path):
    return stat.S_IMODE(os.lstat(path).st_mode)
def set_permissions(path, permissions):
    os.chmod(path, permissions)
permissions = {}
for root, dirs, files in os.walk(source):
    for file in files:
        path = os.path.join(root, file)
        permissions[path] = get_permissions(path)
        os.chmod(path, stat.S_IWUSR)
shutil.copytree(source, destination)
for root, dirs, files in os.walk(destination):
    for file in files:
        path = os.path.join(root, file)
        original_path = os.path.join(source, os.path.relpath(path, destination))
        if original_path in permissions:
            set_permissions(path, permissions[original_path])

在这个示例中，我们定义了get_permissions()和set_permissions()函数，用来获取和设置文件权限。在复制之前，我们记录源文件夹中所有文件的权限，并在复制之后恢复它们。

五、总结

在Python中复制文件夹的方法有很多，最常用和简单的方法是使用shutil模块的shutil.copytree()函数。 通过本文的介绍，你学会了如何使用shutil模块、os模块以及第三方库来复制文件夹，并处理文件权限问题。无论是简单的文件夹复制，还是复杂的文件类型筛选和权限处理，这些方法都能满足你的需求。