From ccfec76b829e7eccd9e08d7f757da3590497f197 Mon Sep 17 00:00:00 2001 From: Cyril Chapellier Date: Thu, 30 Jun 2022 14:59:38 +0200 Subject: Support lambda expression in mypy plugin Avoid `error: INTERNAL ERROR` when the default is a lambda Fixes: #8196 Change-Id: I7346c693519b024c56156db6f4ffc9a45bb748d3 --- lib/sqlalchemy/ext/mypy/infer.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'lib/sqlalchemy') diff --git a/lib/sqlalchemy/ext/mypy/infer.py b/lib/sqlalchemy/ext/mypy/infer.py index 08035d74c..479be40c3 100644 --- a/lib/sqlalchemy/ext/mypy/infer.py +++ b/lib/sqlalchemy/ext/mypy/infer.py @@ -16,6 +16,7 @@ from mypy.nodes import AssignmentStmt from mypy.nodes import CallExpr from mypy.nodes import Expression from mypy.nodes import FuncDef +from mypy.nodes import LambdaExpr from mypy.nodes import MemberExpr from mypy.nodes import NameExpr from mypy.nodes import RefExpr @@ -423,6 +424,10 @@ def _infer_type_from_decl_column( elif isinstance(column_arg, (StrExpr,)): # x = Column("name", String), go to next argument continue + elif isinstance(column_arg, (LambdaExpr,)): + # x = Column("name", String, default=lambda: uuid.uuid4()) + # go to next argument + continue else: assert False -- cgit v1.2.1