ALSO generate .pyi stubs, because I guess you can do that

Author: ryapric

protobuf-grpc/clients/client.py

@@ -16,16 +16,13 @@
 import pb.employees.v1.employees_pb2 as employees_pb2
 import pb.employees.v1.employees_pb2_grpc as employees_pb2_grpc
 
+# Note that if you did NOT specify --pyi_out in the protoc call, you won't get
+# editor help like tab completion for the generated code -- everything comes out
+# as a metaclass unless you also generate those .pyi files
 if __name__ == "__main__":
     with grpc.insecure_channel("127.0.0.1:8080") as channel:
         echo_stub = echo_pb2_grpc.EchoServiceStub(channel)
 
-        # This is pretty infuriating -- Python protobuf code isn't generated
-        # directly, it generates a *Python* generator to generate the code at
-        # runtime (in *_pb2.py). As such, you don't get completion hints in
-        # editors because there's no data types to autofill etc. So, look at
-        # your proto file and determine how you should be setting up your
-        # requests!
         echo_request = echo_pb2.EchoRequest(msg = "Hello, gRPC!")
         echo_response = echo_stub.Echo(echo_request)
         print(echo_response)
@@ -35,6 +32,9 @@
         list_response = employees_stub.ListEmployees(list_request)
         print(list_response)
 
-        get_request = employees_pb2.GetEmployeeRequest(short_name = sys.argv[1])
-        get_response = employees_stub.GetEmployee(get_request)
-        print(get_response)
+        if len(sys.argv) < 2:
+            print("warning: not calling GetEmployee() because you must provide an employee's short name on the CLI. Did you look at the output of the ListEmployees() call?")
+        else:
+            get_request = employees_pb2.GetEmployeeRequest(short_name = sys.argv[1])
+            get_response = employees_stub.GetEmployee(get_request)
+            print(get_response)

protobuf-grpc/pb/echo/v1/echo_pb2.pyi

@@ -0,0 +1,17 @@
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Optional as _Optional
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class EchoRequest(_message.Message):
+    __slots__ = ["msg"]
+    MSG_FIELD_NUMBER: _ClassVar[int]
+    msg: str
+    def __init__(self, msg: _Optional[str] = ...) -> None: ...
+
+class EchoResponse(_message.Message):
+    __slots__ = ["msg"]
+    MSG_FIELD_NUMBER: _ClassVar[int]
+    msg: str
+    def __init__(self, msg: _Optional[str] = ...) -> None: ...

protobuf-grpc/pb/employees/v1/employees_pb2.pyi

@@ -0,0 +1,39 @@
+from google.api import annotations_pb2 as _annotations_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class Employee(_message.Message):
+    __slots__ = ["birthday", "full_name", "id"]
+    BIRTHDAY_FIELD_NUMBER: _ClassVar[int]
+    FULL_NAME_FIELD_NUMBER: _ClassVar[int]
+    ID_FIELD_NUMBER: _ClassVar[int]
+    birthday: str
+    full_name: str
+    id: int
+    def __init__(self, id: _Optional[int] = ..., full_name: _Optional[str] = ..., birthday: _Optional[str] = ...) -> None: ...
+
+class GetEmployeeRequest(_message.Message):
+    __slots__ = ["short_name"]
+    SHORT_NAME_FIELD_NUMBER: _ClassVar[int]
+    short_name: str
+    def __init__(self, short_name: _Optional[str] = ...) -> None: ...
+
+class GetEmployeeResponse(_message.Message):
+    __slots__ = ["employee"]
+    EMPLOYEE_FIELD_NUMBER: _ClassVar[int]
+    employee: Employee
+    def __init__(self, employee: _Optional[_Union[Employee, _Mapping]] = ...) -> None: ...
+
+class ListEmployeesRequest(_message.Message):
+    __slots__ = []
+    def __init__(self) -> None: ...
+
+class ListEmployeesResponse(_message.Message):
+    __slots__ = ["short_names"]
+    SHORT_NAMES_FIELD_NUMBER: _ClassVar[int]
+    short_names: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, short_names: _Optional[_Iterable[str]] = ...) -> None: ...

protobuf-grpc/proto/Makefile

@@ -48,6 +48,7 @@ generate: ensure-outdir
 		--openapiv2_opt=logtostderr=true \
     --openapiv2_opt=generate_unbound_methods=true \
 		--python_out=$(outdir) \
+		--pyi_out=$(outdir) \
 		--descriptor_set_out=./protoset \
 		--include_imports \
 		$(proto_files)