diff --git a/setup.py b/setup.py index d6313ed..d869f6a 100644 --- a/setup.py +++ b/setup.py @@ -4,8 +4,9 @@ from setuptools.command.build_py import build_py as _build import os.path import subprocess +import shutil -PROTOC_BIN = "/usr/bin/protoc" +PROTOC_EXEC = "protoc" CURRENT_DIR = os.path.abspath( os.path.dirname( __file__ ) ) @@ -13,11 +14,12 @@ class ProtobufBuilder(_build): def run(self): # check if protobuf is installed - if not os.path.isfile(PROTOC_BIN): + exec_path = shutil.which(PROTOC_EXEC) + if exec_path is None: raise Exception("You should install protobuf compiler") print("Building protobuf file") - subprocess.run([PROTOC_BIN, + subprocess.run([exec_path, "--proto_path=" + CURRENT_DIR, "--python_out=" + CURRENT_DIR + "/gpapi/", CURRENT_DIR + "/googleplay.proto"])