seitime-frappe/py/webnotes/widgets/form/run_method.py
2012-02-23 12:35:45 +05:30

86 lines
No EOL
2.7 KiB
Python

# Copyright (c) 2012 Web Notes Technologies Pvt Ltd (http://erpnext.com)
#
# MIT License (MIT)
#
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the "Software"),
# to deal in the Software without restriction, including without limitation
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
# and/or sell copies of the Software, and to permit persons to whom the
# Software is furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
# INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
# PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
# CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE
# OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#
import webnotes
@webnotes.whitelist()
def runserverobj():
"""
Run server objects
"""
import webnotes.model.code
from webnotes.model.doclist import DocList
from webnotes.utils import cint
form = webnotes.form
doclist = None
method = form.getvalue('method')
arg = form.getvalue('arg')
dt = form.getvalue('doctype')
dn = form.getvalue('docname')
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:
doclist = DocList()
doclist.from_compressed(form.getvalue('docs'), dn)
so = doclist.make_obj()
doclist.check_if_latest()
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.getvalue('as_csv')):
make_csv_output(r, so.doc.doctype)
else:
webnotes.response['message'] = r
webnotes.response['docs'] =[so.doc] + 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:
writer.writerow(r)
f.seek(0)
webnotes.response['result'] = f.read()
webnotes.response['type'] = 'csv'
webnotes.response['doctype'] = dt.replace(' ','')