pip: entry_point: Add support for exit codes

bazelbuild · Oct 14, 2021 · bc30a71 · bc30a71
1 parent 562ffc7
commit bc30a71
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/pip_install/extract_wheels/lib/bazel.py b/python/pip_install/extract_wheels/lib/bazel.py
@@ -31,9 +31,10 @@ def generate_entry_point_contents(entry_point: str, shebang: str = "#!/usr/bin/e
     module, method = entry_point.split(":", 1)
     return textwrap.dedent("""\
         {shebang}
+        import sys
+        from {module} import {method}
         if __name__ == "__main__":
-            from {module} import {method}
-            {method}()
+            sys.exit({method}())
         """.format(
         shebang=shebang,
         module=module,