aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/ipython/py2/IPython/utils/dir2.py
blob: fb0cd719ef05be78d6d058295b41e3997e10014b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
# encoding: utf-8 
"""A fancy version of Python's builtin :func:`dir` function. 
""" 
 
# Copyright (c) IPython Development Team.
# Distributed under the terms of the Modified BSD License.
 
import inspect
from .py3compat import string_types 
 
 
def safe_hasattr(obj, attr): 
    """In recent versions of Python, hasattr() only catches AttributeError. 
    This catches all errors. 
    """ 
    try: 
        getattr(obj, attr) 
        return True 
    except: 
        return False 
 
 
def dir2(obj): 
    """dir2(obj) -> list of strings 
 
    Extended version of the Python builtin dir(), which does a few extra 
    checks. 
 
    This version is guaranteed to return only a list of true strings, whereas 
    dir() returns anything that objects inject into themselves, even if they 
    are later not really valid for attribute access (many extension libraries 
    have such bugs). 
    """ 
 
    # Start building the attribute list via dir(), and then complete it 
    # with a few extra special-purpose calls. 
 
    try: 
        words = set(dir(obj)) 
    except Exception: 
        # TypeError: dir(obj) does not return a list 
        words = set() 
 
    # filter out non-string attributes which may be stuffed by dir() calls 
    # and poor coding in third-party modules 
 
    words = [w for w in words if isinstance(w, string_types)] 
    return sorted(words) 


def get_real_method(obj, name):
    """Like getattr, but with a few extra sanity checks:

    - If obj is a class, ignore its methods
    - Check if obj is a proxy that claims to have all attributes
    - Catch attribute access failing with any exception
    - Check that the attribute is a callable object

    Returns the method or None.
    """
    if inspect.isclass(obj):
        return None

    try:
        canary = getattr(obj, '_ipython_canary_method_should_not_exist_', None)
    except Exception:
        return None

    if canary is not None:
        # It claimed to have an attribute it should never have
        return None

    try:
        m = getattr(obj, name, None)
    except Exception:
        return None

    if callable(m):
        return m

    return None