bazel: add py_pex_binary support from aspect_rules_py

Author: AlexTereshenkov

BUILD.bazel

@@ -1,12 +1,19 @@
 load("@bazel_skylib//rules:common_settings.bzl", "string_flag")
+load("@buildifier_prebuilt//:rules.bzl", "buildifier")
 load("@rules_python//python:py_binary.bzl", "py_binary")
 
+buildifier(
+    name = "buildifier",
+    lint_mode = "fix",
+    mode = "fix",
+)
+
 # use default Python toolchain
 py_binary(
     name = "main",
     srcs = ["main.py"],
     deps = [
-      "@pypi//loguru",
+        "@pypi//loguru",
     ],
 )
 
@@ -15,37 +22,38 @@ py_binary(
     name = "main-3_10",
     srcs = ["main.py"],
     main = "main.py",
+    python_version = "3.10",
     deps = [
-      "@pypi//loguru",
+        "@pypi//loguru",
     ],
-    python_version="3.10"
 )
 
 # use particular Python toolchain (3.11)
 py_binary(
     name = "main-3_11",
-    main = "main.py",
     srcs = ["main.py"],
+    main = "main.py",
+    python_version = "3.11",
     deps = [
-      "@pypi//loguru",
+        "@pypi//loguru",
     ],
-    python_version="3.11"
 )
 
 genrule(
     name = "python-helper",
     outs = ["data.json"],
-    toolchains=["@rules_python//python:current_py_toolchain"],
     # Path to Python interpreter via a MAKE variable
     # https://rules-python.readthedocs.io/en/latest/toolchains.html#toolchain-usage-in-other-rules
     cmd = "$(PYTHON3) -c \"import json; print(json.dumps({'a': 'b'}, indent=2))\" > $@",
+    toolchains = ["@rules_python//python:current_py_toolchain"],
 )
 
 # flag to request using a local Python interpreter
 config_setting(
     name = "is_py_local",
     flag_values = {":py": "local-3_12"},
 )
+
 string_flag(
     name = "py",
     build_setting_default = "",

MODULE.bazel

@@ -1,12 +1,18 @@
+bazel_dep(name = "buildifier_prebuilt", version = "8.0.3", dev_dependency = True)
+
 bazel_dep(name = "bazel_skylib", version = "1.7.1")
 bazel_dep(name = "rules_python", version = "1.5.0-rc4")
+bazel_dep(name = "aspect_rules_py", version = "1.6.0")
 
 python = use_extension("@rules_python//python/extensions:python.bzl", "python")
 python.defaults(python_version = "3.11")
 
 # register multiple toolchains with different Python versions
 python.toolchain(python_version = "3.10")
-python.toolchain(python_version = "3.11", configure_coverage_tool = True)
+python.toolchain(
+    configure_coverage_tool = True,
+    python_version = "3.11",
+)
 
 pip = use_extension("@rules_python//python/extensions:pip.bzl", "pip")