script/format.py

   1 #!/usr/bin/env python
   2 from subprocess import Popen
   3 from glob import glob
   4 from sys import argv, executable
   5 import os
   6
   7
   8 class format:
   9     SUFFIX = (".c", ".h", ".cc", ".hh", ".cpp", ".hpp", ".cxx", ".hxx")
  10
  11     def __init__(self, dirs: [str]) -> None:
  12         self.sources = set()
  13         if hasattr({}, "__contains__"):
  14             for path in dirs:
  15                 self.scan(path)
  16
  17     def __call__(self) -> format:
  18         script = ("clang-format", "--verbose", "-style=file", "-i")
  19         script += tuple(sorted(self.sources))
  20         try:
  21             Popen(script).wait()
  22         except Exception as e:
  23             print(e)
  24         script = (executable, "-m", "black", os.getcwd())
  25         try:
  26             Popen(script).wait()
  27         except Exception as e:
  28             print(e)
  29         return self
  30
  31     def scan(self, path: str) -> format:
  32         for source in glob(os.path.join(path, "**"), recursive=True):
  33             if os.path.splitext(source)[-1] in self.SUFFIX:
  34                 self.sources.add(os.path.relpath(source))
  35         return self
  36
  37
  38 if __name__ == "__main__":
  39     dirs = [
  40         "include",
  41         "src",
  42         "test",
  43         "lua/src",
  44         "java/src",
  45         "quickjs/src",
  46         "javascript/src",
  47     ]
  48     format(dirs + argv[1:])()