save on local machine without failure

This commit is contained in:
Matthew Stidham
2021-01-22 17:04:50 -08:00
committed by Jed Reynolds
parent 708c785b0c
commit 9d7e518bd3

View File

@@ -202,19 +202,23 @@ python3 ./test_ipv4_variable_time.py
#Create directory #Create directory
if args.report_file is None: if args.report_file is None:
homedir = str(datetime.datetime.now().strftime("%Y-%m-%d-%H-%M")).replace(':','-')+'test_ipv4_variable_time' try:
path = os.path.join('/home/lanforge/report-data/',homedir) homedir = str(datetime.datetime.now().strftime("%Y-%m-%d-%H-%M")).replace(':','-')+'test_ipv4_variable_time'
os.mkdir(path) path = os.path.join('/home/lanforge/report-data/',homedir)
os.mkdir(path)
except:
path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
print('Saving file to local directory')
else: else:
pass pass
if args.report_file is None: if args.report_file is None:
if args.output_format in ['csv','json','html','hdf','stata','pickle','pdf','png','df','parquet','xlsx']: if args.output_format in ['csv','json','html','hdf','stata','pickle','pdf','png','df','parquet','xlsx']:
report_f='/home/lanforge/report-data/'+homedir+'/data.' + args.output_format report_f=path+'/data.' + args.output_format
output=args.output_format output=args.output_format
else: else:
print('Defaulting data file output type to Excel') print('Defaulting data file output type to Excel')
report_f='/home/lanforge/report-data/'+homedir+'/data.xlsx' report_f=path+'/data.xlsx'
output='xlsx' output='xlsx'
else: else:
report_f=args.report_file report_f=args.report_file
@@ -257,7 +261,7 @@ python3 ./test_ipv4_variable_time.py
col_names=None col_names=None
else: else:
col_names = args.col_names col_names = args.col_names
print(col_names) print(report_f)
ip_var_test.l3cxprofile.monitor(col_names=col_names, ip_var_test.l3cxprofile.monitor(col_names=col_names,
report_file=report_f, report_file=report_f,
duration_sec=ip_var_test.local_realm.parse_time(args.test_duration).total_seconds(), duration_sec=ip_var_test.local_realm.parse_time(args.test_duration).total_seconds(),