Brian S. Stephan
c81f4cd139
this should all be tested now: 1. invoking against precompiled _pb2.py files provided by user 2. invoking against .proto files provided by user which must be compiled 3. invoking with a special option to use shipped (by us) .proto files which must be compiled 4. erroring because none of the above occurred this took some reorganization, but this should finally give me stability in using this in GP2040-CE's build process Signed-off-by: Brian S. Stephan <bss@incorporeal.org>
78 lines
3.4 KiB
Python
78 lines
3.4 KiB
Python
"""Initialize the package and get dependencies.
|
|
|
|
SPDX-FileCopyrightText: © 2023 Brian S. Stephan <bss@incorporeal.org>
|
|
SPDX-License-Identifier: GPL-3.0-or-later
|
|
"""
|
|
import argparse
|
|
import importlib
|
|
import logging
|
|
import os
|
|
import pathlib
|
|
import sys
|
|
|
|
import grpc
|
|
|
|
# dynamically generate version number
|
|
try:
|
|
# packaged/pip install -e . value
|
|
from ._version import version as __version__
|
|
except ImportError:
|
|
# local clone value
|
|
from setuptools_scm import get_version
|
|
__version__ = get_version(root='..', relative_to=__file__)
|
|
|
|
# configure basic logging and logger for this module
|
|
root = logging.getLogger()
|
|
root.setLevel(logging.DEBUG)
|
|
handler = logging.StreamHandler(sys.stderr)
|
|
formatter = logging.Formatter('%(asctime)s %(levelname)8s [%(name)s] %(message)s')
|
|
handler.setFormatter(formatter)
|
|
root.addHandler(handler)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# parse flags that are common to many tools (e.g. adding paths for finding .proto files)
|
|
core_parser = argparse.ArgumentParser(add_help=False)
|
|
core_parser.add_argument('-v', '--version', action='version',
|
|
version=f"gp2040ce-binary-tools {__version__} (Python {sys.version})",
|
|
help="show version information and exit")
|
|
core_parser.add_argument('-d', '--debug', action='store_true', help="enable debug logging")
|
|
core_parser.add_argument('-P', '--proto-files-path', type=pathlib.Path, default=list(), action='append',
|
|
help="path to .proto files to read, including dependencies; you will likely need "
|
|
"to supply this twice, once for GP2040-CE's .proto files and once for nanopb's")
|
|
core_parser.add_argument('-S', '--use-shipped-fallback', action='store_true',
|
|
help="utilize shipped (potentially stale) .proto files because you can't supply your own")
|
|
args, _ = core_parser.parse_known_args()
|
|
for path in args.proto_files_path:
|
|
sys.path.append(os.path.abspath(os.path.expanduser(path)))
|
|
|
|
if args.debug:
|
|
handler.setLevel(logging.DEBUG)
|
|
else:
|
|
handler.setLevel(logging.WARNING)
|
|
|
|
|
|
def get_config_pb2(with_fallback: bool = args.use_shipped_fallback):
|
|
"""Retrieve prebuilt _pb2 file or attempt to compile it live."""
|
|
# try to just import a precompiled module if we have been given it in our path
|
|
# (perhaps someone already compiled it for us for whatever reason)
|
|
try:
|
|
logger.debug("Trying precompiled protobuf modules...")
|
|
return importlib.import_module('config_pb2')
|
|
except ModuleNotFoundError:
|
|
# no found precompiled config, try to compile the proto files in realtime
|
|
# because it's possible someone put them on the path
|
|
try:
|
|
logger.debug("No precompiled protobuf modules found, invoking gRPC tool to compile config.proto...")
|
|
return grpc.protos('config.proto')
|
|
except (ModuleNotFoundError, TypeError):
|
|
# (TypeError could be the windows bug https://github.com/protocolbuffers/protobuf/issues/14345)
|
|
if not with_fallback:
|
|
raise
|
|
|
|
# that failed, import the snapshot (may be lagging what's in GP2040-CE)
|
|
logger.warning("using the fallback .proto files! please supply your files with -P if you can!")
|
|
sys.path.append(os.path.join(pathlib.Path(__file__).parent.resolve(), 'proto_snapshot'))
|
|
logger.debug("Invoking gRPC tool again to compile shipped config.proto...")
|
|
return grpc.protos('config.proto')
|