pikapython/port/linux/format.py
2023-07-17 18:38:13 +08:00

52 lines
1.8 KiB
Python

import os
import sys
import subprocess
from pathlib import Path
def format_files_in_dir(dir_path, dir_skip):
# 遍历目录及其子目录中的所有文件
for filepath in dir_path.rglob('*'):
# Skip if the current file's directory or any of its parents are in dir_skip
if any(str(filepath).startswith(dir) for dir in dir_skip):
continue
# 只处理.c, .h, .cpp文件
if filepath.suffix in ['.c', '.h', '.cpp']:
try:
# 使用clang-format命令格式化文件
subprocess.run(
['clang-format', '-i', str(filepath)], check=True)
print(f'Formatted file: {filepath}')
except subprocess.CalledProcessError as e:
print(
f'Error formatting file: {filepath}. Error message: {str(e)}')
# 确保文件最后一行是空行
with open(filepath, 'r+', encoding='utf-8') as file:
lines = file.readlines()
if lines and not lines[-1].endswith('\n'):
# 如果最后一行不是空行,添加一个
file.write('\n')
print(f'Added newline to file: {filepath}')
# 从命令行参数获取目录列表
dirs = sys.argv[1:]
# 如果没有命令行参数, 则使用默认目录
if len(dirs) == 0:
dirs = ['package/pikascript/pikascript-lib',
'package/pikascript/pikascript-core', 'test']
dir_skip = ['package/pikascript/pikascript-lib/re']
# 对每个目录进行处理
for dir_path_str in dirs:
dir_path = Path(dir_path_str)
if dir_path.is_dir():
format_files_in_dir(dir_path, dir_skip)
print(f'Finished processing directory: {dir_path_str}')
else:
print(f"'{dir_path_str}' is not a directory. Skipping...")