diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2021-03-18 15:26:05 -0400 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2021-03-18 16:13:10 -0400 |
| commit | ea844985d08ec6de49e82bb6a5878765bbeeb17b (patch) | |
| tree | f03a52681ef6098a1722ceb97e974cc19a2059fe /lib/sqlalchemy | |
| parent | 3fec5028e695ad138aa46a0ae66c55e8bcb653f6 (diff) | |
| download | sqlalchemy-ea844985d08ec6de49e82bb6a5878765bbeeb17b.tar.gz | |
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
Diffstat (limited to 'lib/sqlalchemy')
| -rw-r--r-- | lib/sqlalchemy/ext/mypy/decl_class.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 |