From ea844985d08ec6de49e82bb6a5878765bbeeb17b Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Thu, 18 Mar 2021 15:26:05 -0400 Subject: check for MemberExpr looking for column argument Fixed issue in MyPy extension which crashed on detecting the type of a :class:`.Column` if the type were given with a module prefix like ``sa.Integer()``. Fixes: sqlalchemy/sqlalchemy2-stubs/#2 Change-Id: I71f53a6ced501ae144e28ce255cf3f50ea2b2e84 --- lib/sqlalchemy/ext/mypy/decl_class.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'lib') diff --git a/lib/sqlalchemy/ext/mypy/decl_class.py b/lib/sqlalchemy/ext/mypy/decl_class.py index f5215ca1c..7a0c251c3 100644 --- a/lib/sqlalchemy/ext/mypy/decl_class.py +++ b/lib/sqlalchemy/ext/mypy/decl_class.py @@ -693,7 +693,7 @@ def _infer_type_from_decl_column( # x = Column(String(50)) callee = column_arg.callee break - elif isinstance(column_arg, nodes.NameExpr): + elif isinstance(column_arg, (nodes.NameExpr, nodes.MemberExpr)): if isinstance(column_arg.node, TypeInfo): # x = Column(String) callee = column_arg -- cgit v1.2.1