ddnet/scripts/fix_style.py

42 lines
1.1 KiB
Python
Raw Normal View History

2020-09-10 20:58:30 +00:00
#!/usr/bin/env python3
from collections import defaultdict
import os
import re
import subprocess
import sys
os.chdir(os.path.dirname(__file__) + "/..")
ignore_files = ["src/engine/keys.h", "src/engine/client/keynames.h"]
def recursive_file_list(path):
result = []
for dirpath, dirnames, filenames in os.walk(path):
result += filter(lambda p: p not in ignore_files, [os.path.join(dirpath, filename) for filename in filenames])
return result
def filter_cpp(filenames):
return [filename for filename in filenames
if any(filename.endswith(ext) for ext in ".c .cpp .h".split())]
def reformat(filenames):
subprocess.check_call(["clang-format", "-i"] + filenames)
def warn(filenames):
return subprocess.call(["clang-format", "-Werror", "--dry-run"] + filenames)
def main():
import argparse
p = argparse.ArgumentParser(description="Check and fix style of changed files")
p.add_argument("-n", "--dry-run", action="store_true", help="Don't fix, only warn")
args = p.parse_args()
filenames = filter_cpp(recursive_file_list("src"))
if not args.dry_run:
reformat(filenames)
else:
sys.exit(warn(filenames))
if __name__ == "__main__":
main()