AWS Lambda Python3.9运行时exec执行代码丢空格报错求解决方案

问题核心 #

Lambda环境中传递给exec的字符串被意外去除空格，导致Python无法正确解析函数定义（比如def twoSum()变成deftwoSum()），最终抛出name 'twoSum' is not defined错误。

具体解决办法 #

• 先确认问题根源：在exec执行前，把要执行的字符串打印到CloudWatch日志，验证是否真的丢失了空格。示例代码：

  def lambda_handler(event, context):
      user_code = event.get('user_code', '')
      print("待执行代码内容:", repr(user_code))  # 用repr能看到所有字符，包括空格/换行
      exec(user_code, globals())
      # ...后续逻辑
  

  执行后去Lambda控制台的日志里查看输出，确认字符串是否被篡改。

• 强制修复代码格式：

  • 引入代码格式化库：使用autopep8或black这类工具自动修复语法空格问题。需要把依赖打包进Lambda部署包（因为Lambda默认没有这些库）。示例：

    import autopep8
    
    def lambda_handler(event, context):
        user_code = event['user_code']
        # 自动格式化代码，修复空格、缩进等语法问题
        formatted_code = autopep8.fix_code(user_code)
        exec(formatted_code, globals())
        return twoSum([2,7,11,15], 9)
    

  • 轻量正则修复（无依赖）：如果不想引入第三方库，用正则表达式修复最容易出问题的def关键字后的空格：

    import re
    
    def lambda_handler(event, context):
        user_code = event['user_code']
        # 给def后面的函数名补空格
        fixed_code = re.sub(r'def(\w+)', r'def \1', user_code)
        # 可选：修复其他语法位置的空格，比如赋值、运算符前后
        fixed_code = re.sub(r'([+\-*/=()])', r' \1 ', fixed_code).replace('  ', ' ')
        exec(fixed_code, globals())
        return twoSum([2,7,11,15], 9)
    

• 检查输入传递链路：如果用户代码是通过API Gateway、SQS等触发器传递的，确认传递过程中没有被压缩、转义或截断。比如API Gateway的请求体如果是JSON，要确保字符串字段没有被错误处理（比如前端传递时不要做minify压缩）。

• 替换exec的替代方案：如果exec的方式始终有问题，可以把用户函数转换成可调用对象。比如用types.FunctionType动态创建函数，但这种方式需要提取函数的代码逻辑，相对复杂，适合特定场景。

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