使用 argparse 打开 json 文件？

Question

我最近刚开始使用 argparse，我正在尝试使用它来打开一个 json 文件以及其他一些东西，比如输出到一个 csv 文件中。

import argparse
import os
import json
import csv
import pandas as pd

parser = argparse.ArgumentParser(description='Json to Csv')
parser.add_argument('file', type=argparse.FileType('r'))
#parser.add_argument('file', type=str)
args = parser.parse_args()

with args.file as json_data:

print(json_data)
#argparse_dict = vars(args)
#argparse_dict.update(json_data)


baseline_dict = {}
try:
    baseline_dict = json.load(json_data)
except:
    print("JSON Baseline File {}: Unable to LOAD")

results_dict = baseline_dict["results"]

"""with open(args.file) as json_data:
baseline_dict = {}
try:
    baseline_dict = json.load(json_data)
except:
    print("JSON Baseline File {}: Unable to LOAD")"""

#将新字典变成数据框 df = pd.DataFrame(results_dict)

当我尝试在终端中使用 argparse 打开 json 文件时，出现此错误。我知道它正在提取 json 文件，因为我可以在将其调用为 json_data 之前让它运行。

with args.file as json_data:
JSON Baseline File {}: Unable to LOAD
Traceback (most recent call last):

Answer 1

因为您使用argparse.FileType 作为参数type，所以args.file已经是一个打开的文件句柄，就好像它是由open 返回的一样。如果您想确保它确定性地关闭（并给它一个不合格的名称），您只需要：

with args.file as json_data:

或者只是：

with args.file:

如果不需要使用本地名称的别名。

Answer 2

可以通过两种方式加载文件的脚本：

import argparse, json

parser = argparse.ArgumentParser()
parser.add_argument('--file', type=argparse.FileType('r'))
parser.add_argument('--name')
args = parser.parse_args()
print(args)

if args.file is not None:
    print(json.load(args.file))
    args.file.close()
    
if args.name is not None:
    with open(args.name) as f:
         print(json.load(f))
         

样本运行：

1307:~/mypy$ python3 stack62557562.py 
Namespace(file=None, name=None)
1307:~/mypy$ python3 stack62557562.py --file foo.json
Namespace(file=<_io.TextIOWrapper name='foo.json' mode='r' encoding='UTF-8'>, name=None)
{'foo': 12, 'bar': 'twelve'}
1307:~/mypy$ python3 stack62557562.py --name foo.json
Namespace(file=None, name='foo.json')
{'foo': 12, 'bar': 'twelve'}
1307:~/mypy$ python3 stack62557562.py --name foo.json --file foo.json
Namespace(file=<_io.TextIOWrapper name='foo.json' mode='r' encoding='UTF-8'>, name='foo.json')
{'foo': 12, 'bar': 'twelve'}
{'foo': 12, 'bar': 'twelve'}