47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
|
import subprocess
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
|
||
|
|
proto_file = "proto/document_parser.proto"
|
||
|
|
proto_path = "proto"
|
||
|
|
python_out = "proto"
|
||
|
|
grpc_python_out = "proto"
|
||
|
|
|
||
|
|
def generate_grpc():
|
||
|
|
"""Generate gRPC Python code from proto file"""
|
||
|
|
print(f"Generating gRPC code from {proto_file}...")
|
||
|
|
|
||
|
|
cmd = [
|
||
|
|
sys.executable,
|
||
|
|
"-m",
|
||
|
|
"grpc_tools.protoc",
|
||
|
|
f"--proto_path={proto_path}",
|
||
|
|
f"--python_out={python_out}",
|
||
|
|
f"--grpc_python_out={grpc_python_out}",
|
||
|
|
proto_file,
|
||
|
|
]
|
||
|
|
|
||
|
|
try:
|
||
|
|
subprocess.run(cmd, check=True)
|
||
|
|
print("gRPC code generated successfully!")
|
||
|
|
|
||
|
|
pb2_file = os.path.join(python_out, "document_parser_pb2.py")
|
||
|
|
pb2_grpc_file = os.path.join(python_out, "document_parser_pb2_grpc.py")
|
||
|
|
|
||
|
|
if os.path.exists(pb2_file) and os.path.exists(pb2_grpc_file):
|
||
|
|
print(f"Generated files:")
|
||
|
|
print(f" - {pb2_file}")
|
||
|
|
print(f" - {pb2_grpc_file}")
|
||
|
|
else:
|
||
|
|
print("Warning: Expected files not found")
|
||
|
|
|
||
|
|
except subprocess.CalledProcessError as e:
|
||
|
|
print(f"Error generating gRPC code: {e}")
|
||
|
|
sys.exit(1)
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Unexpected error: {e}")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
generate_grpc()
|