def get(self, request, modelname, version):
"""
Retrieve parameters of model version
"""
current_path = os.getcwd()
temp_dir = tempfile.mkdtemp(prefix="series_", dir=None)
os.chdir(temp_dir)
flame_status = manage.action_series(modelname, version)
if flame_status[0] == False:
os.chdir(current_path)
return JsonResponse({'error': flame_status[1]},
status=status.HTTP_404_NOT_FOUND)
with open(os.path.abspath('training_series.sdf'), 'r') as f:
series_content = f.read()
os.chdir(current_path)
shutil.rmtree(temp_dir)
return Response(series_content, status=status.HTTP_200_OK)
def get(self, request, modelname, version):
"""
Retrieve parameters of model version
"""
current_path = os.getcwd()
temp_dir = tempfile.mkdtemp(prefix="series_", dir=None)
os.chdir(temp_dir)
flame_status = manage.action_series(modelname, version)
if flame_status[0] == False:
os.chdir(current_path)
return JsonResponse({'error': flame_status[1]},
status=status.HTTP_404_NOT_FOUND)
file = open(os.path.abspath('training_series.sdf'), 'rb')
response = HttpResponse(FileWrapper(file), content_type='text')
response[
'Content-Disposition'] = 'attachment; filename=training_series.sdf'
os.chdir(current_path)
shutil.rmtree(temp_dir)
return response
def manage_cmd(args):
'''
Calls diverse model or space maintenance commands
'''
version = utils.intver(args.version)
if args.space is not None or 'searches' in args.action :
import flame.smanage as smanage
if args.action == 'new':
success, results = smanage.action_new(args.space)
elif args.action == 'kill':
success, results = smanage.action_kill(args.space)
elif args.action == 'remove':
success, results = smanage.action_remove(args.space, version)
elif args.action == 'publish':
success, results = smanage.action_publish(args.space)
elif args.action == 'list':
success, results = smanage.action_list(args.space)
elif args.action == 'parameters':
success, results = smanage.action_parameters(args.space, version)
elif args.action == 'info':
success, results = smanage.action_info(args.space, version)
elif args.action == 'dir':
success, results = smanage.action_dir()
elif args.action == 'searches_result':
success, results = smanage.action_searches_result(args.label)
else:
success = False
results = "Specified manage action is not defined"
else:
import flame.manage as manage
if args.action == 'new':
success, results = manage.action_new(args.endpoint)
elif args.action == 'kill':
success, results = manage.action_kill(args.endpoint)
elif args.action == 'remove':
success, results = manage.action_remove(args.endpoint, version)
elif args.action == 'publish':
success, results = manage.action_publish(args.endpoint)
elif args.action == 'list':
success, results = manage.action_list(args.endpoint)
elif args.action == 'export':
success, results = manage.action_export(args.endpoint)
elif args.action == 'info':
success, results = manage.action_info(args.endpoint, version)
elif args.action == 'refresh':
if args.version == None:
version = None
success, results = manage.action_refresh(args.endpoint, version)
elif args.action == 'series':
success, results = manage.action_series(args.endpoint, version)
elif args.action == 'results':
success, results = manage.action_results(args.endpoint, version)
elif args.action == 'parameters':
success, results = manage.action_parameters(args.endpoint, version)
elif args.action == 'documentation':
success, results = manage.action_documentation(args.endpoint,
version, args.documentation_file)
elif args.action == 'model_template':
success, results = manage.action_model_template(args.endpoint,
version, args.documentation_file)
elif args.action == 'prediction_template':
success, results = manage.action_prediction_template(args.endpoint, version)
elif args.action == 'import':
success, results = manage.action_import(args.infile)
elif args.action == 'dir':
success, results = manage.action_dir()
elif args.action == 'report':
success, results = manage.action_report()
elif args.action == 'list':
success, results = manage.action_list(args.endpoint)
elif args.action == 'predictions':
success, results = manage.action_predictions_list()
elif args.action == 'predictions_result':
success, results = manage.action_predictions_result(args.label)
elif args.action == 'predictions_remove':
success, results = manage.action_predictions_remove(args.label)
elif args.action == 'label':
success, results = manage.action_label(args.endpoint, version, args.label)
else:
success = False
results = "Specified manage action is not defined"
return success, results