From ae8cba38bd650397e9bf2283dd47e07ee0f3a2e0 Mon Sep 17 00:00:00 2001
From: Vincent Turato <vincentturato@gmail.com>
Date: Mon, 9 Aug 2021 16:41:17 -0700
Subject: [PATCH] Fix index/handler logic if handler in subdirectory

---
 packages/@aws-cdk/aws-lambda-python/lib/function.ts       | 2 +-
 packages/@aws-cdk/aws-lambda-python/test/function.test.ts | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/packages/@aws-cdk/aws-lambda-python/lib/function.ts b/packages/@aws-cdk/aws-lambda-python/lib/function.ts
index 733c115c0383d..ce2ce953e269d 100644
--- a/packages/@aws-cdk/aws-lambda-python/lib/function.ts
+++ b/packages/@aws-cdk/aws-lambda-python/lib/function.ts
@@ -113,7 +113,7 @@ export class PythonFunction extends lambda.Function {
         assetHashType: props.assetHashType,
         assetHash: props.assetHash,
       }),
-      handler: `${index.slice(0, -3)}.${handler}`,
+      handler: `${index.replace('.py', '').replace('/', '.')}.${handler}`,
     });
   }
 }
diff --git a/packages/@aws-cdk/aws-lambda-python/test/function.test.ts b/packages/@aws-cdk/aws-lambda-python/test/function.test.ts
index a6bc4778e14f2..7cd57514e5984 100644
--- a/packages/@aws-cdk/aws-lambda-python/test/function.test.ts
+++ b/packages/@aws-cdk/aws-lambda-python/test/function.test.ts
@@ -70,7 +70,7 @@ test('PythonFunction with index in a subdirectory', () => {
   }));
 
   expect(stack).toHaveResource('AWS::Lambda::Function', {
-    Handler: 'inner/custom_index.custom_handler',
+    Handler: 'inner.custom_index.custom_handler',
   });
 });