76 lines
No EOL
2 KiB
Python
76 lines
No EOL
2 KiB
Python
# Copyright (c) 2013, Web Notes Technologies Pvt. Ltd.
|
|
# MIT License. See license.txt
|
|
|
|
from __future__ import unicode_literals
|
|
import webnotes
|
|
from webnotes import _
|
|
|
|
@webnotes.whitelist()
|
|
def runserverobj():
|
|
"""
|
|
Run server objects
|
|
"""
|
|
import webnotes.model.code
|
|
from webnotes.model.bean import Bean
|
|
from webnotes.utils import cint
|
|
|
|
wrapper = None
|
|
method = webnotes.form_dict.get('method')
|
|
arg = webnotes.form_dict.get('args', webnotes.form_dict.get("arg"))
|
|
dt = webnotes.form_dict.get('doctype')
|
|
dn = webnotes.form_dict.get('docname')
|
|
|
|
webnotes.response["docs"] = []
|
|
|
|
if dt: # not called from a doctype (from a page)
|
|
if not dn: dn = dt # single
|
|
so = webnotes.model.code.get_obj(dt, dn)
|
|
|
|
else:
|
|
bean = Bean()
|
|
bean.from_compressed(webnotes.form_dict.get('docs'), dn)
|
|
if not bean.has_read_perm():
|
|
webnotes.msgprint(_("No Permission"), raise_exception = True)
|
|
so = bean.make_controller()
|
|
bean.check_if_latest(method="runserverobj")
|
|
|
|
check_guest_access(so.doc)
|
|
|
|
if so:
|
|
r = webnotes.model.code.run_server_obj(so, method, arg)
|
|
if r:
|
|
#build output as csv
|
|
if cint(webnotes.form_dict.get('as_csv')):
|
|
make_csv_output(r, so.doc.doctype)
|
|
else:
|
|
webnotes.response['message'] = r
|
|
|
|
webnotes.response['docs'] += so.doclist
|
|
|
|
def check_guest_access(doc):
|
|
if webnotes.session['user']=='Guest' and not webnotes.conn.sql("select name from tabDocPerm where role='Guest' and parent=%s and ifnull(`read`,0)=1", doc.doctype):
|
|
webnotes.msgprint("Guest not allowed to call this object")
|
|
raise Exception
|
|
|
|
def make_csv_output(res, dt):
|
|
"""send method response as downloadable CSV file"""
|
|
import webnotes
|
|
|
|
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)
|
|
|
|
webnotes.response['result'] = unicode(f.read(), 'utf-8')
|
|
webnotes.response['type'] = 'csv'
|
|
webnotes.response['doctype'] = dt.replace(' ','') |