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

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

我正在使用 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 命令行中的名称编写文件？

解决方法

您的代码，添加了 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])