pub/python-betterproto
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Supports running plugin.py standalone by reading from a dump-file, so its possible to debug it.

Browse Source
This commit is contained in:
boukeversteegh 2020-07-05 12:13:59 +02:00
parent 0af0cf4bfb
commit 98d00f0d21
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
betterproto/plugin.py
View File

@ -386,6 +386,10 @@ def main():
request = plugin.CodeGeneratorRequest()
request.ParseFromString(data)
dump_file = os.getenv("DUMP_FILE")
if dump_file:
dump_request(dump_file, request)
# Create response
response = plugin.CodeGeneratorResponse()
@ -399,5 +403,16 @@ def main():
sys.stdout.buffer.write(output)
def dump_request(dump_file: str, request: CodeGeneratorRequest):
"""
For developers: Supports running plugin.py standalone so its possible to debug it.
Run protoc (or generate.py) with DUMP_FILE="yourfile.bin" to write the request to a file.
Then run plugin.py from your IDE in debugging mode, and redirect stdin to the file.
"""
with open(str(dump_file), "wb") as fh:
sys.stderr.write(f"\033[31mWriting: {dump_file}\033[0m\n")
fh.write(request.SerializeToString())
if __name__ == "__main__":
main()