summaryrefslogtreecommitdiffstats
path: root/Lib/idlelib/CallTips.py
blob: b3629c447d79643d495425d0872b94b4f595bdb4 (plain)
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
"""CallTips.py - An IDLE Extension to Jog Your Memory

Call Tips are floating windows which display function, class, and method
parameter and docstring information when you type an opening parenthesis, and
which disappear when you type a closing parenthesis.

"""
import re
import sys
import types
import inspect

from idlelib import CallTipWindow
from idlelib.HyperParser import HyperParser

import __main__

class CallTips:

    menudefs = [
        ('edit', [
            ("Show call tip", "<<force-open-calltip>>"),
        ])
    ]

    def __init__(self, editwin=None):
        if editwin is None:  # subprocess and test
            self.editwin = None
        else:
            self.editwin = editwin
            self.text = editwin.text
            self.active_calltip = None
            self._calltip_window = self._make_tk_calltip_window

    def close(self):
        self._calltip_window = None

    def _make_tk_calltip_window(self):
        # See __init__ for usage
        return CallTipWindow.CallTip(self.text)

    def _remove_calltip_window(self, event=None):
        if self.active_calltip:
            self.active_calltip.hidetip()
            self.active_calltip = None

    def force_open_calltip_event(self, event):
        "The user selected the menu entry or hotkey, open the tip."
        self.open_calltip(True)

    def try_open_calltip_event(self, event):
        """Happens when it would be nice to open a CallTip, but not really
        necessary, for example after an opening bracket, so function calls
        won't be made.
        """
        self.open_calltip(False)

    def refresh_calltip_event(self, event):
        if self.active_calltip and self.active_calltip.is_active():
            self.open_calltip(False)

    def open_calltip(self, evalfuncs):
        self._remove_calltip_window()

        hp = HyperParser(self.editwin, "insert")
        sur_paren = hp.get_surrounding_brackets('(')
        if not sur_paren:
            return
        hp.set_index(sur_paren[0])
        expression  = hp.get_expression()
        if not expression:
            return
        if not evalfuncs and (expression.find('(') != -1):
            return
        argspec = self.fetch_tip(expression)
        if not argspec:
            return
        self.active_calltip = self._calltip_window()
        self.active_calltip.showtip(argspec, sur_paren[0], sur_paren[1])

    def fetch_tip(self, expression):
        """Return the argument list and docstring of a function or class.

        If there is a Python subprocess, get the calltip there.  Otherwise,
        either this fetch_tip() is running in the subprocess or it was
        called in an IDLE running without the subprocess.

        The subprocess environment is that of the most recently run script.  If
        two unrelated modules are being edited some calltips in the current
        module may be inoperative if the module was not the last to run.

        To find methods, fetch_tip must be fed a fully qualified name.

        """
        try:
            rpcclt = self.editwin.flist.pyshell.interp.rpcclt
        except AttributeError:
            rpcclt = None
        if rpcclt:
            return rpcclt.remotecall("exec", "get_the_calltip",
                                     (expression,), {})
        else:
            return get_argspec(get_entity(expression))

def get_entity(expression):
    """Return the object corresponding to expression evaluated
    in a namespace spanning sys.modules and __main.dict__.
    """
    if expression:
        namespace = sys.modules.copy()
        namespace.update(__main__.__dict__)
        try:
            return eval(expression, namespace)
        except BaseException:
            # An uncaught exception closes idle, and eval can raise any
            # exception, especially if user classes are involved.
            return None

# The following are used in both get_argspec and tests
_first_param = re.compile('(?<=\()\w*\,?\s*')
_default_callable_argspec = "No docstring, see docs."

def get_argspec(ob):
    '''Return a string describing the arguments and return of a callable object.

    For Python-coded functions and methods, the first line is introspected.
    Delete 'self' parameter for classes (.__init__) and bound methods.
    The last line is the first line of the doc string.  For builtins, this typically
    includes the arguments in addition to the return value.

    '''
    argspec = ""
    if hasattr(ob, '__call__'):
        if isinstance(ob, type):
            fob = getattr(ob, '__init__', None)
        elif isinstance(ob.__call__, types.MethodType):
            fob = ob.__call__
        else:
            fob = ob
        if isinstance(fob, (types.FunctionType, types.MethodType)):
            argspec = inspect.formatargspec(*inspect.getfullargspec(fob))
            if (isinstance(ob, (type, types.MethodType)) or
                    isinstance(ob.__call__, types.MethodType)):
                argspec = _first_param.sub("", argspec)

        if isinstance(ob.__call__, types.MethodType):
            doc = ob.__call__.__doc__
        else:
            doc = getattr(ob, "__doc__", "")
        if doc:
            doc = doc.lstrip()
            pos = doc.find("\n")
            if pos < 0 or pos > 70:
                pos = 70
            if argspec:
                argspec += "\n"
            argspec += doc[:pos]
        if not argspec:
            argspec = _default_callable_argspec
    return argspec

