Script to convert standard comma-separated CSV to TabSON format

csv2tabson.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

import csv
import sys
import re
import json

args = sys.argv
ZEROLENGTH_AS_NULL = False

if len(args) == 1:
    print "Usage: python csv2tabson.py srcfile dstfile"
    exit()

elif len(args) == 2:    
    srcfile = args[1]
    dstfile = None
    
else:
    srcfile = args[1]
    dstfile = args[2]
    
is_float = re.compile('^[-+]?[0-9]*\.[0-9]+$')
is_int = re.compile('^[-+]?\d+$')

def fix_data_type(cell):
    if len(cell) == 0 and ZEROLENGTH_AS_NULL is True:
        return None
    elif is_float.match(cell):
        return float(cell)
    elif is_int.match(cell):
        return int(cell)
    else:
        return cell
            

jsondata = {'header':[],'data':[]}

with open(srcfile, 'rb') as csvfile:
    reader = csv.reader(csvfile, delimiter=',', quotechar='"')
    is_first_row = True
    for row in reader:
        if is_first_row is True:
            jsondata['header'] = row
            is_first_row = False
        else:
            jsondata['data'].append(map(fix_data_type,row))

if dstfile is not None:
    with open(dstfile,'w') as dist:
        json.dump(jsondata,dist,separators=(',', ':'))
        dist.close()
else:
    print json.dumps(jsondata,separators= (',', ':'))