A simple JSON to CSV file converter

json_to_csv.py

import os
import sys
import csv
import codecs
import json
reload(sys)
sys.setdefaultencoding('utf-8')

'''
Determine the different keys present in JSON
'''
def get_keys(data):
    keys = set()
    for k in data:
        for obj in data[k]:
            keys.add(obj)
    return keys

'''
Convert JSON to CSV:
1. Identify the keys in JSON
2. Pretty print CSV file
'''
def do_conversion(sep):
    print('I am in do_conversion')
    file_name = 'test.json'
    with codecs.open(file_name, 'r', encoding='utf-8') as data_file:
        data = json.load(data_file)
        keys = get_keys(data)
        print 'jsonKey' + sep + sep.join(keys)
        for k in data:
            csvLine = []
            csvLine.append(k)
            for key in keys:
                if(key in data[k]):
                    csvLine.append(data[k][key])
                else:
                    csvLine.append('')
            csvString = sep.join(csvLine)
            print csvString + os.linesep

do_conversion(';')