file_format.py 1.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. import sys
  4. if len(sys.argv) < 2:
  5. print("Invalid usage of file_format.py, it should be called with a path to one or multiple files.")
  6. sys.exit(1)
  7. BOM = b"\xef\xbb\xbf"
  8. changed = []
  9. invalid = []
  10. for file in sys.argv[1:]:
  11. try:
  12. with open(file, "rt", encoding="utf-8") as f:
  13. original = f.read()
  14. except UnicodeDecodeError:
  15. invalid.append(file)
  16. continue
  17. if original == "":
  18. continue
  19. revamp = "\n".join([line.rstrip("\n\r\t ") for line in original.splitlines(True)]).rstrip("\n") + "\n"
  20. new_raw = revamp.encode(encoding="utf-8")
  21. if new_raw.startswith(BOM):
  22. new_raw = new_raw[len(BOM) :]
  23. with open(file, "rb") as f:
  24. old_raw = f.read()
  25. if old_raw != new_raw:
  26. changed.append(file)
  27. with open(file, "wb") as f:
  28. f.write(new_raw)
  29. if changed:
  30. for file in changed:
  31. print(f"FIXED: {file}")
  32. if invalid:
  33. for file in invalid:
  34. print(f"REQUIRES MANUAL CHANGES: {file}")
  35. sys.exit(1)