diff --git a/bf_pulp_utils/bf_pulp_utils/psrfits_requantisation.py b/bf_pulp_utils/bf_pulp_utils/psrfits_requantisation.py index 7aaf82aeee54c3b51c7c4cc050756ec20531faf7..db136ac3be3ef40f01305471c123e3553e62c274 100644 --- a/bf_pulp_utils/bf_pulp_utils/psrfits_requantisation.py +++ b/bf_pulp_utils/bf_pulp_utils/psrfits_requantisation.py @@ -48,7 +48,7 @@ def comparison_summary(sasid, input_path, output_path, input_size, output_size, json.dump(summary, fp) # writing out the summary comparison between old and new tarballs -def formatted_comparison_summary(sasid, input_tarball, new_output_name, to_delete, to_extract): +def formatted_comparison_summary(sasid, input_tarball, new_output_name, input_size, output_size, to_delete, to_extract): in_filename = input_tarball.split('/')[-1] out_filename = new_output_name.split('/')[-1] @@ -59,9 +59,7 @@ def formatted_comparison_summary(sasid, input_tarball, new_output_name, to_delet fsum.write((" " + " | ".join([in_filename, out_filename])+ " ").center(98, "*") + "\n") #fsum.write("-"*98+"\n") - input_size = os.path.getsize(input_tarball) - output_size = os.path.getsize(new_output_name) - ratio = (float)(output_size) / input_size + ratio = output_size / input_size factors = {10: "kB", 20: "MB", 30: "GB", 40: "TB"} for ii in factors.keys(): if (float)(input_size) / 2 ** ii < 1: @@ -292,7 +290,7 @@ def main(): # writing out the summary comparison between old and new tarballs comparison_summary(sasid, input_tarball, input_tarball, input_size, output_size, to_delete, to_add, rfilevel) - formatted_comparison_summary(sasid, input_tarball, input_tarball, to_delete, to_add) + formatted_comparison_summary(sasid, input_tarball, input_tarball, input_size, output_size, to_delete, to_add) if __name__ == '__main__': main()