spotbugsComparison.py 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. #!/usr/bin/env python3
  2. import argparse
  3. import defusedxml.ElementTree as ET
  4. import spotbugsSummary
  5. def print_comparison(old: dict, new: dict, link_base: str, link_new: str):
  6. all_keys = sorted(set(list(old.keys()) + list(new.keys())))
  7. output = "<table><tr><th>Category</th>"
  8. old_header = f"<a href='{link_base}'>Base</a>" if link_base is not None else "Base"
  9. output += f"<th>{old_header}</th>"
  10. new_header = f"<a href='{link_new}'>New</a>" if link_new is not None else "New"
  11. output += f"<th>{new_header}</th>"
  12. output += "</tr>"
  13. for category in all_keys:
  14. category_count_old = old[category] if category in old else 0
  15. category_count_new = new[category] if category in new else 0
  16. new_str = f"<b>{category_count_new}</b>" if category_count_new != category_count_old else str(category_count_new)
  17. output += "<tr>"
  18. output += f"<td>{category}</td>"
  19. output += f"<td>{category_count_old}</td>"
  20. output += f"<td>{new_str}</td>"
  21. output += "</tr>"
  22. output += "<tr>"
  23. output += "<td><b>Total</b></td>"
  24. output += f"<td><b>{sum(old.values())}</b></td>"
  25. output += f"<td><b>{sum(new.values())}</b></td>"
  26. output += "</tr>"
  27. output += "</table>"
  28. print(output)
  29. if __name__ == "__main__":
  30. parser = argparse.ArgumentParser()
  31. parser.add_argument("base_file", help="base file for comparison")
  32. parser.add_argument("new_file", help="new file for comparison")
  33. parser.add_argument("--link-base", help="http link to base html report")
  34. parser.add_argument("--link-new", help="http link to new html report")
  35. args = parser.parse_args()
  36. base_tree = ET.parse(args.base_file)
  37. base_summary = spotbugsSummary.get_counts(base_tree)
  38. new_tree = ET.parse(args.new_file)
  39. new_summary = spotbugsSummary.get_counts(new_tree)
  40. print_comparison(base_summary, new_summary, args.link_base, args.link_new)