1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
import sys
from google.protobuf import text_format
from foo import bar_pb2
from com_envoyproxy_protoc_gen_validate.python.protoc_gen_validate import validator
from typing import List
def main(filenames: List[str]) -> None:
if not filenames:
print("No inputs provided; exiting")
success_count = 0
for filename in filenames:
bars = bar_pb2.Bars()
fullname = bars.DESCRIPTOR.name
try:
with open(filename, 'r') as fh:
text_format.Parse(fh.read(), bars)
except IOError as error:
msg = f"Failed to open file '{filename}': {error}"
raise RuntimeError(msg) from error
except text_format.ParseError as error:
msg = f"Failed to parse file '{filename}' as a {fullname} textproto: {error}"
raise RuntimeError(msg) from error
try:
validator.validate(bars)
except validator.ValidationFailed as error:
print(
f"Failed to validate file '{filename}' as a {fullname} textproto: {error}",
file=sys.stderr,
)
else:
print(f"Successfully validated file '{filename}' as a {fullname} textproto")
success_count += 1
failure_count = len(filenames) - success_count
if failure_count:
s = "s" if failure_count > 1 else ""
msg = f"Failed to validate {failure_count} file{s}"
raise RuntimeError(msg)
if __name__ == "__main__":
try:
main(sys.argv[1:])
except RuntimeError as error:
print(error, file=sys.stderr)
sys.exit(1)
|
