147 lines
2.9 KiB
Python
147 lines
2.9 KiB
Python
|
load("@arrdem_source_pypi//:requirements.bzl",
|
||
|
|
_py_requirement = "requirement"
|
||
|
|
)
|
||
|
|
|
||
|
|
load("@rules_python//python:defs.bzl",
|
||
|
|
_py_binary = "py_binary",
|
||
|
|
_py_test = "py_test",
|
||
|
|
_py_library = "py_library",
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def py_requirement(*args, **kwargs):
|
||
|
|
"""A re-export of requirement()"""
|
||
|
|
return _py_requirement(*args, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
def py_test(python_version=None, **kwargs):
|
||
|
|
"""A re-export of py_test()"""
|
||
|
|
|
||
|
|
if python_version and python_version != "PY3":
|
||
|
|
fail("py3k only!")
|
||
|
|
|
||
|
|
return _py_test(
|
||
|
|
python_version="PY3",
|
||
|
|
**kwargs,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def py_pytest(name, srcs, deps, main=None, python_version=None, args=None, **kwargs):
|
||
|
|
"""A py_test target which uses pytest."""
|
||
|
|
|
||
|
|
if python_version and python_version != "PY3":
|
||
|
|
fail("py3k only!")
|
||
|
|
|
||
|
|
f = "//tools/python:bzl_pytest_shim.py"
|
||
|
|
|
||
|
|
deps = [
|
||
|
|
py_requirement("pytest"),
|
||
|
|
py_requirement("jedi"),
|
||
|
|
py_requirement("pytest-pudb"),
|
||
|
|
] + deps
|
||
|
|
|
||
|
|
srcs = [f] + srcs
|
||
|
|
|
||
|
|
t = py_test(
|
||
|
|
name = name,
|
||
|
|
srcs = srcs,
|
||
|
|
main = f,
|
||
|
|
args = args,
|
||
|
|
python_version="PY3",
|
||
|
|
deps = deps,
|
||
|
|
**kwargs,
|
||
|
|
)
|
||
|
|
|
||
|
|
# FIXME (arrdem 2020-09-27):
|
||
|
|
# This really needs to be a py_image_test.
|
||
|
|
# Not clear how to achieve that.
|
||
|
|
# py_image(
|
||
|
|
# name = name + ".containerized",
|
||
|
|
# main = f,
|
||
|
|
# args = args,
|
||
|
|
# srcs = srcs,
|
||
|
|
# deps = deps,
|
||
|
|
# **kwargs,
|
||
|
|
# )
|
||
|
|
|
||
|
|
return t
|
||
|
|
|
||
|
|
|
||
|
|
def py_unittest(srcs=[], **kwargs):
|
||
|
|
"""A helper for running unittest tests"""
|
||
|
|
|
||
|
|
f = "//tools/python:bzl_unittest_shim.py"
|
||
|
|
return py_test(
|
||
|
|
main = f,
|
||
|
|
srcs = [f] + srcs,
|
||
|
|
**kwargs
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def py_binary(python_version=None, main=None, srcs=None, **kwargs):
|
||
|
|
"""A re-export of py_binary()"""
|
||
|
|
|
||
|
|
if python_version and python_version != "PY3":
|
||
|
|
fail("py3k only!")
|
||
|
|
|
||
|
|
srcs = srcs or []
|
||
|
|
if main not in srcs:
|
||
|
|
srcs = [main] + srcs
|
||
|
|
|
||
|
|
return _py_binary(
|
||
|
|
python_version = "PY3",
|
||
|
|
main = main,
|
||
|
|
srcs = srcs,
|
||
|
|
**kwargs,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def py_library(srcs_version=None, **kwargs):
|
||
|
|
"""A re-export of py_library()"""
|
||
|
|
|
||
|
|
if srcs_version and srcs_version != "PY3":
|
||
|
|
fail("py3k only!")
|
||
|
|
|
||
|
|
return _py_library(
|
||
|
|
srcs_version="PY3",
|
||
|
|
**kwargs
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
ResourceGroupInfo = provider(
|
||
|
|
fields = {
|
||
|
|
"srcs": "files to use from Python",
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def _resource_impl(ctx):
|
||
|
|
srcs = []
|
||
|
|
for target in ctx.attr.srcs:
|
||
|
|
srcs.extend(target.files.to_list())
|
||
|
|
transitive_srcs = depset(direct = srcs)
|
||
|
|
|
||
|
|
return [
|
||
|
|
ResourceGroupInfo(
|
||
|
|
srcs = ctx.attr.srcs,
|
||
|
|
),
|
||
|
|
PyInfo(
|
||
|
|
has_py2_only_sources = False,
|
||
|
|
has_py3_only_sources = True,
|
||
|
|
uses_shared_libraries = False,
|
||
|
|
transitive_sources = transitive_srcs,
|
||
|
|
),
|
||
|
|
]
|
||
|
|
|
||
|
|
py_resources = rule(
|
||
|
|
implementation = _resource_impl,
|
||
|
|
attrs = {
|
||
|
|
"srcs": attr.label_list(
|
||
|
|
allow_empty = True,
|
||
|
|
mandatory = True,
|
||
|
|
allow_files = True,
|
||
|
|
doc = "Files to hand through to Python",
|
||
|
|
),
|
||
|
|
},
|
||
|
|
)
|