diff options
| author | mike bayer <mike_mp@zzzcomputing.com> | 2021-04-12 19:38:13 +0000 |
|---|---|---|
| committer | Gerrit Code Review <gerrit@ci3.zzzcomputing.com> | 2021-04-12 19:38:13 +0000 |
| commit | b7facab8936addf022bca7a3dbbff96ce7c85227 (patch) | |
| tree | 87c6aa45835c4f49f2085723c11f94002af8844f /lib/sqlalchemy/ext/mypy/util.py | |
| parent | 8ad68e72c6d88abbddc77546e12ff6170d3dc221 (diff) | |
| parent | fb11e2e2e99c16263228df2d814b3bc8305e8e5e (diff) | |
| download | sqlalchemy-b7facab8936addf022bca7a3dbbff96ce7c85227.tar.gz | |
Merge "Update mypy plugin to only use public plugin API"
Diffstat (limited to 'lib/sqlalchemy/ext/mypy/util.py')
| -rw-r--r-- | lib/sqlalchemy/ext/mypy/util.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/lib/sqlalchemy/ext/mypy/util.py b/lib/sqlalchemy/ext/mypy/util.py index becce3ebe..1c1e56d2c 100644 --- a/lib/sqlalchemy/ext/mypy/util.py +++ b/lib/sqlalchemy/ext/mypy/util.py @@ -11,6 +11,7 @@ from mypy.nodes import JsonDict from mypy.nodes import NameExpr from mypy.nodes import SymbolTableNode from mypy.nodes import TypeInfo +from mypy.plugin import ClassDefContext from mypy.plugin import SemanticAnalyzerPluginInterface from mypy.plugins.common import deserialize_and_fixup_type from mypy.types import Instance @@ -68,7 +69,7 @@ def fail(api: SemanticAnalyzerPluginInterface, msg: str, ctx: Context): def add_global( - ctx: SemanticAnalyzerPluginInterface, + ctx: ClassDefContext, module: str, symbol_name: str, asname: str, @@ -127,7 +128,7 @@ def _unbound_to_instance( ), ) - node = api.lookup(typ.name, typ) + node = api.lookup_qualified(typ.name, typ) if node is not None and isinstance(node, SymbolTableNode): bound_type = node.node @@ -147,7 +148,7 @@ def _unbound_to_instance( def _info_for_cls(cls, api): if cls.info is CLASSDEF_NO_INFO: - sym = api.lookup(cls.name, cls) + sym = api.lookup_qualified(cls.name, cls) if sym.node and isinstance(sym.node, TypeInfo): info = sym.node else: |