#################################################
#
# Test code tests CallTips.fetch_tip, get_entity, and get_argspec

def main():
    # Putting expected in docstrings results in doubled tips for test
    def t1(): "()"
    def t2(a, b=None): "(a, b=None)"
    def t3(a, *args): "(a, *args)"
    def t4(*args): "(*args)"
    def t5(a, b=None, *args, **kw): "(a, b=None, *args, **kw)"

    class TC(object):
        "(ai=None, *b)"
        def __init__(self, ai=None, *b): "(self, ai=None, *b)"
        def t1(self): "(self)"
        def t2(self, ai, b=None): "(self, ai, b=None)"
        def t3(self, ai, *args): "(self, ai, *args)"
        def t4(self, *args): "(self, *args)"
        def t5(self, ai, b=None, *args, **kw): "(self, ai, b=None, *args, **kw)"
        def t6(no, self): "(no, self)"
        @classmethod
        def cm(cls, a): "(cls, a)"
        @staticmethod
        def sm(b): "(b)"
        def __call__(self, ci): "(self, ci)"

    tc = TC()

    # Python classes that inherit builtin methods
    class Int(int):  "Int(x[, base]) -> integer"
    class List(list): "List() -> new empty list"
    # Simulate builtin with no docstring for default argspec test
    class SB:  __call__ = None

    __main__.__dict__.update(locals())  # required for get_entity eval()

    num_tests = num_fail = 0
    tip = CallTips().fetch_tip

    def test(expression, expected):
        nonlocal num_tests, num_fail
        num_tests += 1
        argspec = tip(expression)
        if argspec != expected:
            num_fail += 1
            fmt = "%s - expected\n%r\n - but got\n%r"
            print(fmt % (expression, expected, argspec))

    def test_builtins():
        # if first line of a possibly multiline compiled docstring changes,
        # must change corresponding test string
        test('int',  "int(x[, base]) -> integer")
        test('Int',  Int.__doc__)
        test('types.MethodType', "method(function, instance)")
        test('list', "list() -> new empty list")
        test('List', List.__doc__)
        test('list.__new__',
               'T.__new__(S, ...) -> a new object with type S, a subtype of T')
        test('list.__init__',
               'x.__init__(...) initializes x; see help(type(x)) for signature')
        append_doc =  "L.append(object) -> None -- append object to end"
        test('list.append', append_doc)
        test('[].append', append_doc)
        test('List.append', append_doc)
        test('SB()', _default_callable_argspec)

    def test_funcs():
        for func  in (t1, t2, t3, t4, t5, TC,):
            fdoc = func.__doc__
            test(func.__name__, fdoc + "\n" + fdoc)
        for func in (TC.t1, TC.t2, TC.t3, TC.t4, TC.t5, TC.t6, TC.sm,
                     TC.__call__):
            fdoc = func.__doc__
            test('TC.'+func.__name__, fdoc + "\n" + fdoc)
        fdoc = TC.cm.__func__.__doc__
        test('TC.cm.__func__', fdoc + "\n" + fdoc)

    def test_methods():
        # test that first parameter is correctly removed from argspec
        # using _first_param re to calculate expected masks re errors
        for meth, mdoc  in ((tc.t1, "()"), (tc.t4, "(*args)"), (tc.t6, "(self)"),
                            (TC.cm, "(a)"),):
            test('tc.'+meth.__name__, mdoc + "\n" + meth.__doc__)
        test('tc', "(ci)" + "\n" + tc.__call__.__doc__)
        # directly test that re works to delete unicode parameter name
        uni = "(A\u0391\u0410\u05d0\u0627\u0905\u1e00\u3042, a)"  # various As
        assert _first_param.sub('', uni) == '(a)'

    def test_non_callables():
        # expression evaluates, but not to a callable
        for expr in ('0', '0.0' 'num_tests', b'num_tests', '[]', '{}'):
            test(expr, '')
        # expression does not evaluate, but raises an exception
        for expr in ('1a', 'xyx', 'num_tests.xyz', '[int][1]', '{0:int}[1]'):
            test(expr, '')

    test_builtins()
    test_funcs()
    test_non_callables()
    test_methods()

    print("%d of %d tests failed" % (num_fail, num_tests))

if __name__ == '__main__':
    main()