import json
import argparse


def parse_args():
    parser = argparse.ArgumentParser(description='parse json file')

    parser.add_argument('--dir', help='the json file dir')
    args = parser.parse_args()
    return args


def load_json(file_dir):
    with open(file_dir, 'r') as load_f:
        load_dict = json.load(load_f)
    return load_dict


def main():
    args = parse_args()
    file_dir = args.dir
    load_dict = load_json(file_dir)
    for key in load_dict.keys():
        print(key, ', the key in json file')
    print('the key in annotations: ')
    for key in load_dict['annotations'][0].keys():
        print(key)


if __name__ == '__main__':
    main()