mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-08-17 05:03:44 +08:00
Adeeb Shihadeh
39 lines
1.2 KiB
Python
Executable File
39 lines
1.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import argparse
|
|
import tokenize
|
|
|
|
|
|
# TODO: remove this once https://github.com/astral-sh/ruff/issues/8705 is closed
|
|
def check_indentation(filename: str, indent_width: int = 2) -> bool:
|
|
failed = False
|
|
indent_stack = [0]
|
|
|
|
with tokenize.open(filename) as f:
|
|
tokens = tokenize.generate_tokens(f.readline)
|
|
for token in tokens:
|
|
if token.type == tokenize.INDENT:
|
|
indentation = token.string
|
|
width = len(indentation)
|
|
expected = indent_stack[-1] + indent_width
|
|
|
|
if indentation != " " * expected:
|
|
found = "indentation containing tabs" if "\t" in indentation else f"{width} spaces"
|
|
print(f"{filename}:{token.start[0]}:1: expected {expected} spaces, found {found}")
|
|
failed = True
|
|
indent_stack.append(width)
|
|
elif token.type == tokenize.DEDENT:
|
|
indent_stack.pop()
|
|
|
|
return failed
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Check Python block indentation.")
|
|
parser.add_argument("filenames", nargs="+")
|
|
args = parser.parse_args()
|
|
|
|
failed = False
|
|
for filename in args.filenames:
|
|
failed |= check_indentation(filename)
|
|
raise SystemExit(failed)
|