feat: Add uv support for python packaging (#723)

Co-authored-by: Anton Babenko <[email protected]>
Co-authored-by: Anton Babenko <[email protected]>

Author: rivhar

examples/build-package/README.md

@@ -5,6 +5,7 @@ Configuration in this directory creates deployment packages in a variety of comb
 This example demonstrates various packaging scenarios including:
 - Python packages with pip requirements
 - Poetry-based Python packages
+- UV-based Python packages
 - Node.js packages with npm
 - Docker-based builds
 - Quiet packaging - suppressing Poetry/pip/npm output during builds using `quiet_archive_local_exec = true`
@@ -46,12 +47,16 @@ Note that this example may create resources which cost money. Run `terraform des
 | <a name="module_lambda_layer"></a> [lambda\_layer](#module\_lambda\_layer) | ../../ | n/a |
 | <a name="module_lambda_layer_pip_requirements"></a> [lambda\_layer\_pip\_requirements](#module\_lambda\_layer\_pip\_requirements) | ../.. | n/a |
 | <a name="module_lambda_layer_poetry"></a> [lambda\_layer\_poetry](#module\_lambda\_layer\_poetry) | ../../ | n/a |
+| <a name="module_lambda_layer_uv"></a> [lambda\_layer\_uv](#module\_lambda\_layer\_uv) | ../../ | n/a |
 | <a name="module_npm_package_with_commands_and_patterns"></a> [npm\_package\_with\_commands\_and\_patterns](#module\_npm\_package\_with\_commands\_and\_patterns) | ../../ | n/a |
 | <a name="module_package_dir"></a> [package\_dir](#module\_package\_dir) | ../../ | n/a |
 | <a name="module_package_dir_pip_dir"></a> [package\_dir\_pip\_dir](#module\_package\_dir\_pip\_dir) | ../../ | n/a |
 | <a name="module_package_dir_poetry"></a> [package\_dir\_poetry](#module\_package\_dir\_poetry) | ../../ | n/a |
 | <a name="module_package_dir_poetry_no_docker"></a> [package\_dir\_poetry\_no\_docker](#module\_package\_dir\_poetry\_no\_docker) | ../../ | n/a |
 | <a name="module_package_dir_poetry_quiet"></a> [package\_dir\_poetry\_quiet](#module\_package\_dir\_poetry\_quiet) | ../../ | n/a |
+| <a name="module_package_dir_uv"></a> [package\_dir\_uv](#module\_package\_dir\_uv) | ../../ | n/a |
+| <a name="module_package_dir_uv_no_docker"></a> [package\_dir\_uv\_no\_docker](#module\_package\_dir\_uv\_no\_docker) | ../../ | n/a |
+| <a name="module_package_dir_uv_no_lock"></a> [package\_dir\_uv\_no\_lock](#module\_package\_dir\_uv\_no\_lock) | ../../ | n/a |
 | <a name="module_package_dir_with_npm_install"></a> [package\_dir\_with\_npm\_install](#module\_package\_dir\_with\_npm\_install) | ../../ | n/a |
 | <a name="module_package_dir_with_npm_install_lock_file"></a> [package\_dir\_with\_npm\_install\_lock\_file](#module\_package\_dir\_with\_npm\_install\_lock\_file) | ../../ | n/a |
 | <a name="module_package_dir_without_npm_install"></a> [package\_dir\_without\_npm\_install](#module\_package\_dir\_without\_npm\_install) | ../../ | n/a |

examples/build-package/main.tf

@@ -137,6 +137,84 @@ module "package_dir_poetry_quiet" {
   quiet_archive_local_exec = true # Suppress Poetry/pip output during packaging
 }
 
+# Create zip-archive of a single directory where "uv export" & "pip install" will be executed
+module "package_dir_uv_no_lock" {
+  source = "../../"
+
+  create_function = false
+  runtime         = "python3.12"
+
+  source_path = [
+    {
+      path       = "${path.module}/../fixtures/python-app-uv-no-lock"
+      uv_install = true
+    }
+  ]
+
+  artifacts_dir = "${path.root}/builds/package_dir_uv_no_lock/"
+}
+
+# Create zip-archive of a single directory where "uv export" & "pip install" will be executed (using docker)
+module "package_dir_uv" {
+  source = "../../"
+
+  create_function = false
+
+  build_in_docker = true
+  runtime         = "python3.12"
+  docker_image    = "build-python-uv"
+  docker_file     = "${path.module}/../fixtures/python-app-uv/docker/Dockerfile"
+
+  source_path = [
+    {
+      path       = "${path.module}/../fixtures/python-app-uv"
+      uv_install = true
+    }
+  ]
+  artifacts_dir = "${path.root}/builds/package_dir_uv/"
+}
+
+# Create zip-archive of a single directory where "uv export" & "pip install" will be executed (not using docker)
+module "package_dir_uv_no_docker" {
+  source = "../../"
+
+  create_function = false
+
+  runtime = "python3.12"
+
+  source_path = [
+    {
+      path       = "${path.module}/../fixtures/python-app-uv"
+      uv_install = true
+    }
+  ]
+  artifacts_dir = "${path.root}/builds/package_dir_uv_no_docker/"
+}
+
+# Create a Lambda Layer using uv for dependency management (using docker)
+module "lambda_layer_uv" {
+  source = "../../"
+
+  create_layer        = true
+  layer_name          = "${random_pet.this.id}-layer-uv"
+  compatible_runtimes = ["python3.12"]
+  runtime             = "python3.12"
+
+  build_in_docker = true
+  docker_image    = "build-python-uv"
+  docker_file     = "${path.module}/../fixtures/python-app-uv/docker/Dockerfile"
+
+  source_path = [
+    {
+      path          = "${path.module}/../fixtures/python-app-uv"
+      uv_install    = true
+      prefix_in_zip = "python"
+    }
+  ]
+  hash_extra    = "uv-extra-hash-to-prevent-conflicts-with-module.package_dir"
+  artifacts_dir = "${path.root}/builds/lambda_layer_uv/"
+}
+
 # Create zip-archive of a single directory without running "pip install" (which is default for python runtime)
 module "package_dir_without_pip_install" {
   source = "../../"

examples/fixtures/python-app-uv-no-lock/pyproject.toml

@@ -0,0 +1,11 @@
+[project]
+name = "uv_lambda"
+version = "0.1.0"
+description = "Fixture project to test uv without uv.lock"
+requires-python = ">=3.12"
+
+dependencies = [
+    "requests>=2.31.0"
+]
+
+[tool.uv]

examples/fixtures/python-app-uv-no-lock/uv_lambda/index.py

@@ -0,0 +1,2 @@
+def lambda_handler(event, context):
+    return {"statusCode": 200, "body": "Hello from uv (no lock)!"}

examples/fixtures/python-app-uv/docker/Dockerfile

@@ -0,0 +1,7 @@
+FROM public.ecr.aws/sam/build-python3.12
+
+LABEL description="AWS Lambda build container with uv"
+
+RUN pip install uv==0.9.21
+
+WORKDIR /var/task

examples/fixtures/python-app-uv/pyproject.toml

@@ -0,0 +1,18 @@
+[project]
+name = "uv_lambda"
+version = "0.1.0"
+requires-python = ">=3.12"
+dependencies = [
+    "requests==2.31.0",
+]
+
+[tool.uv]
+# This marker is for our Terraform parser to detect UV
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["uv_lambda"]
+include = ["uv_lambda/*"]