如何使用 Python 中的 argparse 和 csv 库编写文件？

Posted 2023-03-25

【中文标题】如何使用 Python 中的 argparse 和 csv 库编写文件？

【英文标题】：How to write a file using the argparse and csv libraries in Python?

【发布时间】：2021-04-14 14:28:32

【问题描述】：

我正在使用 Python 创建报告生成器，并尝试从命令行读取和写入文件名。例如，我希望能够生成输出文件（带有我在命令行中键入的名称），例如：

python generator.py -p product.csv -s sales.csv --p product_report.csv

这里，generator.py 是 Python 脚本，我正在读取两个文件 product.csv 和 sales.csv。在脚本中，我计算收入，我想将结果输出到product_report.csv 文件中。我当前的代码如下所示：

import csv
import argparse

parser = argparse.ArgumentParser()

parser.add_argument("-p", type=argparse.FileType("r"))
parser.add_argument("-s", type=argparse.FileType("r"))
parser.add_argument("--p", type=argparse.FileType('w', encoding='UTF-8'))
args = parser.parse_args()
args_write_report = parser.parse_args(["--p"])

with args.p as Product, args.s as Sales, args_write_report as ProductReport:

    calculated_revenue= 'apple': '1000', 'banana': '22224'

    writer_product = csv.writer(ProductReport)
    writer_product.writerow(['Product', 'GrossRevenue'])
    
    for key, value in calculated_revenue.items():
        writer_product.writerow([key, value])

然后我运行命令行，我得到这个错误：

generator.py: error: argument --p: expected one argument

我尝试搜索 argparse 写入函数，但因为我使用的是 csv 库，我想我可以使用 argparse 函数（用于输出文件名）创建一个空文件，并且继续使用csv 将输出写入该文件。但这不起作用，我也不太明白错误消息。 如何使用csv 库在命令行中使用argparse 编写文件？

【问题讨论】：

—p 不是可选参数，而是一个位置参数（因为有两个 -s），所以你不应该在命令行中输入—p。跨度>

如果您在编写 csv 文件时遇到问题，这是一个单独的问题，但阅读文档可能会有所帮助，因为它提供了有用的示例 docs.python.org/3/library/csv.html?highlight=csv

@barny 实际上我在编写 CSV 文件时没有问题。在我刚刚使用 with open as ProductReport 之前它工作正常。但是现在我希望用户能够通过使用argparse 输入命令行来命名文件，这就是我遇到问题的地方。

与FileType 斗争的另一种方法是只使用argparse 来接受文件名——用于输入和输出。 FileType 生成的 opened 文件在 with 上下文中无法正常工作。

parser.parse_args(["--p", "filename"]) 非常适合测试，但对于生产用途，您需要从命令行获取标志和名称。你知道怎么做，不是吗？否则使用argparse 没有什么意义。

【参考方案1】：

您的代码加上 print(args) 行会产生：

1338:~/mypy$ touch product.csv
1339:~/mypy$ touch sales.csv
1339:~/mypy$ python3 stack65636491.py -p product.csv -s sales.csv --p product_report.csv
Namespace(p=<_io.TextIOWrapper name='product_report.csv' mode='w' encoding='UTF-8'>, s=<_io.TextIOWrapper name='sales.csv' mode='r' encoding='UTF-8'>)
usage: stack65636491.py [-h] [-p P] [-s S] [--p P]
stack65636491.py: error: argument --p: expected one argument

注意args.p 是一个以写入模式打开的文件。这来自 '--p' 论点。 args.s' 是一个打开的读取文件。 '--p' 参数覆盖了 '-p' 参数（它们具有相同的dest）

错误来自多余的parser.parse_args(["--p"]) 行。

更改这些行：

parser.add_argument("-r", type=argparse.FileType('w', encoding='UTF-8'))
args = parser.parse_args()
print(args)
#args_write_report = parser.parse_args(["--p"])

with args.p as Product, args.s as Sales, args.r as ProductReport:

和测试：

1344:~/mypy$ python3 stack65636491.py -p product.csv -s sales.csv -r product_report.csv
Namespace(p=<_io.TextIOWrapper name='product.csv' mode='r' encoding='UTF-8'>, r=<_io.TextIOWrapper name='product_report.csv' mode='w' encoding='UTF-8'>, s=<_io.TextIOWrapper name='sales.csv' mode='r' encoding='UTF-8'>)
1345:~/mypy$ cat product_report.csv 
Product,GrossRevenue
apple,1000
banana,22224

现在它打开3个文件而不互相踩踏，然后继续写入新的。

===

没有FileType的更简单的版本：

parser = argparse.ArgumentParser()
parser.add_argument("-r")
args = parser.parse_args()
print(args)

with open(args.r, 'w') as ProductReport:
    calculated_revenue= 'apple': '1000', 'banana': '22224'

    writer_product = csv.writer(ProductReport)
    writer_product.writerow(['Product', 'GrossRevenue'])
    
    for key, value in calculated_revenue.items():
        writer_product.writerow([key, value])