在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文档以获取更多信息。