setup.py: Add grpc_tools proto include path

Change-Id: I577f6fbd7ea4ec204927baed7ca20570aaab8e0d

1 files changed, 5 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 47618ca..8e54d32 100755
--- a/setup.py
+++ b/setup.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 from setuptools import setup, Command
 from setuptools.command.build_py import build_py
+import pkg_resources
 import os
 
 package_directory = os.path.dirname(os.path.realpath(__file__))
@@ -22,9 +23,12 @@ class BuildGrpc(Command):
     def run(self):
         from grpc_tools import protoc
 
+        proto_include = pkg_resources.resource_filename('grpc_tools', '_proto')
+
         protoc.main([
             'grpc_tools.protoc',
-            '-I=proto',
+            '-Iproto',
+            f'-I{proto_include}',
             '--python_out=.',
             '--custom_grpc_out=.',
             'blueberry/host.proto',