import json
import pickle
import os
import datetime

with open("fio_result.json") as fio_raw:
    fio_data=json.load(fio_raw)

r_iops=[];
r_io_bytes=[];
r_io_runtime=[];
r_lat=[];
w_iops=[];
w_io_bytes=[];
w_io_runtime=[];
w_lat=[];



total_jobs=len(fio_data["jobs"])

for x in range (0,int(total_jobs)):
    r_iops.append(fio_data["jobs"][x]["read"]["iops"])
    r_io_bytes.append(fio_data["jobs"][x]["read"]["io_bytes"])
    r_io_runtime.append(fio_data["jobs"][x]["read"]["runtime"])
    r_lat.append(fio_data["jobs"][x]["read"]["lat"]["mean"])
    w_iops.append(fio_data["jobs"][x]["write"]["iops"])
    w_io_bytes.append(fio_data["jobs"][x]["write"]["io_bytes"])
    w_io_runtime.append(fio_data["jobs"][x]["write"]["runtime"])
    w_lat.append(fio_data["jobs"][x]["write"]["lat"]["mean"])



FIO_result_dict={};

for x in range (0,total_jobs):
    FIO_result_dict['Job_'+str(x)]={};
    FIO_result_dict['Job_'+str(x)]['read']={};
    FIO_result_dict['Job_'+str(x)]['read']['Total_IO_Bytes']=r_io_bytes[x]
    FIO_result_dict['Job_'+str(x)]['read']['IO/sec']=r_iops[x]
    FIO_result_dict['Job_'+str(x)]['read']['IO_runtime (millisec)']=r_io_runtime[x]
    FIO_result_dict['Job_'+str(x)]['read']['mean_IO_latenchy (microsec)']=r_lat[x]
    
    FIO_result_dict['Job_'+str(x)]['write']={};
    FIO_result_dict['Job_'+str(x)]['write']['Total_IO_Bytes']=w_io_bytes[x]
    FIO_result_dict['Job_'+str(x)]['write']['IO/sec']=w_iops[x]
    FIO_result_dict['Job_'+str(x)]['write']['IO_runtime (millisec)']=w_io_runtime[x]
    FIO_result_dict['Job_'+str(x)]['write']['mean_IO_latenchy (microsec)']=w_lat[x]



host_name = (os.popen("hostname").read().rstrip())
report_time = str(datetime.datetime.utcnow().isoformat())
os.system("mv fio_result.json "+str(host_name)+"-"+report_time+".log")
with open('./result_temp','w+')as out_fio_result:
    pickle.dump(FIO_result_dict,out_fio_result)