Merge pull request #3 from parknow/feature/nodeps-pip-parameter

--no-deps pip parameter

Author: rojopolis

README.md

@@ -16,8 +16,10 @@ change.
 ```
 module "python_lambda_archive" {
     source = "rojopolis/lambda-python-archive/aws"
-    src_dir = "${path.module}/python"
-    output_path = "${path.module}/lambda.zip"
+
+    src_dir              = "${path.module}/python"
+    output_path          = "${path.module}/lambda.zip"
+    install_dependencies = false
 }
 
 resource "aws_iam_role" "iam_for_lambda" {

main.tf

@@ -1,7 +1,8 @@
 data "external" "lambda_archive" {
     program = ["python3", "${path.module}/scripts/build_lambda.py"]
     query = {
-        src_dir     = "${var.src_dir}"
-        output_path = "${var.output_path}"
+        src_dir              = var.src_dir
+        output_path          = var.output_path
+        install_dependencies = var.install_dependencies
     }
 }

outputs.tf

@@ -1,9 +1,9 @@
 output "archive_path" {
     description = "Path of the archive file."
-    value       = "${data.external.lambda_archive.result.archive}"
+    value       = data.external.lambda_archive.result.archive
 }
 
 output "source_code_hash" {
     description = "Base64 encoded SHA256 hash of the archive file."
-    value       = "${data.external.lambda_archive.result.base64sha256}"
+    value       = data.external.lambda_archive.result.base64sha256
 }

scripts/build_lambda.py

@@ -13,7 +13,7 @@
 import tempfile
 import zipfile
 
-def build(src_dir, output_path):
+def build(src_dir, output_path, install_dependencies):
     with tempfile.TemporaryDirectory() as build_dir:
         copy_tree(src_dir, build_dir)
         if os.path.exists(os.path.join(src_dir, 'requirements.txt')):
@@ -24,7 +24,8 @@ def build(src_dir, output_path):
                  'install',
                  '--ignore-installed',
                  '--target', build_dir,
-                 '-r', os.path.join(build_dir, 'requirements.txt')],
+                 '-r', os.path.join(build_dir, 'requirements.txt'),
+                 *(['--no-deps'] if install_dependencies == 'false' else [])],
                  check=True,
                  stdout=subprocess.DEVNULL,
             )
@@ -71,5 +72,5 @@ def get_hash(output_path):
     logging.basicConfig(level='DEBUG')
     query = json.loads(sys.stdin.read())
     logging.debug(query)
-    archive = build(query['src_dir'], query['output_path'])
-    print(json.dumps({'archive': archive, "base64sha256":get_hash(archive)}))
+    archive = build(query['src_dir'], query['output_path'], query['install_dependencies'])
+    print(json.dumps({'archive': archive, 'base64sha256':get_hash(archive)}))

variables.tf

@@ -1,9 +1,15 @@
 variable "src_dir" {
     description = "Path to root of Python source to package."
-    type        = "string"
+    type        = string
 }
 
 variable "output_path" {
     description = "The output of the archive file."
-    type        = "string"
+    type        = string
+}
+
+variable "install_dependencies" {
+    description = "Whether to install pip dependecies"
+    type        = bool
+    default     = true
 }