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')