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

Question

我正在使用 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?

编写文件

Answer 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])

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

How to write a file using the argparse and csv libraries in Python?

python

csv

argparse