1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-16 03:18:09 +00:00
trezor-firmware/protob/check.py
2019-03-21 12:30:37 +01:00

30 lines
843 B
Python
Executable File

#!/usr/bin/env python3
from glob import glob
import os
import re
import sys
error = False
MYDIR = os.path.dirname(__file__)
EXPECTED_PREFIX_RE = re.compile(r"messages-(\w+)\.proto")
for fn in sorted(glob(os.path.join(MYDIR, "messages-*.proto"))):
with open(fn, "rt") as f:
prefix = EXPECTED_PREFIX_RE.search(fn).group(1).capitalize()
if prefix in ["Bitcoin", "Bootloader", "Common", "Crypto", "Management"]:
continue
if prefix == "Nem":
prefix = "NEM"
for line in f:
line = line.strip().split(" ")
if line[0] not in ["enum", "message"]:
continue
if not line[1].startswith(prefix) and not line[1].startswith("Debug" + prefix):
print("ERROR:", fn, line[1])
error = True
if error:
sys.exit(1)