def api_simulationFrame(frame_id):
#TODO(robnagler) startTime is reportParametersHash; need version on URL and/or param names in URL
keys = [
'simulationType', 'simulationId', 'modelName', 'animationArgs',
'frameIndex', 'startTime'
]
data = dict(zip(keys, frame_id.split('*')))
template = sirepo.template.import_module(data)
data['report'] = template.get_animation_name(data)
run_dir = simulation_db.simulation_run_dir(data)
model_data = simulation_db.read_json(
run_dir.join(template_common.INPUT_BASE_NAME))
frame = template.get_simulation_frame(run_dir, data, model_data)
response = _json_response(frame)
if 'error' not in frame and template.WANT_BROWSER_FRAME_CACHE:
now = datetime.datetime.utcnow()
expires = now + datetime.timedelta(365)
response.headers['Cache-Control'] = 'public, max-age=31536000'
response.headers['Expires'] = expires.strftime(
"%a, %d %b %Y %H:%M:%S GMT")
response.headers['Last-Modified'] = now.strftime(
"%a, %d %b %Y %H:%M:%S GMT")
else:
_no_cache(response)
return response
def api_simulationFrame(frame_id):
#TODO(robnagler) startTime is reportParametersHash; need version on URL and/or param names in URL
keys = ['simulationType', 'simulationId', 'modelName', 'animationArgs', 'frameIndex', 'startTime']
data = dict(zip(keys, frame_id.split('*')))
template = sirepo.template.import_module(data)
data['report'] = template.get_animation_name(data)
run_dir = simulation_db.simulation_run_dir(data)
model_data = simulation_db.read_json(run_dir.join(template_common.INPUT_BASE_NAME))
if feature_config.cfg.runner_daemon:
# XX TODO: it would be better if the frontend passed the jhash to this
# call. Since it doesn't, we have to read it out of the run_dir, which
# creates a race condition -- we might return a frame from a different
# version of the report than the one the frontend expects.
jhash = template_common.report_parameters_hash(model_data)
frame = runner_client.run_extract_job(
run_dir, jhash, 'get_simulation_frame', data,
)
else:
frame = template.get_simulation_frame(run_dir, data, model_data)
resp = http_reply.gen_json(frame)
if 'error' not in frame and template.WANT_BROWSER_FRAME_CACHE:
now = datetime.datetime.utcnow()
expires = now + datetime.timedelta(365)
resp.headers['Cache-Control'] = 'public, max-age=31536000'
resp.headers['Expires'] = expires.strftime("%a, %d %b %Y %H:%M:%S GMT")
resp.headers['Last-Modified'] = now.strftime("%a, %d %b %Y %H:%M:%S GMT")
else:
_no_cache(resp)
return resp
def app_download_data_file(simulation_type, simulation_id, model, frame):
data = {
'simulationType': simulation_type,
'simulationId': simulation_id,
'modelName': model,
}
frame = int(frame)
template = sirepo.template.import_module(data)
if frame >= 0:
data['report'] = template.get_animation_name(data)
else:
data['report'] = model
run_dir = simulation_db.simulation_run_dir(data)
filename, content, content_type = template.get_data_file(run_dir, model, frame)
return _as_attachment(flask.make_response(content), content_type, filename)
def app_download_data_file(simulation_type, simulation_id, model, frame):
data = {
'simulationType': simulation_type,
'simulationId': simulation_id,
'modelName': model,
}
frame = int(frame)
template = sirepo.template.import_module(data)
if frame >= 0:
data['report'] = template.get_animation_name(data)
else:
data['report'] = model
run_dir = simulation_db.simulation_run_dir(data)
filename, content, content_type = template.get_data_file(run_dir, model, frame)
return _as_attachment(flask.make_response(content), content_type, filename)
def api_downloadDataFile(simulation_type, simulation_id, model, frame, suffix=None):
data = {
'simulationType': sirepo.template.assert_sim_type(simulation_type),
'simulationId': simulation_id,
'modelName': model,
}
options = pkcollections.Dict(data)
options.suffix = suffix
frame = int(frame)
template = sirepo.template.import_module(data)
if frame >= 0:
data['report'] = template.get_animation_name(data)
else:
data['report'] = model
run_dir = simulation_db.simulation_run_dir(data)
filename, content, content_type = template.get_data_file(run_dir, model, frame, options=options)
return _as_attachment(flask.make_response(content), content_type, filename)
def app_simulation_frame(frame_id):
#TODO(robnagler) startTime is reportParametersHash; need version on URL and/or param names in URL
keys = ['simulationType', 'simulationId', 'modelName', 'animationArgs', 'frameIndex', 'startTime']
data = dict(zip(keys, frame_id.split('*')))
template = sirepo.template.import_module(data)
data['report'] = template.get_animation_name(data)
run_dir = simulation_db.simulation_run_dir(data)
model_data = simulation_db.read_json(run_dir.join(template_common.INPUT_BASE_NAME))
response = _json_response(template.get_simulation_frame(run_dir, data, model_data))
if template.WANT_BROWSER_FRAME_CACHE:
now = datetime.datetime.utcnow()
expires = now + datetime.timedelta(365)
response.headers['Cache-Control'] = 'public, max-age=31536000'
response.headers['Expires'] = expires.strftime("%a, %d %b %Y %H:%M:%S GMT")
response.headers['Last-Modified'] = now.strftime("%a, %d %b %Y %H:%M:%S GMT")
else:
_no_cache(response)
return response