Skip to content
Snippets Groups Projects
generate-stubs 496 B
Newer Older
#!/bin/sh

set -eu

PROTOS_DIR=./protos
STUBS_DIR=hgitaly/stub

for PROTO in lint shared blob commit diff ref repository-service; do
    python -m grpc_tools.protoc \
      -I${PROTOS_DIR} \
      --python_out=${STUBS_DIR} \
      --grpc_python_out=${STUBS_DIR} \
      ${PROTOS_DIR}/${PROTO}.proto

    MODULE=$(echo $PROTO | sed 's/-/_/g')

    sed -E -i 's/^import ([[:alnum:]_]+_pb2)/from . import \1/g' \
      ${STUBS_DIR}/${MODULE}_pb2.py \
      ${STUBS_DIR}/${MODULE}_pb2_grpc.py

done