diff options
Diffstat (limited to 'generator')
| -rwxr-xr-x | generator/nanopb_generator.py | 23 | ||||
| -rw-r--r-- | generator/proto/Makefile | 9 |
2 files changed, 13 insertions, 19 deletions
diff --git a/generator/nanopb_generator.py b/generator/nanopb_generator.py index 0002409b..86c554a2 100755 --- a/generator/nanopb_generator.py +++ b/generator/nanopb_generator.py @@ -4,7 +4,7 @@ nanopb_version = "nanopb-0.2.5-dev" try: - import google.protobuf.descriptor_pb2 as descriptor + import google, distutils.util # bbfreeze seems to need these import google.protobuf.text_format as text_format except: print @@ -15,18 +15,8 @@ except: print raise -try: - import proto.nanopb_pb2 as nanopb_pb2 -except: - print - print "***************************************************************" - print "*** Could not import the precompiled nanopb_pb2.py. ***" - print "*** Run 'make' in the 'generator' folder to update the file.***" - print "***************************************************************" - print - raise - - +import proto.nanopb_pb2 as nanopb_pb2 +import proto.descriptor_pb2 as descriptor @@ -1025,6 +1015,13 @@ def main_cli(): def main_plugin(): '''Main function when invoked as a protoc plugin.''' + import sys + if sys.platform == "win32": + import os, msvcrt + # Set stdin and stdout to binary mode + msvcrt.setmode(sys.stdin.fileno(), os.O_BINARY) + msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY) + import proto.plugin_pb2 as plugin_pb2 data = sys.stdin.read() request = plugin_pb2.CodeGeneratorRequest.FromString(data) diff --git a/generator/proto/Makefile b/generator/proto/Makefile index fea99a9c..ee1390a5 100644 --- a/generator/proto/Makefile +++ b/generator/proto/Makefile @@ -1,7 +1,4 @@ -all: nanopb_pb2.py plugin_pb2.py +all: nanopb_pb2.py plugin_pb2.py descriptor_pb2.py -nanopb_pb2.py: nanopb.proto - protoc --python_out=. nanopb.proto - -plugin_pb2.py: plugin.proto - protoc --python_out=. plugin.proto +%_pb2.py: %.proto + protoc --python_out=. $< |