py-protobuf package: fix import error for Python 2.x (#12873)

2019-09-20 17:52:11 -07:00 · 2019-09-20 17:52:11 -07:00 · af8f245f48
commit af8f245f48
parent 5397606b34
1 changed files with 8 additions and 0 deletions
--- a/var/spack/repos/builtin/packages/py-protobuf/package.py
+++ b/var/spack/repos/builtin/packages/py-protobuf/package.py
@ -42,3 +42,11 @@ def build_args(self, spec, prefix):
    def install_args(self, spec, prefix):
        args = super(PyProtobuf, self).install_args(spec, prefix)
        return args + ['--cpp_implementation']
+
+    @run_after('install')
+    def fix_import_error(self):
+        if str(self.spec['python'].version.up_to(1)) == '2':
+            touch = which('touch')
+            touch(self.prefix + '/' +
+                  self.spec['python'].package.site_packages_dir +
+                  '/google/__init__.py')