# Copyright (c) 2013, Web Notes Technologies Pvt. Ltd. and Contributors
# MIT License. See license.txt 

from __future__ import unicode_literals
import frappe
from frappe import _

@frappe.whitelist()
def runserverobj():
	"""
		Run server objects
	"""
	from frappe.utils import cint

	wrapper = None
	method = frappe.form_dict.get('method')
	arg = frappe.form_dict.get('args', frappe.form_dict.get("arg"))
	dt = frappe.form_dict.get('doctype')
	dn = frappe.form_dict.get('docname')
	
	if dt: # not called from a doctype (from a page)
		if not dn: dn = dt # single
		doc = frappe.get_doc(dt, dn)

	else:
		doc = frappe.get_doc(json.loads(frappe.form_dict.get('docs')))
		doc.check_if_latest()

	if not doc.has_permission("read"):
		frappe.msgprint(_("No Permission"), raise_exception = True)
	
	if doc:
		r = doc.run_method(method, arg)
		if r:
			#build output as csv
			if cint(frappe.form_dict.get('as_csv')):
				make_csv_output(r, doc.doctype)
			else:
				frappe.response['message'] = r
		
		frappe.response.docs.append(doc)

def make_csv_output(res, dt):
	"""send method response as downloadable CSV file"""
	import frappe
	
	from cStringIO import StringIO
	import csv
		
	f = StringIO()
	writer = csv.writer(f)
	for r in res:
		row = []
		for v in r:
			if isinstance(v, basestring):
				v = v.encode("utf-8")
			row.append(v)
		writer.writerow(row)
	
	f.seek(0)
						
	frappe.response['result'] = unicode(f.read(), 'utf-8')
	frappe.response['type'] = 'csv'
	frappe.response['doctype'] = dt.replace(' ','')