我正在尝试创建一个自定义构建规则,以便从py_binary输出构建一个pip包。要创建我的pip包,我想调用一个shell脚本。 shell脚本通过从py_binary输出的runfiles创建一个zip文件来构建一个pip包。
例如假设我有
py_binary(
name = "some_binary",
srcs = ["some_binary.py"],
srcs_version = "PY2AND3",
)
构建此规则会产生
bazel-bin/some_binary.runfiles
我现在想创建一个自定义构建规则,它将使用位置bazel-bin / some_binary.runfiles
调用我的shell脚本我尝试创建一个宏
def build_pip_package(
name, py_binary=None, setup_file=None):
"""Create a pip package from a py_binary.
The source file should be a text file with python formatting i.e.
Args:
name: Name for the rule.
py_binary: Build rule producing the py_binary
setup_file: Build rule producing the setup.py file to use to produce
the package.
"""
output= "somefile"
native.genrule(
name=name,
outs=[output],
cmd="echo $(location //:build_pip_package_script) "
+ "--py_runfiles_path=$(locations %s)" % py_binary
+ " --setup_file=$(location %s) " % setup_file,
tools=["//:build_pip_package_script"],
srcs=[setup_file, py_binary])
最终用
调用我的shell脚本bazel-out/local-fastbuild/bin/some_binary/model_train some_binary.py
如何使用与some_binary目标对应的runfiles目录的位置调用我的脚本。
答案 0 :(得分:6)
如果将py_binary放在tools属性而不是srcs中,bazel将包含runfiles树。您可以通过"$(location %s).runfiles" % py_binary