import sys

from pathlib import Path

TARGET_DIR = Path("src/python_testing/")

# This is a linter that checks for 'except ChipStackError' key words. The idea is to avoid using

# try/except ChipStackError blocks and replace them with 'with' syntax. In case you still want to

# use try/except you have to comment '# chipstack-ok' in the same line and justify it.

def is_exception_line(line):

return ("except ChipStackError" in line) and ("# chipstack-ok" not in line)

failures = []

for py_file in TARGET_DIR.rglob("*.py"):

with open(py_file, encoding="utf-8") as f:

lines = f.readlines()

for i, line in enumerate(lines):

if is_exception_line(line):

failures.append(f"{py_file}: {i+1}: 'except ChipStackError' without # chipstack-ok -> {line.strip()}")

if failures:

print("\n".join(failures))

sys.exit(1)

else:

print("No 'except ChipStackError' found in testing Python files")