json2csv.py

import sublime, sublime_plugin

import simplejson as json
import csv
import cStringIO
import os
from types import *

class Json2csvCommand(sublime_plugin.TextCommand):

	# Get the content of the json file or selection
	def get_json_content(self):
		regions = self.view.sel()
		if len(regions) > 0 and not regions[0].empty():
			selection = regions[0]
		else:
			selection = sublime.Region(0, self.view.size())

		return self.view.substr(selection)

	# Write data to csv
	def write_csv(self, csv_buffer):
		write_header = True
		item_keys    = []
		writer       = csv.writer(csv_buffer)

		for item in self.data:
			for key in item:
				if not key in item_keys:
					item_keys.append(key)
		writer.writerow(item_keys)
		for item in self.data:
			item_values = []
			for key in item_keys:
				value = item.get(key, '')
				if type(value) is StringTypes:
					item_values.append(value.encode('utf-8'))
				else:
					item_values.append(value)
			writer.writerow(item_values)


	def run(self, edit):
		self.data            = []
		self.source_filename = self.view.file_name()
		#print self.source_filename

		json_content = self.get_json_content()

		# Validate json data
		try:
			#print "json data is valid"
			self.data = json.loads(json_content)
			
			if self.source_filename:
				dest_filename = os.path.splitext(self.source_filename)[0] + '.csv'
				with open(dest_filename, 'wb') as csv_buffer:
					self.write_csv(csv_buffer)

				sublime.active_window().open_file(dest_filename)
			else:
				csv_buffer = cStringIO.StringIO()
				self.write_csv(csv_buffer)

				window = self.view.window()
				csv_file_view = window.new_file()
				selection = sublime.Region(0, csv_file_view.size())
				#print selection

				edit = csv_file_view.begin_edit('json2csv')
				csv_file_view.replace(edit, selection, csv_buffer.getvalue())
				csv_file_view.end_edit(edit)				

		except Exception:
			import sys
			exc = sys.exc_info()[1]
			sublime.status_message(str(exc))
			#print str(exc)