def save_query(request, design_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
form = QueryForm()
api = get_api(request.user)
app_names = api.jars()
try:
form.bind(request.POST)
form.query.fields['appName'].choices = ((key, key)
for key in app_names)
if form.is_valid():
design = save_design(request, form, query_type, design, True)
response['design_id'] = design.id
response['status'] = 0
else:
response['message'] = smart_str(form.query.errors) + smart_str(
form.saveform.errors)
except RuntimeError, e:
response['message'] = str(e)
def execute(request, design_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_server = get_query_server_config(app_name)
query_type = beeswax.models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
try:
query_form = get_query_form(request)
if query_form.is_valid():
query_str = query_form.query.cleaned_data["query"]
explain = request.GET.get('explain', 'false').lower() == 'true'
design = save_design(request, query_form, query_type, design,
False)
if query_form.query.cleaned_data['is_parameterized']:
# Parameterized query
parameterization_form_cls = make_parameterization_form(
query_str)
if parameterization_form_cls:
parameterization_form = parameterization_form_cls(
request.REQUEST, prefix="parameterization")
if parameterization_form.is_valid():
real_query = substitute_variables(
query_str, parameterization_form.cleaned_data)
query = HQLdesign(query_form, query_type=query_type)
query._data_dict['query']['query'] = real_query
try:
if explain:
return explain_directly(
request, query, design, query_server)
else:
return execute_directly(
request, query, design, query_server)
except Exception, ex:
db = dbms.get(request.user, query_server)
error_message, log = expand_exception(ex, db)
response['message'] = error_message
return HttpResponse(json.dumps(response),
mimetype="application/json")
else:
response['errors'] = parameterization_form.errors
return HttpResponse(json.dumps(response),
mimetype="application/json")
# Non-parameterized query
query = HQLdesign(query_form, query_type=query_type)
if request.GET.get('explain', 'false').lower() == 'true':
return explain_directly(request, query, design, query_server)
else:
return execute_directly(request, query, design, query_server)
else:
def save_query_design(request, design_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_type = beeswax.models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
try:
query_form = get_query_form(request)
if query_form.is_valid():
design = save_design(request, query_form, query_type, design, True)
response['design_id'] = design.id
response['status'] = 0
else:
response['errors'] = {
'query': [query_form.query.errors],
'settings': query_form.settings.errors,
'file_resources': query_form.file_resources.errors,
'functions': query_form.functions.errors,
'saveform': query_form.saveform.errors,
}
except RuntimeError, e:
response['message'] = str(e)
def save_query(request, design_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
form = QueryForm()
api = get_api(request.user)
app_names = api.jars()
try:
form.bind(request.POST)
form.query.fields['appName'].choices = ((key, key) for key in app_names)
if form.is_valid():
design = save_design(request, form, query_type, design, True)
response['design_id'] = design.id
response['status'] = 0
else:
response['message'] = smart_str(form.query.errors) + smart_str(form.saveform.errors)
except RuntimeError, e:
response['message'] = str(e)
def save_query(request, design_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
try:
save_form = SaveForm(request.POST.copy())
query_form = get_query_form(request, design_id)
if query_form.is_valid() and save_form.is_valid():
design = save_design(request, save_form, query_form, query_type,
design, True)
response['design_id'] = design.id
response['status'] = 0
else:
response['errors'] = query_form.errors
except RuntimeError as e:
response['message'] = str(e)
return JsonResponse(response)
def save_query_design(request, design_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_type = beeswax.models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
try:
query_form = get_query_form(request)
if query_form.is_valid():
design = save_design(request, query_form, query_type, design, True)
response['design_id'] = design.id
response['status'] = 0
else:
response['errors'] = {
'query': [query_form.query.errors],
'settings': query_form.settings.errors,
'file_resources': query_form.file_resources.errors,
'functions': query_form.functions.errors,
'saveform': query_form.saveform.errors,
}
except RuntimeError as e:
response['message'] = str(e)
return JsonResponse(response)
def execute_query(request, design_id=None, table=None):
authorized_get_design(request, design_id)
error_message, log = None, None
form = query_form()
action = request.path
design = safe_get_design(request, SavedQuery.HQL, design_id)
on_success_url = request.REQUEST.get('on_success_url')
for _ in range(1):
if request.method == 'POST':
form.bind(request.POST)
to_explain = request.POST.has_key('button-explain')
to_submit = request.POST.has_key('button-submit')
# Always validate the saveform, which will tell us whether it needs explicit saving
if not form.is_valid():
break
to_save = form.saveform.cleaned_data['save']
to_saveas = form.saveform.cleaned_data['saveas']
if to_saveas and not design.is_auto:
# Save As only affects a previously saved query
design = design.clone()
if to_submit or to_save or to_saveas or to_explain:
explicit_save = to_save or to_saveas
design = save_design(request, form, SavedQuery.HQL, design, explicit_save)
action = urlresolvers.reverse(execute_query, kwargs=dict(design_id=design.id))
# We're not going to process the form. Simply re-render it.
if not to_explain and not to_submit:
break
query_str = _strip_trailing_semicolon(form.query.cleaned_data["query"])
query_server = db_utils.get_query_server(form.query_servers.cleaned_data["server"])
# (Optional) Parameterization.
parameterization = get_parameterization(request, query_str, form, design, to_explain)
if parameterization:
return parameterization
query_msg = make_beeswax_query(request, query_str, form)
try:
if to_explain:
return expl_d(request, query_str, query_msg, design, query_server)
else:
notify = form.query.cleaned_data.get('email_notify', False)
return e_d(request, query_msg, design=design,
on_success_url=on_success_url,
notify=notify)
except BeeswaxException, ex:
error_message, log = expand_exception(ex)
# Fall through to re-render the execute form.
else:
# GET request
if design.id is not None:
data = HQLdesign.loads(design.data).get_query_dict()
form.bind(data)
form.saveform.set_data(design.name, design.desc)
else:
# New design
form.bind()
def execute(request, design_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_server = get_query_server_config(app_name)
query_type = beeswax.models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
try:
query_form = get_query_form(request)
if query_form.is_valid():
query_str = query_form.query.cleaned_data["query"]
explain = request.GET.get('explain', 'false').lower() == 'true'
design = save_design(request, query_form, query_type, design, False)
if query_form.query.cleaned_data['is_parameterized']:
# Parameterized query
parameterization_form_cls = make_parameterization_form(query_str)
if parameterization_form_cls:
parameterization_form = parameterization_form_cls(request.REQUEST, prefix="parameterization")
if parameterization_form.is_valid():
parameters = parameterization_form.cleaned_data
real_query = substitute_variables(query_str, parameters)
query = HQLdesign(query_form, query_type=query_type)
query._data_dict['query']['query'] = real_query
try:
if explain:
return explain_directly(request, query, design, query_server)
else:
return execute_directly(request, query, design, query_server, parameters=parameters)
except Exception, ex:
db = dbms.get(request.user, query_server)
error_message, log = expand_exception(ex, db)
response['message'] = error_message
return JsonResponse(response)
else:
response['errors'] = parameterization_form.errors
return JsonResponse(response)
# Non-parameterized query
query = HQLdesign(query_form, query_type=query_type)
if request.GET.get('explain', 'false').lower() == 'true':
return explain_directly(request, query, design, query_server)
else:
return execute_directly(request, query, design, query_server)
else:
def fetch_saved_query(request, design_id):
response = {"status": -1, "message": ""}
if request.method != "GET":
response["message"] = _("A GET request is required.")
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
response["design"] = design_to_dict(design)
return HttpResponse(json.dumps(response), mimetype="application/json")
def fetch_saved_design(request, design_id):
response = {'status': 0, 'message': ''}
if request.method != 'GET':
response['message'] = _('A GET request is required.')
app_name = get_app_name(request)
query_type = beeswax.models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
response['design'] = design_to_dict(design)
return JsonResponse(response)
def fetch_saved_design(request, design_id):
response = {'status': 0, 'message': ''}
if request.method != 'GET':
response['message'] = _('A GET request is required.')
app_name = get_app_name(request)
query_type = beeswax.models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
response['design'] = design_to_dict(design)
return JsonResponse(response)
def fetch_saved_query(request, design_id):
response = {'status': -1, 'message': ''}
if request.method != 'GET':
response['message'] = _('A GET request is required.')
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
response['design'] = design_to_dict(design)
return HttpResponse(json.dumps(response), mimetype="application/json")
def fetch_saved_query(request, design_id):
response = {'status': -1, 'message': ''}
if request.method != 'GET':
response['message'] = _('A GET request is required.')
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
response['design'] = design_to_dict(design)
return HttpResponse(json.dumps(response), mimetype="application/json")
def execute(request, design_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
try:
form = get_query_form(request)
if form.is_valid():
#design = save_design(request, SaveForm(), form, query_type, design)
# query = SQLdesign(form, query_type=query_type)
# query_server = dbms.get_query_server_config(request.POST.get('server'))
# db = dbms.get(request.user, query_server)
# query_history = db.execute_query(query, design)
# query_history.last_state = beeswax_models.QueryHistory.STATE.expired.index
# query_history.save()
params = '\n'.join([
'%(name)s=%(value)s' % param
for param in json.loads(form.cleaned_data['params'])
])
try:
api = get_api(request.user)
results = api.submit_job(
form.cleaned_data['appName'],
form.cleaned_data['classPath'],
data=params,
context=None if form.cleaned_data['autoContext'] else
form.cleaned_data['context'],
sync=False)
if results['status'] == 'STARTED':
response['status'] = 0
response['results'] = results
else:
response['message'] = str(results[1]['result'])
response['design'] = design.id
except Exception, e:
response['message'] = str(e)
else:
def execute_query(request, design_id=None, query_history_id=None):
"""
View function for executing an arbitrary synchronously query.
"""
action = request.path
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
return render('execute.mako', request, {
'action': action,
'design': design,
'autocomplete_base_url': reverse('rdbms:api_autocomplete_databases', kwargs={}),
'can_edit_name': design.id and not design.is_auto,
})
def execute_query(request, design_id=None, query_history_id=None):
"""
View function for executing an arbitrary synchronously query.
"""
action = request.path
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
return render('execute.mako', request, {
'action': action,
'doc_id': json.dumps(design.id and design.doc.get().id or -1),
'design': design,
'autocomplete_base_url': reverse('rdbms:api_autocomplete_databases', kwargs={}),
'can_edit_name': design.id and not design.is_auto,
})
def execute(request, design_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
try:
form = get_query_form(request)
if form.is_valid():
#design = save_design(request, SaveForm(), form, query_type, design)
# query = SQLdesign(form, query_type=query_type)
# query_server = dbms.get_query_server_config(request.POST.get('server'))
# db = dbms.get(request.user, query_server)
# query_history = db.execute_query(query, design)
# query_history.last_state = beeswax_models.QueryHistory.STATE.expired.index
# query_history.save()
params = '\n'.join(['%(name)s=%(value)s' % param for param in json.loads(form.cleaned_data['params'])])
try:
api = get_api(request.user)
results = api.submit_job(
form.cleaned_data['appName'],
form.cleaned_data['classPath'],
data=params,
context=None if form.cleaned_data['autoContext'] else form.cleaned_data['context'],
sync=False
)
if results['status'] == 'STARTED':
response['status'] = 0
response['results'] = results
else:
response['message'] = str(results[1]['result'])
response['design'] = design.id
except Exception, e:
response['message'] = str(e)
else:
def editor(request, design_id=None):
if design_id is not None and not design_id.isdigit():
job_id, design_id = design_id, None
else:
job_id = None
action = request.path
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
return render('editor.mako', request, {
'action': action,
'design': design,
'design_json': json.dumps(design_to_dict(design)),
'can_edit_name': design.id and not design.is_auto,
'job_id': job_id,
})
def execute_query(request, design_id=None):
"""
View function for executing an arbitrary synchronously query.
"""
action = request.path
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
return render(
"execute.mako",
request,
{
"action": action,
"design": design,
"autocomplete_base_url": reverse("rdbms:api_autocomplete_databases", kwargs={}),
"can_edit_name": design.id and not design.is_auto,
},
)
def execute_query(request, design_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
try:
form = get_query_form(request, design_id)
if form.is_valid():
design = save_design(request, SaveForm(), form, query_type, design)
query = SQLdesign(form, query_type=query_type)
query_server = dbms.get_query_server_config(
request.POST.get('server'))
db = dbms.get(request.user, query_server)
query_history = db.execute_query(query, design)
query_history.last_state = beeswax_models.QueryHistory.STATE.expired.value
query_history.save()
try:
db.use(form.cleaned_data['database'])
datatable = db.execute_and_wait(query)
results = db.client.create_result(datatable)
response['status'] = 0
response['results'] = results_to_dict(results)
response['design'] = design.id
except Exception as e:
response['status'] = -1
response['message'] = str(e)
else:
response['message'] = _('There was an error with your query.')
response['errors'] = form.errors
except RuntimeError as e:
response['message'] = str(e)
return JsonResponse(response, encoder=ResultEncoder)
def save_query(request, query_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_type = beeswax.models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, query_id)
try:
query_form = get_query_form(request)
if query_form.is_valid():
design = save_design(request, query_form, query_type, design, True)
response['design_id'] = design.id
response['status'] = 0
else:
response['errors'] = query_form.errors
except RuntimeError, e:
response['message'] = str(e)
def editor(request, design_id=None, query_history_id=None):
api = get_api(request.user)
jobs = api.jobs()
if design_id is not None and not design_id.isdigit():
job_id, design_id = design_id, None
else:
job_id = None
action = request.path
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
return render(
'editor.mako', request, {
'action': action,
'design': design,
'design_json': json.dumps(design_to_dict(design)),
'can_edit_name': design.id and not design.is_auto,
'job_id': job_id,
})
def save_query(request, design_id=None):
response = {"status": -1, "message": ""}
if request.method != "POST":
response["message"] = _("A POST request is required.")
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
try:
save_form = SaveForm(request.POST.copy())
query_form = get_query_form(request, design_id)
if query_form.is_valid() and save_form.is_valid():
design = save_design(request, save_form, query_form, query_type, design, True)
response["design_id"] = design.id
response["status"] = 0
else:
response["errors"] = query_form.errors
except RuntimeError, e:
response["message"] = str(e)
def execute_query(request, design_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
try:
form = get_query_form(request, design_id)
if form.is_valid():
design = save_design(request, SaveForm(), form, query_type, design)
query = SQLdesign(form, query_type=query_type)
query_server = dbms.get_query_server_config(request.POST.get('server'))
db = dbms.get(request.user, query_server)
query_history = db.execute_query(query, design)
query_history.last_state = beeswax_models.QueryHistory.STATE.expired.index
query_history.save()
try:
db.use(form.cleaned_data['database'])
datatable = db.execute_and_wait(query)
results = db.client.create_result(datatable)
response['status'] = 0
response['results'] = results_to_dict(results)
response['design'] = design.id
except Exception, e:
response['status'] = -1
response['message'] = str(e)
else:
def editor(request, design_id=None, query_history_id=None):
api = get_api(request.user)
jobs = api.jobs()
if design_id is not None and not design_id.isdigit():
job_id, design_id = design_id, None
else:
job_id = None
action = request.path
app_name = get_app_name(request)
query_type = beeswax_models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
return render(
"editor.mako",
request,
{
"action": action,
"design": design,
"design_json": json.dumps(design_to_dict(design)),
"can_edit_name": design.id and not design.is_auto,
"job_id": job_id,
},
)
def execute_query(request, design_id=None):
"""
View function for executing an arbitrary query.
It understands the optional GET/POST params:
on_success_url
If given, it will be displayed when the query is successfully finished.
Otherwise, it will display the view query results page by default.
"""
authorized_get_design(request, design_id)
request.session['start_time'] = time.time(
) # FIXME: add job id to not intersect simultaneous jobs
error_message = None
form = QueryForm()
action = request.path
log = None
app_name = get_app_name(request)
query_type = SavedQuery.TYPES_MAPPING['beeswax']
design = safe_get_design(request, query_type, design_id)
on_success_url = request.REQUEST.get('on_success_url')
query_server = get_query_server_config(app_name)
db = dbms.get(request.user, query_server)
databases = _get_db_choices(request)
if request.method == 'POST':
form.bind(request.POST)
form.query.fields[
'database'].choices = databases # Could not do it in the form
to_explain = request.POST.has_key('button-explain')
to_submit = request.POST.has_key('button-submit')
# Always validate the saveform, which will tell us whether it needs explicit saving
if form.is_valid():
to_save = form.saveform.cleaned_data['save']
to_saveas = form.saveform.cleaned_data['saveas']
if to_save or to_saveas:
if 'beeswax-autosave' in request.session:
del request.session['beeswax-autosave']
if to_saveas and not design.is_auto:
# Save As only affects a previously saved query
design = design.clone()
if to_submit or to_save or to_saveas or to_explain:
explicit_save = to_save or to_saveas
design = save_design(request, form, query_type, design,
explicit_save)
action = urlresolvers.reverse(app_name + ':execute_query',
kwargs=dict(design_id=design.id))
if to_explain or to_submit:
query_str = form.query.cleaned_data["query"]
if conf.CHECK_PARTITION_CLAUSE_IN_QUERY.get():
query_str = _strip_trailing_semicolon(query_str)
# check query. if a select query on partitioned table without partition keys,
# intercept it and raise a PopupException.
_check_partition_clause_in_query(
form.query.cleaned_data.get('database', None),
query_str, db)
# (Optional) Parameterization.
parameterization = get_parameterization(
request, query_str, form, design, to_explain)
if parameterization:
return parameterization
try:
query = HQLdesign(form, query_type=query_type)
if to_explain:
return explain_directly(request, query, design,
query_server)
else:
download = request.POST.has_key('download')
download_format = form.query.cleaned_data.get(
'download_format', None)
if not download_format: download_format = None
if download_format in common.DL_FORMATS:
request.session['dl_status'] = True
return execute_directly(
request,
query,
query_server,
design,
on_success_url=on_success_url,
download_format=download_format,
download=download)
except QueryServerException, ex:
error_message, log = expand_exception(ex, db)
def execute_query(request, design_id=None):
"""
View function for executing an arbitrary query.
It understands the optional GET/POST params:
on_success_url
If given, it will be displayed when the query is successfully finished.
Otherwise, it will display the view query results page by default.
"""
authorized_get_design(request, design_id)
request.session['start_time'] = time.time() # FIXME: add job id to not intersect simultaneous jobs
error_message = None
form = QueryForm()
action = request.path
log = None
app_name = get_app_name(request)
query_type = SavedQuery.TYPES_MAPPING['beeswax']
design = safe_get_design(request, query_type, design_id)
on_success_url = request.REQUEST.get('on_success_url')
query_server = get_query_server_config(app_name)
db = dbms.get(request.user, query_server)
databases = _get_db_choices(request)
if request.method == 'POST':
form.bind(request.POST)
form.query.fields['database'].choices = databases # Could not do it in the form
to_explain = request.POST.has_key('button-explain')
to_submit = request.POST.has_key('button-submit')
# Always validate the saveform, which will tell us whether it needs explicit saving
if form.is_valid():
to_save = form.saveform.cleaned_data['save']
to_saveas = form.saveform.cleaned_data['saveas']
if to_save or to_saveas:
if 'beeswax-autosave' in request.session:
del request.session['beeswax-autosave']
if to_saveas and not design.is_auto:
# Save As only affects a previously saved query
design = design.clone()
if to_submit or to_save or to_saveas or to_explain:
explicit_save = to_save or to_saveas
design = save_design(request, form, query_type, design, explicit_save)
action = urlresolvers.reverse(app_name + ':execute_query', kwargs=dict(design_id=design.id))
if to_explain or to_submit:
query_str = form.query.cleaned_data["query"]
if conf.CHECK_PARTITION_CLAUSE_IN_QUERY.get():
query_str = _strip_trailing_semicolon(query_str)
# check query. if a select query on partitioned table without partition keys,
# intercept it and raise a PopupException.
_check_partition_clause_in_query(form.query.cleaned_data.get('database', None), query_str, db)
# (Optional) Parameterization.
parameterization = get_parameterization(request, query_str, form, design, to_explain)
if parameterization:
return parameterization
try:
query = HQLdesign(form, query_type=query_type)
if to_explain:
return explain_directly(request, query, design, query_server)
else:
download = request.POST.has_key('download')
download_format = form.query.cleaned_data.get('download_format', None)
if not download_format: download_format = None
if download_format in common.DL_FORMATS:
request.session['dl_status'] = True
return execute_directly(request, query, query_server, design, on_success_url=on_success_url, download_format=download_format, download=download)
except QueryServerException, ex:
error_message, log = expand_exception(ex, db)
def execute_query(request, design_id=None, table=None):
authorized_get_design(request, design_id)
error_message, log = None, None
form = query_form()
action = request.path
design = safe_get_design(request, SavedQuery.HQL, design_id)
on_success_url = request.REQUEST.get('on_success_url')
for _ in range(1):
if request.method == 'POST':
form.bind(request.POST)
to_explain = request.POST.has_key('button-explain')
to_submit = request.POST.has_key('button-submit')
# Always validate the saveform, which will tell us whether it needs explicit saving
if not form.is_valid():
break
to_save = form.saveform.cleaned_data['save']
to_saveas = form.saveform.cleaned_data['saveas']
if to_saveas and not design.is_auto:
# Save As only affects a previously saved query
design = design.clone()
if to_submit or to_save or to_saveas or to_explain:
explicit_save = to_save or to_saveas
design = save_design(request, form, SavedQuery.HQL, design,
explicit_save)
action = urlresolvers.reverse(execute_query,
kwargs=dict(design_id=design.id))
# We're not going to process the form. Simply re-render it.
if not to_explain and not to_submit:
break
query_str = _strip_trailing_semicolon(
form.query.cleaned_data["query"])
query_server = db_utils.get_query_server(
form.query_servers.cleaned_data["server"])
# (Optional) Parameterization.
parameterization = get_parameterization(request, query_str, form,
design, to_explain)
if parameterization:
return parameterization
query_msg = make_beeswax_query(request, query_str, form)
try:
if to_explain:
return expl_d(request, query_str, query_msg, design,
query_server)
else:
notify = form.query.cleaned_data.get('email_notify', False)
return e_d(request,
query_msg,
design=design,
on_success_url=on_success_url,
notify=notify)
except BeeswaxException, ex:
error_message, log = expand_exception(ex)
# Fall through to re-render the execute form.
else:
# GET request
if design.id is not None:
data = HQLdesign.loads(design.data).get_query_dict()
form.bind(data)
form.saveform.set_data(design.name, design.desc)
else:
# New design
form.bind()
def execute(request, design_id=None):
response = {'status': -1, 'message': ''}
if request.method != 'POST':
response['message'] = _('A POST request is required.')
app_name = get_app_name(request)
query_server = get_query_server_config(app_name)
query_type = beeswax.models.SavedQuery.TYPES_MAPPING[app_name]
design = safe_get_design(request, query_type, design_id)
try:
query_form = get_query_form(request)
if query_form.is_valid():
query_str = query_form.query.cleaned_data["query"]
explain = request.GET.get('explain', 'false').lower() == 'true'
design = save_design(request, query_form, query_type, design, False)
if query_form.query.cleaned_data['is_parameterized']:
# Parameterized query
parameterization_form_cls = make_parameterization_form(query_str)
if parameterization_form_cls:
parameterization_form = parameterization_form_cls(request.POST.get('query-query', ''), prefix="parameterization")
if parameterization_form.is_valid():
parameters = parameterization_form.cleaned_data
real_query = substitute_variables(query_str, parameters)
query = HQLdesign(query_form, query_type=query_type)
query._data_dict['query']['query'] = real_query
try:
if explain:
return explain_directly(request, query_server, query)
else:
return execute_directly(request, query, design, query_server, parameters=parameters)
except Exception as ex:
db = dbms.get(request.user, query_server)
error_message, log = expand_exception(ex, db)
response['message'] = error_message
return JsonResponse(response)
else:
response['errors'] = parameterization_form.errors
return JsonResponse(response)
# Non-parameterized query
query = HQLdesign(query_form, query_type=query_type)
if request.GET.get('explain', 'false').lower() == 'true':
return explain_directly(request, query_server, query)
else:
return execute_directly(request, query, design, query_server)
else:
response['message'] = _('There was an error with your query.')
response['errors'] = {
'query': [query_form.query.errors],
'settings': query_form.settings.errors,
'file_resources': query_form.file_resources.errors,
'functions': query_form.functions.errors,
}
except RuntimeError as e:
response['message']= str(e)
return JsonResponse(response)
