shutil和fileinput模块：文件操作的最佳实践

在Python中，shutil和fileinput模块是处理文件和输入/输出(I/O)操作的有力工具。shutil模块提供了一种在Python中操作文件的高级接口，而fileinput模块则允许我们轻松地读取多个输入流。

shutil模块

shutil模块是Python的标准库之一，提供了很多用于处理文件和文件夹的函数。下面是一些常用的shutil函数：

shutil.copy(src, dst): 复制源文件(src)到目标路径(dst)。
shutil.move(src, dst): 将源文件(src)移动到目标路径(dst)。
shutil.copytree(src, dst): 复制源文件夹(src)到目标路径(dst)。
shutil.rmtree(path): 删除包含空文件夹的路径。
shutil.chown(path, user=None, group=None): 改变路径(path)的所有者。
shutil.get_terminal_size(): 返回终端窗口的大小。
shutil.disk_usage(path): 返回指定路径的磁盘使用情况。
shutil.make_archive(base_name, format, root_dir=None, verbose=0, dry_run=0): 创建归档文件。
shutil.register_rmtree(dir): 在rmtree函数中注册文件夹，以便在删除时通知rmtree函数。

以下是shutil模块的使用示例：

复制代码

pythonimport shutil
import os

# 创建文件夹
os.makedirs('test_folder')

# 复制文件
shutil.copy('source.txt', 'test_folder/destination.txt')

# 移动文件
shutil.move('source.txt', 'test_folder/source.txt')

# 删除文件夹，包括其中的所有文件和子文件夹
shutil.rmtree('test_folder')
import shutil
import os

# 创建文件夹
os.makedirs('test_folder')

# 复制文件
shutil.copy('source.txt', 'test_folder/destination.txt')

# 移动文件
shutil.move('source.txt', 'test_folder/source.txt')

# 删除文件夹，包括其中的所有文件和子文件夹
shutil.rmtree('test_folder')

fileinput模块

fileinput模块允许用户在Python程序中轻松地迭代多个输入流，包括标准输入、文件和目录。fileinput模块的主要功能是处理多个输入流，这对于处理命令行参数、文件名和标准输入非常有用。它支持在每个文件上执行操作，并在操作完成后自动关闭文件。以下是一些常用的fileinput函数：

fileinput.input(files=None, inplace=False, backup='', bufsize=-1, mode='b', openhook=None): 从文件中读取行。
fileinput.FileInput(files=None, inplace=False, backup='', bufsize=-1, mode='b', openhook=None): 从文件中读取行，并保持打开的文件句柄。
fileinput.hook_compressed(method=None, mode='b'): 为open函数添加压缩支持。
fileinput.hook_encoded(encoding=None): 为open函数添加编码支持。
fileinput.close(): 关闭所有打开的文件句柄。
fileinput.isfirstline(): 如果当前行是第一个行，则返回True。
fileinput.isstdin(): 如果当前行来自标准输入，则返回True。
fileinput.nextfile(): 关闭当前文件并前进到下一个文件。如果已经处于最后一个文件，则返回None。
fileinput.filename(): 返回当前文件的名称。如果已经处于最后一个文件，则返回None。
fileinput.lineno(): 返回当前行的行号。如果已经处于最后一个文件，则返回None。
fileinput.filelineno(): 返回当前文件的行号。如果已经处于最后一个文件，则返回None。
fileinput.fileno(): 返回当前文件的文件描述符。如果已经处于最后一个文件，则返回None。
fileinput.readline(): 读取并返回当前文件的下一行，不包括结尾的换行符（如果有的话）。如果已经处于最后一个文件，则返回None。
fileinput.readlines(): 读取并返回当前文件的剩余所有行，每行作为一个字符串列表。如果已经处于最后一个文件，则返回None。
fileinput.next():next()：读取并返回当前文件的下一行，不包括结尾的换行符（如果有的话）。如果已经处于最后一个文件，则引发StopIteration异常。

以下是使用fileinput模块的示例代码：

复制代码

pythonimport fileinput

# 从文件中读取行并输出到控制台
for line in fileinput.input('test.txt'):
    print(line)

# 在文件上执行操作并输出结果
with fileinput.FileInput('test.txt') as f:
    for line in f:
        # 对每一行进行处理
        processed_line = line.strip().upper()
        print(processed_line)
import fileinput

# 从文件中读取行并输出到控制台
for line in fileinput.input('test.txt'):
    print(line)

# 在文件上执行操作并输出结果
with fileinput.FileInput('test.txt') as f:
    for line in f:
        # 对每一行进行处理
        processed_line = line.strip().upper()
        print(processed_line)

在上面的示例中，我们使用fileinput.input()函数从文件中读取行，并将其打印到控制台。我们还使用fileinput.FileInput()函数创建一个文件对象，并在其上迭代每一行。对于每一行，我们执行一些操作（例如去除首尾的空白字符并转换为大写字母），然后将其打印到控制台。

除了上述示例外，fileinput模块还提供了许多其他功能，例如支持从标准输入读取行、支持文件备份、支持指定缓冲区大小等等。您可以查看Python文档以获取更多信息。