def stateio_import(ct_id):
'''
Background task to import state
'''
# check for livy session
_check_livy_session()
ct = models.CombineBackgroundTask.objects.get(pk=int(ct_id))
logger.info('using %s' % ct)
# begin import
sio_client = models.StateIOClient()
sio_client.import_state(stateio_id=ct.task_params['stateio_id'],
import_name=ct.task_params['import_name'],
export_path=ct.task_params['export_path'])
def stateio_export(ct_id):
'''
Background task to export state
'''
# check for livy session
_check_livy_session()
ct = models.CombineBackgroundTask.objects.get(pk=int(ct_id))
logger.info('using %s' % ct)
# begin export
sio_client = models.StateIOClient()
sio_client.export_state(
stateio_id=ct.task_params['stateio_id'],
jobs=ct.task_params['jobs'],
record_groups=ct.task_params['record_groups'],
orgs=ct.task_params['orgs'],
config_scenarios=ct.task_params['config_scenarios'],
export_name=ct.task_params['export_name'],
compress=True,
compression_format='zip')