From 524971e05b36e34f71cd01f83031711e13d589a2 Mon Sep 17 00:00:00 2001
From: Hiroshi Toyama <hiroshi.toyama@dena.com>
Date: Wed, 6 Feb 2019 19:39:32 +0900
Subject: [PATCH] fix sls invoke local not works.

---
 unzip_requirements.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/unzip_requirements.py b/unzip_requirements.py
index 4e3b9e51..9ba3ad54 100644
--- a/unzip_requirements.py
+++ b/unzip_requirements.py
@@ -13,8 +13,9 @@
     if os.path.exists(tempdir):
         shutil.rmtree(tempdir)
 
-    zip_requirements = os.path.join(
-        os.environ.get('LAMBDA_TASK_ROOT', os.getcwd()), '.requirements.zip')
+    default_lambda_task_root = os.environ.get('LAMBDA_TASK_ROOT', os.getcwd())
+    lambda_task_root = os.getcwd() if os.environ.get('IS_LOCAL') == 'true' else default_lambda_task_root
+    zip_requirements = os.path.join(lambda_task_root, '.requirements.zip')
 
     zipfile.ZipFile(zip_requirements, 'r').extractall(tempdir)
     os.rename(tempdir, pkgdir)  # Atomic