feat(python!): add support for v2 layer

Changes to v1 layer:
* remove python 3.6 support
* aggregated all extra dependencies in "extras" group
* removed unecessary files from Lambda Layer to reduce size

Author: rubenfonseca

layer/Python/Dockerfile

@@ -1,19 +1,24 @@
 FROM public.ecr.aws/lambda/python:3.8
 
-
-
 ARG PACKAGE_SUFFIX=''
 
 USER root
 WORKDIR /tmp
 
-
-# PACKAGE_SUFFIX = '[pydantic]==1.23.0'
-# PACKAGE_SUFFIX = '[pydantic]'
+# PACKAGE_SUFFIX = '[extras]==1.23.0'
+# PACKAGE_SUFFIX = '[extras]'
 # PACKAGE_SUFFIX = '=='1.23.0'
 # PACKAGE_SUFFIX = ''
 
-
 RUN yum update -y && yum install -y zip unzip wget tar gzip
 
-RUN pip install -t /asset/python aws-lambda-powertools$PACKAGE_SUFFIX
+RUN pip install -t /asset/python aws-lambda-powertools$PACKAGE_SUFFIX
+
+# Removing nonessential files
+RUN cd /asset && \
+  rm -rf python/boto* && \
+  find python -name '*.so' -type f -exec strip "{}" \; && \
+  find python -wholename "*/tests/*" -type f -delete && \
+  find python -regex '^.*\(__pycache__\|\.py[co]\)$' -delete && \
+  zip -r9 aws-lambda-powertools.zip ./python && \
+  rm -rf ./python

src/lambda-powertools-layer.ts

@@ -10,8 +10,9 @@ export interface PowertoolsLayerProps {
    * The powertools package version from pypi repository.
    */
   readonly version?: string;
+
   /**
-   * A flag for the pydantic extras dependency, used for parsing.
+   * A flag for the extras dependencies (pydantic, aws-xray-sdk, etc.)
    * This will increase the size of the layer significantly. If you don't use parsing, ignore it.
    */
   readonly includeExtras?: boolean;
@@ -36,7 +37,7 @@ export class LambdaPowertoolsLayer extends lambda.LayerVersion {
    * There are multiple combinations between version and extras package that results in different suffix for the installation.
    * With and without version, with and without extras flag.
    * We construct one suffix here because it is easier to do in code than inside the Dockerfile with bash commands.
-   * For example, if we set extras=true and version=1.22.0 we get '[pydantic]==1.22.0'.
+   * For example, if we set extras=true and version=1.22.0 we get '[extras]==1.22.0'.
    *
    */
   static constructBuildArgs(
@@ -48,7 +49,7 @@ export class LambdaPowertoolsLayer extends lambda.LayerVersion {
     switch (runtimeFamily) {
       case lambda.RuntimeFamily.PYTHON:
         if (includeExtras) {
-          suffix = '[pydantic]';
+          suffix = '[extras]';
         }
         if (version) {
           suffix = `${suffix}==${version}`;
@@ -84,7 +85,7 @@ export class LambdaPowertoolsLayer extends lambda.LayerVersion {
       license: 'MIT-0',
       compatibleRuntimes: getRuntimesFromRuntimeFamily(runtimeFamily),
       description: `Lambda Powertools for ${languageName}${
-        props?.includeExtras ? ' with Pydantic' : ''
+        props?.includeExtras ? ' with Extras' : ''
       } ${props?.version ? `version ${props?.version}` : 'latest version'}`.trim(),
     });
   }
@@ -94,7 +95,6 @@ function getRuntimesFromRuntimeFamily(runtimeFamily: lambda.RuntimeFamily): lamb
   switch (runtimeFamily) {
     case lambda.RuntimeFamily.PYTHON:
       return [
-        lambda.Runtime.PYTHON_3_6,
         lambda.Runtime.PYTHON_3_7,
         lambda.Runtime.PYTHON_3_8,
         lambda.Runtime.PYTHON_3_9,

test/lambda-powertools-python-layer.test.ts

@@ -23,7 +23,6 @@ describe('with no configuration the construct', () => {
   test('matches the python 3.x runtimes', () => {
     template.hasResourceProperties('AWS::Lambda::LayerVersion', {
       CompatibleRuntimes: [
-        'python3.6',
         'python3.7',
         'python3.8',
         'python3.9',
@@ -73,34 +72,34 @@ describe('with version configuration the construct', () => {
     });
 
     Template.fromStack(stack).hasResourceProperties('AWS::Lambda::LayerVersion', {
-      Description: 'Lambda Powertools for Python with Pydantic version 1.22.0',
+      Description: 'Lambda Powertools for Python with Extras version 1.22.0',
     });
 
   });
 
-  test('synthesizes with pyndatic and latest version', () => {
+  test('synthesizes with extras and latest version', () => {
     const stack = new Stack();
     new LambdaPowertoolsLayer(stack, 'LayerExtrasNoVersion', {
       includeExtras: true,
     });
 
     Template.fromStack(stack).hasResourceProperties('AWS::Lambda::LayerVersion', {
-      Description: 'Lambda Powertools for Python with Pydantic latest version',
+      Description: 'Lambda Powertools for Python with Extras latest version',
     });
   });
 });
 
 describe('construct build args for Dockerfile', () => {
-  test('returns pydantic and version', () => {
+  test('returns extras and version', () => {
     const args = LambdaPowertoolsLayer.constructBuildArgs(RuntimeFamily.PYTHON, true, '1.21.0');
 
-    expect(args).toEqual('[pydantic]==1.21.0');
+    expect(args).toEqual('[extras]==1.21.0');
   });
 
-  test('returns only pydantic when no version provided', () => {
+  test('returns only extras when no version provided', () => {
     const args = LambdaPowertoolsLayer.constructBuildArgs(RuntimeFamily.PYTHON, true, undefined);
 
-    expect(args).toEqual('[pydantic]');
+    expect(args).toEqual('[extras]');
   });
 
   test('returns only version when no extras flag provided', () => {