test-example-projects/c_projects/30wklite/tools/check_style.py

#!python
import io
import os
import re
import sys

def check_line_start(line):
	for ch in line:
		if(ch == " "):
			return False
		if(ch != "\t"):
			return True

def check_line_end(line):
	if line.endswith(" \r\n"):
		return False
	if line.endswith("\t\r\n"):
		return False
	return True

def check_line_crlf(line, crlf):
	cr = 0
	lf = 0
	for ch in line:
		if(ch == "\r"):
			cr = cr + 1
		if(ch == "\n"):
			lf = lf + 1
	if crlf == "windows":
		if cr == 1 and lf == 1:
			return True
	if crlf == "unix":
		if cr == 0 and lf == 1:
			return True
	if crlf == "mac":
		if cr == 1 and lf == 0:
			return True
	return False
	
def check_space_line(line):
	for ch in line:
		if ch == " ":
			continue
		if ch == "\t":
			continue
		if ch == "\r":
			continue
		if ch == "\n":
			return False
		return True
	return False

def get_crlf_style(line):
	if line.endswith("\r\n"):
		return "windows"
	elif line.endswith("\n"):
		return "unix"
	elif line.endswith("\r"):
		return "mac"
	return "error"

def check_style(path):
	#print(path)
	crlf = None
	err_count = 0
	line_num = 0
	space_line = 0
	file = io.open(path, "rb")
	for line in file:
		if err_count > 10:
			print(path, "错误过多，已跳过检查")
			break;
		line = line.decode('latin1')
		if crlf is None:
			crlf = get_crlf_style(line)
			if crlf != "windows":
				print(path, "换行格式:", crlf)
		line_num = line_num + 1
		if not check_space_line(line):
			space_line = space_line + 1
			if(space_line > 1):
				print(path, "第%d行:" % (line_num), "连续空行")
				err_count = err_count + 1
			continue
		else:
			space_line = 0
		if not check_line_crlf(line, crlf):
			print(path, "第%d行:" % (line_num), "换行不一致")
			err_count = err_count + 1
			continue
		if not check_line_start(line):
			print(path, "第%d行:" % (line_num), "行首有空格")
			err_count = err_count + 1
			continue
		if not check_line_end(line):
			print(path, "第%d行:" % (line_num), "行尾有空格")
			err_count = err_count + 1
			continue
	file.close()
	
def is_source_code(path):
	path_lower = path.lower()
	if path_lower.endswith(".c"):
		return True
	if path_lower.endswith(".h"):
		return True
	if path_lower.endswith(".s"):
		return True
	return False

def scan_dir_files(dir):
	list = os.listdir(dir)
	for item in list:
		path = dir + "\\" + item
		if os.path.isfile(path):
			if is_source_code(path):
				check_style(path)
		if os.path.isdir(path):
			scan_dir_files(path)
	return None

def main():
	if len(sys.argv) != 2:
		print("python check_style.py <path>")
		return False
	scan_dir_files(sys.argv[1])
	print("检查完毕")
	
main()