Gecko [ lorencats.com ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
brain_argparse.py	1 KB	-rw-r--r--
brain_attrs.py	1.86 KB	-rw-r--r--
brain_builtin_inference.py	26.28 KB	-rw-r--r--
brain_collections.py	2.76 KB	-rw-r--r--
brain_curses.py	3.23 KB	-rw-r--r--
brain_dateutil.py	795 B	-rw-r--r--
brain_fstrings.py	2.07 KB	-rw-r--r--
brain_functools.py	6.16 KB	-rw-r--r--
brain_gi.py	6.65 KB	-rw-r--r--
brain_hashlib.py	2.11 KB	-rw-r--r--
brain_io.py	1.44 KB	-rw-r--r--
brain_mechanize.py	889 B	-rw-r--r--
brain_multiprocessing.py	3.04 KB	-rw-r--r--
brain_namedtuple_enum.py	13.7 KB	-rw-r--r--
brain_nose.py	2.16 KB	-rw-r--r--
brain_numpy.py	14.44 KB	-rw-r--r--
brain_pkg_resources.py	2.21 KB	-rw-r--r--
brain_pytest.py	2.33 KB	-rw-r--r--
brain_qt.py	2.38 KB	-rw-r--r--
brain_random.py	2.38 KB	-rw-r--r--
brain_re.py	1.1 KB	-rw-r--r--
brain_six.py	6 KB	-rw-r--r--
brain_ssl.py	3.47 KB	-rw-r--r--
brain_subprocess.py	3.55 KB	-rw-r--r--
brain_threading.py	767 B	-rw-r--r--
brain_typing.py	2.66 KB	-rw-r--r--
brain_uuid.py	564 B	-rw-r--r--

Code Editor : brain_typing.py

# -*- coding: utf-8 -*-
# Copyright (c) 2017-2018 Claudiu Popa <pcmanticore@gmail.com>
# Copyright (c) 2017 Łukasz Rogalski <rogalski.91@gmail.com>
# Copyright (c) 2017 David Euresti <github@euresti.com>
# Copyright (c) 2018 Bryce Guinta <bryce.paul.guinta@gmail.com>

"""Astroid hooks for typing.py support."""
import typing

from astroid import (
    MANAGER,
    UseInferenceDefault,
    extract_node,
    inference_tip,
    nodes,
    InferenceError,
)

TYPING_NAMEDTUPLE_BASENAMES = {"NamedTuple", "typing.NamedTuple"}
TYPING_TYPEVARS = {"TypeVar", "NewType"}
TYPING_TYPEVARS_QUALIFIED = {"typing.TypeVar", "typing.NewType"}
TYPING_TYPE_TEMPLATE = """
class Meta(type):
    def __getitem__(self, item):
        return self

class {0}(metaclass=Meta):
    pass
"""
TYPING_MEMBERS = set(typing.__all__)

def looks_like_typing_typevar_or_newtype(node):
    func = node.func
    if isinstance(func, nodes.Attribute):
        return func.attrname in TYPING_TYPEVARS
    if isinstance(func, nodes.Name):
        return func.name in TYPING_TYPEVARS
    return False

def infer_typing_typevar_or_newtype(node, context=None):
    """Infer a typing.TypeVar(...) or typing.NewType(...) call"""
    try:
        func = next(node.func.infer(context=context))
    except InferenceError as exc:
        raise UseInferenceDefault from exc

if func.qname() not in TYPING_TYPEVARS_QUALIFIED:
        raise UseInferenceDefault
    if not node.args:
        raise UseInferenceDefault

typename = node.args[0].as_string().strip("'")
    node = extract_node(TYPING_TYPE_TEMPLATE.format(typename))
    return node.infer(context=context)

def _looks_like_typing_subscript(node):
    """Try to figure out if a Subscript node *might* be a typing-related subscript"""
    if isinstance(node, nodes.Name):
        return node.name in TYPING_MEMBERS
    elif isinstance(node, nodes.Attribute):
        return node.attrname in TYPING_MEMBERS
    elif isinstance(node, nodes.Subscript):
        return _looks_like_typing_subscript(node.value)
    return False

def infer_typing_attr(node, context=None):
    """Infer a typing.X[...] subscript"""
    try:
        value = next(node.value.infer())
    except InferenceError as exc:
        raise UseInferenceDefault from exc

if not value.qname().startswith("typing."):
        raise UseInferenceDefault

node = extract_node(TYPING_TYPE_TEMPLATE.format(value.qname().split(".")[-1]))
    return node.infer(context=context)

MANAGER.register_transform(
    nodes.Call,
    inference_tip(infer_typing_typevar_or_newtype),
    looks_like_typing_typevar_or_newtype,
)
MANAGER.register_transform(
    nodes.Subscript, inference_tip(infer_typing_attr), _looks_like_typing_subscript
)

${this.title}

Code Editor : brain_typing.py