json to csv

json2csv.py

#!/usr/bin/env python

__copyright__ = "2023, pozemka"
__license__   = "WTFPL"

import argparse
import json
import pandas
from pathlib import Path

parser = argparse.ArgumentParser(description='Convert JSON to CSV.')
parser.add_argument('JSON', type=str, help='Input JSON')
parser.add_argument('CSV', type=str, help='Output CSV')
args = parser.parse_args()

in_path = Path(args.json)
out_path = Path(args.CSV)

with in_path.open('r', encoding='utf-8') as f:
    data = json.loads(f.read())
    
top_level_keys = list(data.keys())
    
if len(top_level_keys) == 1 :
    data = data[top_level_keys[0]]
    
dataframe = pandas.json_normalize(data)

dataframe.to_csv(out_path, index=False, encoding='utf-8')

# More info about pandas JSON normalization:
# https://github.com/BindiChen/machine-learning/blob/main/data-analysis/028-pandas-json_normalize/pandas-json_normalize.ipynb