def load_location_file(location_file, request):
'''
We would like to merge the current active model into the new model
created by our location file prior to clearing our session
'''
if isdir(location_file):
active_model = get_active_model(request)
new_model = load(location_file)
new_model._cache.enabled = False
if active_model is not None:
active_model._map = new_model._map
active_model._time_step = new_model._time_step
active_model._num_time_steps = new_model._num_time_steps
active_model.merge(new_model)
else:
active_model = new_model
name = split(location_file)[1]
if name != '':
active_model.name = name
init_session_objects(request, force=True)
log.debug("model loaded - begin registering objects")
RegisterObject(active_model, request)
set_active_model(request, active_model.id)
def create_map(request):
'''Creates a Gnome Map object.'''
log_prefix = 'req({0}): create_map():'.format(id(request))
init_session_objects(request)
try:
json_request = ujson.loads(request.body)
except:
raise cors_exception(request, HTTPBadRequest)
if not JSONImplementsOneOf(json_request, implemented_types):
raise cors_exception(request, HTTPNotImplemented)
if 'filename' in json_request:
json_request['filename'] = get_file_path(request,
json_request=json_request)
gnome_sema = request.registry.settings['py_gnome_semaphore']
gnome_sema.acquire()
log.info(' ' + log_prefix + 'semaphore acquired...')
try:
obj = CreateObject(json_request, get_session_objects(request))
except:
raise cors_exception(request,
HTTPUnsupportedMediaType,
with_stacktrace=True)
finally:
gnome_sema.release()
log.info(' ' + log_prefix + 'semaphore released...')
set_session_object(obj, request)
return obj.serialize()
def load_location_file(location_file, request):
'''
We would like to merge the current active model into the new model
created by our location file prior to clearing our session
'''
if isdir(location_file):
active_model = get_active_model(request)
new_model = Model.load(location_file)
new_model._cache.enabled = False
if active_model is not None:
active_model._map = new_model._map
active_model._time_step = new_model._time_step
active_model._num_time_steps = new_model._num_time_steps
active_model.merge(new_model)
else:
active_model = new_model
name = split(location_file)[1]
if name != '':
active_model.name = name
init_session_objects(request, force=True)
log.debug("model loaded - begin registering objects")
RegisterObject(active_model, request)
set_active_model(request, active_model.id)
def create_map(request):
'''Creates a Gnome Map object.'''
log_prefix = 'req({0}): create_map():'.format(id(request))
init_session_objects(request)
try:
json_request = ujson.loads(request.body)
except:
raise cors_exception(request, HTTPBadRequest)
if not JSONImplementsOneOf(json_request, implemented_types):
raise cors_exception(request, HTTPNotImplemented)
if 'filename' in json_request:
json_request['filename'] = get_file_path(request,
json_request=json_request)
gnome_sema = request.registry.settings['py_gnome_semaphore']
gnome_sema.acquire()
log.info(' ' + log_prefix + 'semaphore acquired...')
try:
obj = CreateObject(json_request, get_session_objects(request))
except:
raise cors_exception(request, HTTPUnsupportedMediaType,
with_stacktrace=True)
finally:
gnome_sema.release()
log.info(' ' + log_prefix + 'semaphore released...')
set_session_object(obj, request)
return obj.serialize()
def upload_model(request):
'''
Uploads a new model in the form of a zipfile and registers it as the
current active model.
We are generating our own filename instead of trusting
the incoming filename since that might result in insecure paths.
We may want to eventually use something other than /tmp,
and if you write to an untrusted location you will need to do
some extra work to prevent symlink attacks.
'''
clean_session_dir(request)
file_path = process_upload(request, 'new_model')
# Now that we have our file, we will now try to load the model into
# memory.
# Now that we have our file, is it a zipfile?
if not is_savezip_valid(file_path):
raise cors_response(
request, HTTPBadRequest('Incoming file is not a '
'valid zipfile!'))
# now we try to load our model from the zipfile.
gnome_sema = request.registry.settings['py_gnome_semaphore']
gnome_sema.acquire()
log.info('semaphore acquired.')
try:
log.info('loading our model from zip...')
new_model = load(file_path)
new_model._cache.enabled = False
init_session_objects(request, force=True)
RegisterObject(new_model, request)
log.info('setting active model...')
set_active_model(request, new_model.id)
except:
raise cors_exception(request, HTTPBadRequest, with_stacktrace=True)
finally:
gnome_sema.release()
log.info('semaphore released.')
# We will want to clean up our tempfile when we are done.
os.remove(file_path)
return cors_response(request, Response('OK'))
def create_model(request):
'''
Creates a new model
'''
log_prefix = 'req({0}): create_object():'.format(id(request))
log.info('>>' + log_prefix)
try:
json_request = ujson.loads(request.body.decode('utf-8'))
except Exception:
json_request = None
if json_request and not JSONImplementsOneOf(json_request,
implemented_types):
raise cors_exception(request, HTTPNotImplemented)
session_lock = acquire_session_lock(request)
log.info(' {} session lock acquired (sess:{}, thr_id: {})'.format(
log_prefix, id(session_lock),
current_thread().ident))
try:
clean_session_dir(request)
init_session_objects(request, force=True)
if json_request:
new_model = CreateObject(json_request,
get_session_objects(request))
else:
new_model = Model()
set_session_object(new_model, request)
set_active_model(request, new_model.id)
except Exception:
raise cors_exception(request,
HTTPUnsupportedMediaType,
with_stacktrace=True)
finally:
session_lock.release()
log.info(' {} session lock released (sess:{}, thr_id: {})'.format(
log_prefix, id(session_lock),
current_thread().ident))
log.info('<<' + log_prefix)
return new_model.serialize(options=web_ser_opts)
def upload_model(request):
'''
Uploads a new model in the form of a zipfile and registers it as the
current active model.
We are generating our own filename instead of trusting
the incoming filename since that might result in insecure paths.
We may want to eventually use something other than /tmp,
and if you write to an untrusted location you will need to do
some extra work to prevent symlink attacks.
'''
clean_session_dir(request)
file_path = process_upload(request, 'new_model')
# Now that we have our file, we will now try to load the model into
# memory.
# Now that we have our file, is it a zipfile?
if not is_savezip_valid(file_path):
raise cors_response(request, HTTPBadRequest('Incoming file is not a '
'valid zipfile!'))
# now we try to load our model from the zipfile.
gnome_sema = request.registry.settings['py_gnome_semaphore']
gnome_sema.acquire()
log.info('semaphore acquired.')
try:
log.info('loading our model from zip...')
new_model = load(file_path)
new_model._cache.enabled = False
init_session_objects(request, force=True)
RegisterObject(new_model, request)
log.info('setting active model...')
set_active_model(request, new_model.id)
except:
raise cors_exception(request, HTTPBadRequest, with_stacktrace=True)
finally:
gnome_sema.release()
log.info('semaphore released.')
# We will want to clean up our tempfile when we are done.
os.remove(file_path)
return cors_response(request, Response('OK'))
def create_model(request):
'''
Creates a new model
'''
log_prefix = 'req({0}): create_object():'.format(id(request))
log.info('>>' + log_prefix)
try:
json_request = ujson.loads(request.body)
except:
json_request = None
if json_request and not JSONImplementsOneOf(json_request,
implemented_types):
raise cors_exception(request, HTTPNotImplemented)
gnome_sema = request.registry.settings['py_gnome_semaphore']
gnome_sema.acquire()
log.info(' ' + log_prefix + 'semaphore acquired...')
try:
init_session_objects(request, force=True)
if json_request:
new_model = CreateObject(json_request,
get_session_objects(request))
else:
new_model = Model()
set_session_object(new_model, request)
set_session_object(new_model._map, request)
set_active_model(request, new_model.id)
except:
raise cors_exception(request,
HTTPUnsupportedMediaType,
with_stacktrace=True)
finally:
gnome_sema.release()
log.info(' ' + log_prefix + 'semaphore released...')
log.info('<<' + log_prefix)
return new_model.serialize()
def create_model(request):
'''
Creates a new model
'''
log_prefix = 'req({0}): create_object():'.format(id(request))
log.info('>>' + log_prefix)
try:
json_request = ujson.loads(request.body)
except:
json_request = None
if json_request and not JSONImplementsOneOf(json_request,
implemented_types):
raise cors_exception(request, HTTPNotImplemented)
gnome_sema = request.registry.settings['py_gnome_semaphore']
gnome_sema.acquire()
log.info(' ' + log_prefix + 'semaphore acquired...')
try:
init_session_objects(request, force=True)
if json_request:
new_model = CreateObject(json_request,
get_session_objects(request))
else:
new_model = Model()
set_session_object(new_model, request)
set_session_object(new_model._map, request)
set_active_model(request, new_model.id)
except:
raise cors_exception(request, HTTPUnsupportedMediaType,
with_stacktrace=True)
finally:
gnome_sema.release()
log.info(' ' + log_prefix + 'semaphore released...')
log.info('<<' + log_prefix)
return new_model.serialize()
def get_info(request):
request.session.redis.config_set("notify-keyspace-events", "Ex")
init_session_objects(request, force=False)
return {'id': request.session.session_id}
