2021-06-18 17:57:02 +02:00
|
|
|
#!/usr/bin/env python3
|
|
|
|
|
|
|
|
|
|
import os
|
2023-11-26 15:09:52 +01:00
|
|
|
from pathlib import Path
|
|
|
|
|
from subprocess import check_call
|
2021-06-18 17:57:02 +02:00
|
|
|
|
|
|
|
|
root_dir = Path(__file__).absolute().parent.parent
|
|
|
|
|
os.chdir(root_dir)
|
|
|
|
|
|
|
|
|
|
check_call([
|
2022-11-22 19:20:23 +01:00
|
|
|
"protoc", "--python_out=aioesphomeapi", "-I", "aioesphomeapi",
|
2021-06-18 17:57:02 +02:00
|
|
|
"aioesphomeapi/api.proto", "aioesphomeapi/api_options.proto"
|
|
|
|
|
])
|
|
|
|
|
|
|
|
|
|
# https://github.com/protocolbuffers/protobuf/issues/1491
|
|
|
|
|
api_file = root_dir / 'aioesphomeapi' / 'api_pb2.py'
|
|
|
|
|
content = api_file.read_text().replace(
|
|
|
|
|
"import api_options_pb2 as api__options__pb2",
|
|
|
|
|
"from . import api_options_pb2 as api__options__pb2"
|
|
|
|
|
)
|
|
|
|
|
api_file.write_text(content)
|
|
|
|
|
|
|
|
|
|
for fname in ['api_pb2.py', 'api_options_pb2.py']:
|
|
|
|
|
file = root_dir / 'aioesphomeapi' / fname
|
|
|
|
|
content = '# type: ignore\n' + file.read_text()
|
|
|
|
|
file.write_text(content)
|
