如何在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在该文件中写入输出。但这不起作用，我也不太理解错误消息。如何使用argparse在命令行中使用名称使用csv库编写文件？

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