diff options
| author | mike bayer <mike_mp@zzzcomputing.com> | 2021-03-31 22:30:06 +0000 |
|---|---|---|
| committer | Gerrit Code Review <gerrit@ci3.zzzcomputing.com> | 2021-03-31 22:30:06 +0000 |
| commit | 1c8e221b55082afcecde0c05fe7b00d8018f8015 (patch) | |
| tree | ee75e0c7230d6b665b972c61b3ddb44cb09ee61a /lib/sqlalchemy/ext/mypy/util.py | |
| parent | 62e68beea671215e98cb939bb87af95d6d9f35ee (diff) | |
| parent | 803a2373f58e794499e1e0b476db4c23e8dd7f87 (diff) | |
| download | sqlalchemy-1c8e221b55082afcecde0c05fe7b00d8018f8015.tar.gz | |
Merge "Add DeclarativeMeta to globals"
Diffstat (limited to 'lib/sqlalchemy/ext/mypy/util.py')
| -rw-r--r-- | lib/sqlalchemy/ext/mypy/util.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/lib/sqlalchemy/ext/mypy/util.py b/lib/sqlalchemy/ext/mypy/util.py index e7178a885..7079f3cd7 100644 --- a/lib/sqlalchemy/ext/mypy/util.py +++ b/lib/sqlalchemy/ext/mypy/util.py @@ -18,6 +18,22 @@ def fail(api: SemanticAnalyzerPluginInterface, msg: str, ctx: Context): return api.fail(msg, ctx) +def add_global( + ctx: SemanticAnalyzerPluginInterface, + module: str, + symbol_name: str, + asname: str, +): + module_globals = ctx.api.modules[ctx.api.cur_mod_id].names + + if asname not in module_globals: + lookup_sym: SymbolTableNode = ctx.api.modules[module].names[ + symbol_name + ] + + module_globals[asname] = lookup_sym + + def _get_callexpr_kwarg(callexpr: CallExpr, name: str) -> Optional[NameExpr]: try: arg_idx = callexpr.arg_names.index(name) |