如何用单条导入语句兼容本地与AWS环境导入AWS Lambda层？

以下是几种无需修改导入语句、也不用提前执行SAM构建的实用方案，同时满足本地开发、CI检查和单元测试的需求：

方案1：动态添加本地模块路径（推荐） #

直接在Lambda处理器代码开头加入路径配置，让本地Python能识别层目录，同时不影响Lambda线上环境的运行逻辑。

• 在Lambda处理器文件的最顶部（所有导入语句之前）添加代码：

import sys
from pathlib import Path

# 根据你的实际目录结构调整根目录层级
# 示例：假设处理器文件在 src/lambda/handler.py，层目录在 src/libs/layers/
project_root = Path(__file__).parent.parent.parent
sys.path.append(str(project_root / "src/libs/layers"))

线上Lambda环境中，层会被自动挂载到Python默认搜索路径，这段代码不会产生任何影响；本地运行时，Python会把层目录加入搜索路径，from my_lib import my_func就能正常生效。

• CI环境适配：运行pylint前，通过环境变量指定模块路径即可消除E0401错误：

export PYTHONPATH="$PYTHONPATH:/path/to/your/project/src/libs/layers"
pylint src/

方案2：通过PYTHONPATH统一本地环境 #

无需修改代码，只在本地开发和CI环境中设置PYTHONPATH环境变量，让Python直接识别层目录为顶级模块。

• 本地开发时，在项目根目录执行：

# Linux/macOS
export PYTHONPATH="$PYTHONPATH:./src/libs/layers"

# Windows（PowerShell）
$env:PYTHONPATH += ";./src/libs/layers"

设置后，本地运行代码或单元测试时，from my_lib import my_func就能直接生效。

• CI环境中，将上述命令加入pylint执行的前置步骤即可，和方案1的CI配置一致。

方案3：通过环境变量动态适配导入路径 #

如果不想修改路径或环境变量，可以通过判断运行环境动态切换导入逻辑，代码侵入性稍强但无需额外配置：

import os

# 识别Lambda运行环境
if os.getenv("AWS_EXECUTION_ENV"):
    from my_lib import my_func
else:
    from src.libs.layers.my_lib import my_func

或者用importlib实现更灵活的导入：

import os
import importlib

lib_module = "my_lib" if os.getenv("AWS_EXECUTION_ENV") else "src.libs.layers.my_lib"
my_lib = importlib.import_module(lib_module)
my_func = my_lib.my_func

内容的提问来源于stack exchange，提问作者Matthew Phinney