pocketpy/scripts/format.py
blueloveTH 6805b418b5 ...
2024-08-06 14:00:59 +08:00

23 lines
728 B
Python

import os
import subprocess
def get_all_files(root: str):
for path, _, files in os.walk(root):
for file in files:
fullpath = os.path.join(path, file)
# ignore some files
if fullpath.startswith('include/pybind11'):
continue
if file.startswith('_'):
continue
if not file.endswith('.c') and not file.endswith('.h') and not file.endswith('.hpp'):
continue
yield fullpath
if __name__ == '__main__':
files = []
files.extend(get_all_files('include'))
files.extend(get_all_files('src'))
files.extend(get_all_files('src2'))
subprocess.run(['clang-format', '-i'] + files, check=True)