如何使用Python从命令行写入文件？

在Python中，我们可以使用open()函数来打开一个文件，然后利用读写操作进行文件的输入输出。本文将介绍如何使用Python从命令行写入文件。

使用Python的I/O函数

Python提供了open()函数来打开文件，可以使用w模式来写入文件：

# 打开文件
file = open("example.txt", "w")
# 写入内容
file.write("Hello World!")
# 关闭文件
file.close()

以上代码打开一个名为example.txt的文件，以写入模式打开，写入内容为“Hello World!”并关闭文件。

我们也可以使用with语句块，它会自动关闭文件，不需要我们手动调用close()函数：

with open("example.txt", "w") as file:
    file.write("Hello World!")

使用Python的命令行参数

在命令行输入时，可以通过sys.argv获取参数列表。sys.argv[0]为脚本名称，sys.argv[1]及以后为脚本参数。以下是一个从命令行输入文件名和写入内容的示例：

import sys

# 获取文件名、写入内容
file_name = sys.argv[1]
content = sys.argv[2]

# 写入文件
with open(file_name, "w") as file:
    file.write(content)

运行脚本时，需在命令行中输入文件名和写入内容：

python write_file.py example.txt "Hello World!"

使用Python的argparse模块

argparse是Python自带的命令行参数解析模块，支持解析复杂的命令行参数。以下是一个使用argparse的示例：

import argparse

# 创建解析器对象
parser = argparse.ArgumentParser(description="Write content to a file")

# 添加命令行参数
parser.add_argument("file_name", type=str, help="file name to write")
parser.add_argument("content", type=str, help="content to write")

# 解析命令行参数
args = parser.parse_args()

# 写入文件
with open(args.file_name, "w") as file:
    file.write(args.content)

以上代码使用argparse模块来解析命令行参数，可以为参数添加说明，并对参数类型进行约束。在命令行中输入：

python write_file.py example.txt "Hello World!"

即可将“Hello World!”写入example.txt文件。

使用Python的click模块

click是一个Python自带的命令行框架，用于快速构建命令行工具。以下是一个使用click模块的示例：

import click

# 定义命令行命令
@click.command()
@click.option("--file_name", prompt="Enter file name", help="file name to write")
@click.option("--content", prompt="Enter content", help="content to write")
def write_file(file_name, content):
    # 写入文件
    with open(file_name, "w") as file:
        file.write(content)

# 执行命令行命令
if __name__ == "__main__":
    write_file()

以上代码使用click模块来定义命令行命令，可以为参数添加提示，并提供交互式命令行界面。在命令行输入：