def view_jobflow_with_ajax(context, request):
req_type = request.params.get('type')
resource = get_resource(context)
emr_conn = get_emr_connection(resource)
if req_type == 'runjobflow' and is_runnable_jobflow(context):
run_jobflow(emr_conn, resource, context)
jobflow_ids = get_jobflow_ids(context)
if req_type == 'terminate' and jobflow_ids:
emr_conn.terminate_jobflows(jobflow_ids)
# FIXME: enforce a termination against protection
if req_type == 'getlog':
s3_conn = get_s3_connection(resource)
keys_list = get_s3_log_keys(s3_conn, resource.log_uri, jobflow_ids)
bucket_name = keys_list[0].bucket.name if keys_list else u''
keys = [dict(get_object_attributes(key, ['name', 'size']))
for key in chain.from_iterable(keys_list)]
return {'keys': keys, 'bucket_name': bucket_name}
else: # refresh
res_jobflows = get_jobflows_for_json(emr_conn, jobflow_ids)
set_jobflow_state(context, res_jobflows)
return res_jobflows
def view_jobflow(context, request):
return {
'api': template_api(context, request),
'bootstraps': get_bootstraps(context),
'is_runnable': is_runnable_jobflow(context),
'resource': get_resource(context),
'resource_models': [JobFlow],
}
def view_jobservice(context, request):
return {
'api': template_api(context, request),
'bootstraps': False,
'is_runnable': False,
'resource': get_resource(context),
'resource_models': [EMRJobResource, JobService, Bootstrap],
}
def deferred_default_step_args(node, kw):
jobtype = get_context_data(kw['request'].context, 'jobflow', ['jobtype'])
if jobtype == u'hive':
resource = get_resource(kw['request'].context)
args = _DEFAULT_HIVE_STEP_ARGS
if resource.hive_versions:
args += u'--hive-versions\n{0}\n'.format(resource.hive_versions)
return args
elif jobtype == u'custom-jar':
return _DEFAULT_CUSTOM_JAR_STEP_ARGS
else:
return u''
def view_download_file(context, request):
resource = get_resource(context)
s3_conn = get_s3_connection(resource)
bucket = s3_conn.get_bucket(request.params.get('bucket'))
key_path = request.params.get('key')
key = bucket.lookup(key_path)
response = Response()
fp = get_temporary_file(key.size)
if fp:
key.get_contents_to_file(fp)
fp.seek(0)
con_dis = 'attachment; filename={0}'.format(basename(key_path))
response.content_disposition = con_dis
response.app_iter = fp
else:
response.text = u'Download it yourself, '
response.text += u'because of large file size: {0}'.format(key.size)
return response
