"""Initialize the package and get dependencies.""" 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__) 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"%(prog)s {__version__}") core_parser.add_argument('--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") args, _ = core_parser.parse_known_args() for path in args.proto_files_path: sys.path.append(os.path.abspath(os.path.expanduser(path))) def get_config_pb2(): """Retrieve prebuilt _pb2 file or attempt to compile it live.""" try: return importlib.import_module('config_pb2') except ModuleNotFoundError: if args.proto_files_path: # compile the proto files in realtime, leave them in this package logger.info("Invoking gRPC tool to compile config.proto...") return grpc.protos('config.proto') raise