##// END OF EJS Templates
Merge pull request #11419 from elyashiv/fix-tab-kw-args...
Matthias Bussonnier -
r24768:5a8f4071 merge
parent child Browse files
Show More
@@ -1,2066 +1,2061 b''
1 """Completion for IPython.
1 """Completion for IPython.
2
2
3 This module started as fork of the rlcompleter module in the Python standard
3 This module started as fork of the rlcompleter module in the Python standard
4 library. The original enhancements made to rlcompleter have been sent
4 library. The original enhancements made to rlcompleter have been sent
5 upstream and were accepted as of Python 2.3,
5 upstream and were accepted as of Python 2.3,
6
6
7 This module now support a wide variety of completion mechanism both available
7 This module now support a wide variety of completion mechanism both available
8 for normal classic Python code, as well as completer for IPython specific
8 for normal classic Python code, as well as completer for IPython specific
9 Syntax like magics.
9 Syntax like magics.
10
10
11 Latex and Unicode completion
11 Latex and Unicode completion
12 ============================
12 ============================
13
13
14 IPython and compatible frontends not only can complete your code, but can help
14 IPython and compatible frontends not only can complete your code, but can help
15 you to input a wide range of characters. In particular we allow you to insert
15 you to input a wide range of characters. In particular we allow you to insert
16 a unicode character using the tab completion mechanism.
16 a unicode character using the tab completion mechanism.
17
17
18 Forward latex/unicode completion
18 Forward latex/unicode completion
19 --------------------------------
19 --------------------------------
20
20
21 Forward completion allows you to easily type a unicode character using its latex
21 Forward completion allows you to easily type a unicode character using its latex
22 name, or unicode long description. To do so type a backslash follow by the
22 name, or unicode long description. To do so type a backslash follow by the
23 relevant name and press tab:
23 relevant name and press tab:
24
24
25
25
26 Using latex completion:
26 Using latex completion:
27
27
28 .. code::
28 .. code::
29
29
30 \\alpha<tab>
30 \\alpha<tab>
31 Ξ±
31 Ξ±
32
32
33 or using unicode completion:
33 or using unicode completion:
34
34
35
35
36 .. code::
36 .. code::
37
37
38 \\greek small letter alpha<tab>
38 \\greek small letter alpha<tab>
39 Ξ±
39 Ξ±
40
40
41
41
42 Only valid Python identifiers will complete. Combining characters (like arrow or
42 Only valid Python identifiers will complete. Combining characters (like arrow or
43 dots) are also available, unlike latex they need to be put after the their
43 dots) are also available, unlike latex they need to be put after the their
44 counterpart that is to say, `F\\\\vec<tab>` is correct, not `\\\\vec<tab>F`.
44 counterpart that is to say, `F\\\\vec<tab>` is correct, not `\\\\vec<tab>F`.
45
45
46 Some browsers are known to display combining characters incorrectly.
46 Some browsers are known to display combining characters incorrectly.
47
47
48 Backward latex completion
48 Backward latex completion
49 -------------------------
49 -------------------------
50
50
51 It is sometime challenging to know how to type a character, if you are using
51 It is sometime challenging to know how to type a character, if you are using
52 IPython, or any compatible frontend you can prepend backslash to the character
52 IPython, or any compatible frontend you can prepend backslash to the character
53 and press `<tab>` to expand it to its latex form.
53 and press `<tab>` to expand it to its latex form.
54
54
55 .. code::
55 .. code::
56
56
57 \\Ξ±<tab>
57 \\Ξ±<tab>
58 \\alpha
58 \\alpha
59
59
60
60
61 Both forward and backward completions can be deactivated by setting the
61 Both forward and backward completions can be deactivated by setting the
62 ``Completer.backslash_combining_completions`` option to ``False``.
62 ``Completer.backslash_combining_completions`` option to ``False``.
63
63
64
64
65 Experimental
65 Experimental
66 ============
66 ============
67
67
68 Starting with IPython 6.0, this module can make use of the Jedi library to
68 Starting with IPython 6.0, this module can make use of the Jedi library to
69 generate completions both using static analysis of the code, and dynamically
69 generate completions both using static analysis of the code, and dynamically
70 inspecting multiple namespaces. The APIs attached to this new mechanism is
70 inspecting multiple namespaces. The APIs attached to this new mechanism is
71 unstable and will raise unless use in an :any:`provisionalcompleter` context
71 unstable and will raise unless use in an :any:`provisionalcompleter` context
72 manager.
72 manager.
73
73
74 You will find that the following are experimental:
74 You will find that the following are experimental:
75
75
76 - :any:`provisionalcompleter`
76 - :any:`provisionalcompleter`
77 - :any:`IPCompleter.completions`
77 - :any:`IPCompleter.completions`
78 - :any:`Completion`
78 - :any:`Completion`
79 - :any:`rectify_completions`
79 - :any:`rectify_completions`
80
80
81 .. note::
81 .. note::
82
82
83 better name for :any:`rectify_completions` ?
83 better name for :any:`rectify_completions` ?
84
84
85 We welcome any feedback on these new API, and we also encourage you to try this
85 We welcome any feedback on these new API, and we also encourage you to try this
86 module in debug mode (start IPython with ``--Completer.debug=True``) in order
86 module in debug mode (start IPython with ``--Completer.debug=True``) in order
87 to have extra logging information is :any:`jedi` is crashing, or if current
87 to have extra logging information is :any:`jedi` is crashing, or if current
88 IPython completer pending deprecations are returning results not yet handled
88 IPython completer pending deprecations are returning results not yet handled
89 by :any:`jedi`
89 by :any:`jedi`
90
90
91 Using Jedi for tab completion allow snippets like the following to work without
91 Using Jedi for tab completion allow snippets like the following to work without
92 having to execute any code:
92 having to execute any code:
93
93
94 >>> myvar = ['hello', 42]
94 >>> myvar = ['hello', 42]
95 ... myvar[1].bi<tab>
95 ... myvar[1].bi<tab>
96
96
97 Tab completion will be able to infer that ``myvar[1]`` is a real number without
97 Tab completion will be able to infer that ``myvar[1]`` is a real number without
98 executing any code unlike the previously available ``IPCompleter.greedy``
98 executing any code unlike the previously available ``IPCompleter.greedy``
99 option.
99 option.
100
100
101 Be sure to update :any:`jedi` to the latest stable version or to try the
101 Be sure to update :any:`jedi` to the latest stable version or to try the
102 current development version to get better completions.
102 current development version to get better completions.
103 """
103 """
104
104
105
105
106 # Copyright (c) IPython Development Team.
106 # Copyright (c) IPython Development Team.
107 # Distributed under the terms of the Modified BSD License.
107 # Distributed under the terms of the Modified BSD License.
108 #
108 #
109 # Some of this code originated from rlcompleter in the Python standard library
109 # Some of this code originated from rlcompleter in the Python standard library
110 # Copyright (C) 2001 Python Software Foundation, www.python.org
110 # Copyright (C) 2001 Python Software Foundation, www.python.org
111
111
112
112
113 import __main__
113 import __main__
114 import builtins as builtin_mod
114 import builtins as builtin_mod
115 import glob
115 import glob
116 import time
116 import time
117 import inspect
117 import inspect
118 import itertools
118 import itertools
119 import keyword
119 import keyword
120 import os
120 import os
121 import re
121 import re
122 import sys
122 import sys
123 import unicodedata
123 import unicodedata
124 import string
124 import string
125 import warnings
125 import warnings
126
126
127 from contextlib import contextmanager
127 from contextlib import contextmanager
128 from importlib import import_module
128 from importlib import import_module
129 from typing import Iterator, List, Tuple, Iterable, Union
129 from typing import Iterator, List, Tuple, Iterable, Union
130 from types import SimpleNamespace
130 from types import SimpleNamespace
131
131
132 from traitlets.config.configurable import Configurable
132 from traitlets.config.configurable import Configurable
133 from IPython.core.error import TryNext
133 from IPython.core.error import TryNext
134 from IPython.core.inputtransformer2 import ESC_MAGIC
134 from IPython.core.inputtransformer2 import ESC_MAGIC
135 from IPython.core.latex_symbols import latex_symbols, reverse_latex_symbol
135 from IPython.core.latex_symbols import latex_symbols, reverse_latex_symbol
136 from IPython.core.oinspect import InspectColors
136 from IPython.core.oinspect import InspectColors
137 from IPython.utils import generics
137 from IPython.utils import generics
138 from IPython.utils.dir2 import dir2, get_real_method
138 from IPython.utils.dir2 import dir2, get_real_method
139 from IPython.utils.process import arg_split
139 from IPython.utils.process import arg_split
140 from traitlets import Bool, Enum, observe, Int
140 from traitlets import Bool, Enum, observe, Int
141
141
142 # skip module docstests
142 # skip module docstests
143 skip_doctest = True
143 skip_doctest = True
144
144
145 try:
145 try:
146 import jedi
146 import jedi
147 jedi.settings.case_insensitive_completion = False
147 jedi.settings.case_insensitive_completion = False
148 import jedi.api.helpers
148 import jedi.api.helpers
149 import jedi.api.classes
149 import jedi.api.classes
150 JEDI_INSTALLED = True
150 JEDI_INSTALLED = True
151 except ImportError:
151 except ImportError:
152 JEDI_INSTALLED = False
152 JEDI_INSTALLED = False
153 #-----------------------------------------------------------------------------
153 #-----------------------------------------------------------------------------
154 # Globals
154 # Globals
155 #-----------------------------------------------------------------------------
155 #-----------------------------------------------------------------------------
156
156
157 # Public API
157 # Public API
158 __all__ = ['Completer','IPCompleter']
158 __all__ = ['Completer','IPCompleter']
159
159
160 if sys.platform == 'win32':
160 if sys.platform == 'win32':
161 PROTECTABLES = ' '
161 PROTECTABLES = ' '
162 else:
162 else:
163 PROTECTABLES = ' ()[]{}?=\\|;:\'#*"^&'
163 PROTECTABLES = ' ()[]{}?=\\|;:\'#*"^&'
164
164
165 # Protect against returning an enormous number of completions which the frontend
165 # Protect against returning an enormous number of completions which the frontend
166 # may have trouble processing.
166 # may have trouble processing.
167 MATCHES_LIMIT = 500
167 MATCHES_LIMIT = 500
168
168
169 _deprecation_readline_sentinel = object()
169 _deprecation_readline_sentinel = object()
170
170
171
171
172 class ProvisionalCompleterWarning(FutureWarning):
172 class ProvisionalCompleterWarning(FutureWarning):
173 """
173 """
174 Exception raise by an experimental feature in this module.
174 Exception raise by an experimental feature in this module.
175
175
176 Wrap code in :any:`provisionalcompleter` context manager if you
176 Wrap code in :any:`provisionalcompleter` context manager if you
177 are certain you want to use an unstable feature.
177 are certain you want to use an unstable feature.
178 """
178 """
179 pass
179 pass
180
180
181 warnings.filterwarnings('error', category=ProvisionalCompleterWarning)
181 warnings.filterwarnings('error', category=ProvisionalCompleterWarning)
182
182
183 @contextmanager
183 @contextmanager
184 def provisionalcompleter(action='ignore'):
184 def provisionalcompleter(action='ignore'):
185 """
185 """
186
186
187
187
188 This contest manager has to be used in any place where unstable completer
188 This contest manager has to be used in any place where unstable completer
189 behavior and API may be called.
189 behavior and API may be called.
190
190
191 >>> with provisionalcompleter():
191 >>> with provisionalcompleter():
192 ... completer.do_experimetal_things() # works
192 ... completer.do_experimetal_things() # works
193
193
194 >>> completer.do_experimental_things() # raises.
194 >>> completer.do_experimental_things() # raises.
195
195
196 .. note:: Unstable
196 .. note:: Unstable
197
197
198 By using this context manager you agree that the API in use may change
198 By using this context manager you agree that the API in use may change
199 without warning, and that you won't complain if they do so.
199 without warning, and that you won't complain if they do so.
200
200
201 You also understand that if the API is not to you liking you should report
201 You also understand that if the API is not to you liking you should report
202 a bug to explain your use case upstream and improve the API and will loose
202 a bug to explain your use case upstream and improve the API and will loose
203 credibility if you complain after the API is make stable.
203 credibility if you complain after the API is make stable.
204
204
205 We'll be happy to get your feedback , feature request and improvement on
205 We'll be happy to get your feedback , feature request and improvement on
206 any of the unstable APIs !
206 any of the unstable APIs !
207 """
207 """
208 with warnings.catch_warnings():
208 with warnings.catch_warnings():
209 warnings.filterwarnings(action, category=ProvisionalCompleterWarning)
209 warnings.filterwarnings(action, category=ProvisionalCompleterWarning)
210 yield
210 yield
211
211
212
212
213 def has_open_quotes(s):
213 def has_open_quotes(s):
214 """Return whether a string has open quotes.
214 """Return whether a string has open quotes.
215
215
216 This simply counts whether the number of quote characters of either type in
216 This simply counts whether the number of quote characters of either type in
217 the string is odd.
217 the string is odd.
218
218
219 Returns
219 Returns
220 -------
220 -------
221 If there is an open quote, the quote character is returned. Else, return
221 If there is an open quote, the quote character is returned. Else, return
222 False.
222 False.
223 """
223 """
224 # We check " first, then ', so complex cases with nested quotes will get
224 # We check " first, then ', so complex cases with nested quotes will get
225 # the " to take precedence.
225 # the " to take precedence.
226 if s.count('"') % 2:
226 if s.count('"') % 2:
227 return '"'
227 return '"'
228 elif s.count("'") % 2:
228 elif s.count("'") % 2:
229 return "'"
229 return "'"
230 else:
230 else:
231 return False
231 return False
232
232
233
233
234 def protect_filename(s, protectables=PROTECTABLES):
234 def protect_filename(s, protectables=PROTECTABLES):
235 """Escape a string to protect certain characters."""
235 """Escape a string to protect certain characters."""
236 if set(s) & set(protectables):
236 if set(s) & set(protectables):
237 if sys.platform == "win32":
237 if sys.platform == "win32":
238 return '"' + s + '"'
238 return '"' + s + '"'
239 else:
239 else:
240 return "".join(("\\" + c if c in protectables else c) for c in s)
240 return "".join(("\\" + c if c in protectables else c) for c in s)
241 else:
241 else:
242 return s
242 return s
243
243
244
244
245 def expand_user(path:str) -> Tuple[str, bool, str]:
245 def expand_user(path:str) -> Tuple[str, bool, str]:
246 """Expand ``~``-style usernames in strings.
246 """Expand ``~``-style usernames in strings.
247
247
248 This is similar to :func:`os.path.expanduser`, but it computes and returns
248 This is similar to :func:`os.path.expanduser`, but it computes and returns
249 extra information that will be useful if the input was being used in
249 extra information that will be useful if the input was being used in
250 computing completions, and you wish to return the completions with the
250 computing completions, and you wish to return the completions with the
251 original '~' instead of its expanded value.
251 original '~' instead of its expanded value.
252
252
253 Parameters
253 Parameters
254 ----------
254 ----------
255 path : str
255 path : str
256 String to be expanded. If no ~ is present, the output is the same as the
256 String to be expanded. If no ~ is present, the output is the same as the
257 input.
257 input.
258
258
259 Returns
259 Returns
260 -------
260 -------
261 newpath : str
261 newpath : str
262 Result of ~ expansion in the input path.
262 Result of ~ expansion in the input path.
263 tilde_expand : bool
263 tilde_expand : bool
264 Whether any expansion was performed or not.
264 Whether any expansion was performed or not.
265 tilde_val : str
265 tilde_val : str
266 The value that ~ was replaced with.
266 The value that ~ was replaced with.
267 """
267 """
268 # Default values
268 # Default values
269 tilde_expand = False
269 tilde_expand = False
270 tilde_val = ''
270 tilde_val = ''
271 newpath = path
271 newpath = path
272
272
273 if path.startswith('~'):
273 if path.startswith('~'):
274 tilde_expand = True
274 tilde_expand = True
275 rest = len(path)-1
275 rest = len(path)-1
276 newpath = os.path.expanduser(path)
276 newpath = os.path.expanduser(path)
277 if rest:
277 if rest:
278 tilde_val = newpath[:-rest]
278 tilde_val = newpath[:-rest]
279 else:
279 else:
280 tilde_val = newpath
280 tilde_val = newpath
281
281
282 return newpath, tilde_expand, tilde_val
282 return newpath, tilde_expand, tilde_val
283
283
284
284
285 def compress_user(path:str, tilde_expand:bool, tilde_val:str) -> str:
285 def compress_user(path:str, tilde_expand:bool, tilde_val:str) -> str:
286 """Does the opposite of expand_user, with its outputs.
286 """Does the opposite of expand_user, with its outputs.
287 """
287 """
288 if tilde_expand:
288 if tilde_expand:
289 return path.replace(tilde_val, '~')
289 return path.replace(tilde_val, '~')
290 else:
290 else:
291 return path
291 return path
292
292
293
293
294 def completions_sorting_key(word):
294 def completions_sorting_key(word):
295 """key for sorting completions
295 """key for sorting completions
296
296
297 This does several things:
297 This does several things:
298
298
299 - Demote any completions starting with underscores to the end
299 - Demote any completions starting with underscores to the end
300 - Insert any %magic and %%cellmagic completions in the alphabetical order
300 - Insert any %magic and %%cellmagic completions in the alphabetical order
301 by their name
301 by their name
302 """
302 """
303 prio1, prio2 = 0, 0
303 prio1, prio2 = 0, 0
304
304
305 if word.startswith('__'):
305 if word.startswith('__'):
306 prio1 = 2
306 prio1 = 2
307 elif word.startswith('_'):
307 elif word.startswith('_'):
308 prio1 = 1
308 prio1 = 1
309
309
310 if word.endswith('='):
310 if word.endswith('='):
311 prio1 = -1
311 prio1 = -1
312
312
313 if word.startswith('%%'):
313 if word.startswith('%%'):
314 # If there's another % in there, this is something else, so leave it alone
314 # If there's another % in there, this is something else, so leave it alone
315 if not "%" in word[2:]:
315 if not "%" in word[2:]:
316 word = word[2:]
316 word = word[2:]
317 prio2 = 2
317 prio2 = 2
318 elif word.startswith('%'):
318 elif word.startswith('%'):
319 if not "%" in word[1:]:
319 if not "%" in word[1:]:
320 word = word[1:]
320 word = word[1:]
321 prio2 = 1
321 prio2 = 1
322
322
323 return prio1, word, prio2
323 return prio1, word, prio2
324
324
325
325
326 class _FakeJediCompletion:
326 class _FakeJediCompletion:
327 """
327 """
328 This is a workaround to communicate to the UI that Jedi has crashed and to
328 This is a workaround to communicate to the UI that Jedi has crashed and to
329 report a bug. Will be used only id :any:`IPCompleter.debug` is set to true.
329 report a bug. Will be used only id :any:`IPCompleter.debug` is set to true.
330
330
331 Added in IPython 6.0 so should likely be removed for 7.0
331 Added in IPython 6.0 so should likely be removed for 7.0
332
332
333 """
333 """
334
334
335 def __init__(self, name):
335 def __init__(self, name):
336
336
337 self.name = name
337 self.name = name
338 self.complete = name
338 self.complete = name
339 self.type = 'crashed'
339 self.type = 'crashed'
340 self.name_with_symbols = name
340 self.name_with_symbols = name
341 self.signature = ''
341 self.signature = ''
342 self._origin = 'fake'
342 self._origin = 'fake'
343
343
344 def __repr__(self):
344 def __repr__(self):
345 return '<Fake completion object jedi has crashed>'
345 return '<Fake completion object jedi has crashed>'
346
346
347
347
348 class Completion:
348 class Completion:
349 """
349 """
350 Completion object used and return by IPython completers.
350 Completion object used and return by IPython completers.
351
351
352 .. warning:: Unstable
352 .. warning:: Unstable
353
353
354 This function is unstable, API may change without warning.
354 This function is unstable, API may change without warning.
355 It will also raise unless use in proper context manager.
355 It will also raise unless use in proper context manager.
356
356
357 This act as a middle ground :any:`Completion` object between the
357 This act as a middle ground :any:`Completion` object between the
358 :any:`jedi.api.classes.Completion` object and the Prompt Toolkit completion
358 :any:`jedi.api.classes.Completion` object and the Prompt Toolkit completion
359 object. While Jedi need a lot of information about evaluator and how the
359 object. While Jedi need a lot of information about evaluator and how the
360 code should be ran/inspected, PromptToolkit (and other frontend) mostly
360 code should be ran/inspected, PromptToolkit (and other frontend) mostly
361 need user facing information.
361 need user facing information.
362
362
363 - Which range should be replaced replaced by what.
363 - Which range should be replaced replaced by what.
364 - Some metadata (like completion type), or meta information to displayed to
364 - Some metadata (like completion type), or meta information to displayed to
365 the use user.
365 the use user.
366
366
367 For debugging purpose we can also store the origin of the completion (``jedi``,
367 For debugging purpose we can also store the origin of the completion (``jedi``,
368 ``IPython.python_matches``, ``IPython.magics_matches``...).
368 ``IPython.python_matches``, ``IPython.magics_matches``...).
369 """
369 """
370
370
371 __slots__ = ['start', 'end', 'text', 'type', 'signature', '_origin']
371 __slots__ = ['start', 'end', 'text', 'type', 'signature', '_origin']
372
372
373 def __init__(self, start: int, end: int, text: str, *, type: str=None, _origin='', signature='') -> None:
373 def __init__(self, start: int, end: int, text: str, *, type: str=None, _origin='', signature='') -> None:
374 warnings.warn("``Completion`` is a provisional API (as of IPython 6.0). "
374 warnings.warn("``Completion`` is a provisional API (as of IPython 6.0). "
375 "It may change without warnings. "
375 "It may change without warnings. "
376 "Use in corresponding context manager.",
376 "Use in corresponding context manager.",
377 category=ProvisionalCompleterWarning, stacklevel=2)
377 category=ProvisionalCompleterWarning, stacklevel=2)
378
378
379 self.start = start
379 self.start = start
380 self.end = end
380 self.end = end
381 self.text = text
381 self.text = text
382 self.type = type
382 self.type = type
383 self.signature = signature
383 self.signature = signature
384 self._origin = _origin
384 self._origin = _origin
385
385
386 def __repr__(self):
386 def __repr__(self):
387 return '<Completion start=%s end=%s text=%r type=%r, signature=%r,>' % \
387 return '<Completion start=%s end=%s text=%r type=%r, signature=%r,>' % \
388 (self.start, self.end, self.text, self.type or '?', self.signature or '?')
388 (self.start, self.end, self.text, self.type or '?', self.signature or '?')
389
389
390 def __eq__(self, other)->Bool:
390 def __eq__(self, other)->Bool:
391 """
391 """
392 Equality and hash do not hash the type (as some completer may not be
392 Equality and hash do not hash the type (as some completer may not be
393 able to infer the type), but are use to (partially) de-duplicate
393 able to infer the type), but are use to (partially) de-duplicate
394 completion.
394 completion.
395
395
396 Completely de-duplicating completion is a bit tricker that just
396 Completely de-duplicating completion is a bit tricker that just
397 comparing as it depends on surrounding text, which Completions are not
397 comparing as it depends on surrounding text, which Completions are not
398 aware of.
398 aware of.
399 """
399 """
400 return self.start == other.start and \
400 return self.start == other.start and \
401 self.end == other.end and \
401 self.end == other.end and \
402 self.text == other.text
402 self.text == other.text
403
403
404 def __hash__(self):
404 def __hash__(self):
405 return hash((self.start, self.end, self.text))
405 return hash((self.start, self.end, self.text))
406
406
407
407
408 _IC = Iterable[Completion]
408 _IC = Iterable[Completion]
409
409
410
410
411 def _deduplicate_completions(text: str, completions: _IC)-> _IC:
411 def _deduplicate_completions(text: str, completions: _IC)-> _IC:
412 """
412 """
413 Deduplicate a set of completions.
413 Deduplicate a set of completions.
414
414
415 .. warning:: Unstable
415 .. warning:: Unstable
416
416
417 This function is unstable, API may change without warning.
417 This function is unstable, API may change without warning.
418
418
419 Parameters
419 Parameters
420 ----------
420 ----------
421 text: str
421 text: str
422 text that should be completed.
422 text that should be completed.
423 completions: Iterator[Completion]
423 completions: Iterator[Completion]
424 iterator over the completions to deduplicate
424 iterator over the completions to deduplicate
425
425
426 Yields
426 Yields
427 ------
427 ------
428 `Completions` objects
428 `Completions` objects
429
429
430
430
431 Completions coming from multiple sources, may be different but end up having
431 Completions coming from multiple sources, may be different but end up having
432 the same effect when applied to ``text``. If this is the case, this will
432 the same effect when applied to ``text``. If this is the case, this will
433 consider completions as equal and only emit the first encountered.
433 consider completions as equal and only emit the first encountered.
434
434
435 Not folded in `completions()` yet for debugging purpose, and to detect when
435 Not folded in `completions()` yet for debugging purpose, and to detect when
436 the IPython completer does return things that Jedi does not, but should be
436 the IPython completer does return things that Jedi does not, but should be
437 at some point.
437 at some point.
438 """
438 """
439 completions = list(completions)
439 completions = list(completions)
440 if not completions:
440 if not completions:
441 return
441 return
442
442
443 new_start = min(c.start for c in completions)
443 new_start = min(c.start for c in completions)
444 new_end = max(c.end for c in completions)
444 new_end = max(c.end for c in completions)
445
445
446 seen = set()
446 seen = set()
447 for c in completions:
447 for c in completions:
448 new_text = text[new_start:c.start] + c.text + text[c.end:new_end]
448 new_text = text[new_start:c.start] + c.text + text[c.end:new_end]
449 if new_text not in seen:
449 if new_text not in seen:
450 yield c
450 yield c
451 seen.add(new_text)
451 seen.add(new_text)
452
452
453
453
454 def rectify_completions(text: str, completions: _IC, *, _debug=False)->_IC:
454 def rectify_completions(text: str, completions: _IC, *, _debug=False)->_IC:
455 """
455 """
456 Rectify a set of completions to all have the same ``start`` and ``end``
456 Rectify a set of completions to all have the same ``start`` and ``end``
457
457
458 .. warning:: Unstable
458 .. warning:: Unstable
459
459
460 This function is unstable, API may change without warning.
460 This function is unstable, API may change without warning.
461 It will also raise unless use in proper context manager.
461 It will also raise unless use in proper context manager.
462
462
463 Parameters
463 Parameters
464 ----------
464 ----------
465 text: str
465 text: str
466 text that should be completed.
466 text that should be completed.
467 completions: Iterator[Completion]
467 completions: Iterator[Completion]
468 iterator over the completions to rectify
468 iterator over the completions to rectify
469
469
470
470
471 :any:`jedi.api.classes.Completion` s returned by Jedi may not have the same start and end, though
471 :any:`jedi.api.classes.Completion` s returned by Jedi may not have the same start and end, though
472 the Jupyter Protocol requires them to behave like so. This will readjust
472 the Jupyter Protocol requires them to behave like so. This will readjust
473 the completion to have the same ``start`` and ``end`` by padding both
473 the completion to have the same ``start`` and ``end`` by padding both
474 extremities with surrounding text.
474 extremities with surrounding text.
475
475
476 During stabilisation should support a ``_debug`` option to log which
476 During stabilisation should support a ``_debug`` option to log which
477 completion are return by the IPython completer and not found in Jedi in
477 completion are return by the IPython completer and not found in Jedi in
478 order to make upstream bug report.
478 order to make upstream bug report.
479 """
479 """
480 warnings.warn("`rectify_completions` is a provisional API (as of IPython 6.0). "
480 warnings.warn("`rectify_completions` is a provisional API (as of IPython 6.0). "
481 "It may change without warnings. "
481 "It may change without warnings. "
482 "Use in corresponding context manager.",
482 "Use in corresponding context manager.",
483 category=ProvisionalCompleterWarning, stacklevel=2)
483 category=ProvisionalCompleterWarning, stacklevel=2)
484
484
485 completions = list(completions)
485 completions = list(completions)
486 if not completions:
486 if not completions:
487 return
487 return
488 starts = (c.start for c in completions)
488 starts = (c.start for c in completions)
489 ends = (c.end for c in completions)
489 ends = (c.end for c in completions)
490
490
491 new_start = min(starts)
491 new_start = min(starts)
492 new_end = max(ends)
492 new_end = max(ends)
493
493
494 seen_jedi = set()
494 seen_jedi = set()
495 seen_python_matches = set()
495 seen_python_matches = set()
496 for c in completions:
496 for c in completions:
497 new_text = text[new_start:c.start] + c.text + text[c.end:new_end]
497 new_text = text[new_start:c.start] + c.text + text[c.end:new_end]
498 if c._origin == 'jedi':
498 if c._origin == 'jedi':
499 seen_jedi.add(new_text)
499 seen_jedi.add(new_text)
500 elif c._origin == 'IPCompleter.python_matches':
500 elif c._origin == 'IPCompleter.python_matches':
501 seen_python_matches.add(new_text)
501 seen_python_matches.add(new_text)
502 yield Completion(new_start, new_end, new_text, type=c.type, _origin=c._origin, signature=c.signature)
502 yield Completion(new_start, new_end, new_text, type=c.type, _origin=c._origin, signature=c.signature)
503 diff = seen_python_matches.difference(seen_jedi)
503 diff = seen_python_matches.difference(seen_jedi)
504 if diff and _debug:
504 if diff and _debug:
505 print('IPython.python matches have extras:', diff)
505 print('IPython.python matches have extras:', diff)
506
506
507
507
508 if sys.platform == 'win32':
508 if sys.platform == 'win32':
509 DELIMS = ' \t\n`!@#$^&*()=+[{]}|;\'",<>?'
509 DELIMS = ' \t\n`!@#$^&*()=+[{]}|;\'",<>?'
510 else:
510 else:
511 DELIMS = ' \t\n`!@#$^&*()=+[{]}\\|;:\'",<>?'
511 DELIMS = ' \t\n`!@#$^&*()=+[{]}\\|;:\'",<>?'
512
512
513 GREEDY_DELIMS = ' =\r\n'
513 GREEDY_DELIMS = ' =\r\n'
514
514
515
515
516 class CompletionSplitter(object):
516 class CompletionSplitter(object):
517 """An object to split an input line in a manner similar to readline.
517 """An object to split an input line in a manner similar to readline.
518
518
519 By having our own implementation, we can expose readline-like completion in
519 By having our own implementation, we can expose readline-like completion in
520 a uniform manner to all frontends. This object only needs to be given the
520 a uniform manner to all frontends. This object only needs to be given the
521 line of text to be split and the cursor position on said line, and it
521 line of text to be split and the cursor position on said line, and it
522 returns the 'word' to be completed on at the cursor after splitting the
522 returns the 'word' to be completed on at the cursor after splitting the
523 entire line.
523 entire line.
524
524
525 What characters are used as splitting delimiters can be controlled by
525 What characters are used as splitting delimiters can be controlled by
526 setting the ``delims`` attribute (this is a property that internally
526 setting the ``delims`` attribute (this is a property that internally
527 automatically builds the necessary regular expression)"""
527 automatically builds the necessary regular expression)"""
528
528
529 # Private interface
529 # Private interface
530
530
531 # A string of delimiter characters. The default value makes sense for
531 # A string of delimiter characters. The default value makes sense for
532 # IPython's most typical usage patterns.
532 # IPython's most typical usage patterns.
533 _delims = DELIMS
533 _delims = DELIMS
534
534
535 # The expression (a normal string) to be compiled into a regular expression
535 # The expression (a normal string) to be compiled into a regular expression
536 # for actual splitting. We store it as an attribute mostly for ease of
536 # for actual splitting. We store it as an attribute mostly for ease of
537 # debugging, since this type of code can be so tricky to debug.
537 # debugging, since this type of code can be so tricky to debug.
538 _delim_expr = None
538 _delim_expr = None
539
539
540 # The regular expression that does the actual splitting
540 # The regular expression that does the actual splitting
541 _delim_re = None
541 _delim_re = None
542
542
543 def __init__(self, delims=None):
543 def __init__(self, delims=None):
544 delims = CompletionSplitter._delims if delims is None else delims
544 delims = CompletionSplitter._delims if delims is None else delims
545 self.delims = delims
545 self.delims = delims
546
546
547 @property
547 @property
548 def delims(self):
548 def delims(self):
549 """Return the string of delimiter characters."""
549 """Return the string of delimiter characters."""
550 return self._delims
550 return self._delims
551
551
552 @delims.setter
552 @delims.setter
553 def delims(self, delims):
553 def delims(self, delims):
554 """Set the delimiters for line splitting."""
554 """Set the delimiters for line splitting."""
555 expr = '[' + ''.join('\\'+ c for c in delims) + ']'
555 expr = '[' + ''.join('\\'+ c for c in delims) + ']'
556 self._delim_re = re.compile(expr)
556 self._delim_re = re.compile(expr)
557 self._delims = delims
557 self._delims = delims
558 self._delim_expr = expr
558 self._delim_expr = expr
559
559
560 def split_line(self, line, cursor_pos=None):
560 def split_line(self, line, cursor_pos=None):
561 """Split a line of text with a cursor at the given position.
561 """Split a line of text with a cursor at the given position.
562 """
562 """
563 l = line if cursor_pos is None else line[:cursor_pos]
563 l = line if cursor_pos is None else line[:cursor_pos]
564 return self._delim_re.split(l)[-1]
564 return self._delim_re.split(l)[-1]
565
565
566
566
567
567
568 class Completer(Configurable):
568 class Completer(Configurable):
569
569
570 greedy = Bool(False,
570 greedy = Bool(False,
571 help="""Activate greedy completion
571 help="""Activate greedy completion
572 PENDING DEPRECTION. this is now mostly taken care of with Jedi.
572 PENDING DEPRECTION. this is now mostly taken care of with Jedi.
573
573
574 This will enable completion on elements of lists, results of function calls, etc.,
574 This will enable completion on elements of lists, results of function calls, etc.,
575 but can be unsafe because the code is actually evaluated on TAB.
575 but can be unsafe because the code is actually evaluated on TAB.
576 """
576 """
577 ).tag(config=True)
577 ).tag(config=True)
578
578
579 use_jedi = Bool(default_value=False,
579 use_jedi = Bool(default_value=False,
580 help="Experimental: Use Jedi to generate autocompletions. "
580 help="Experimental: Use Jedi to generate autocompletions. "
581 "Off by default.").tag(config=True)
581 "Off by default.").tag(config=True)
582
582
583 jedi_compute_type_timeout = Int(default_value=400,
583 jedi_compute_type_timeout = Int(default_value=400,
584 help="""Experimental: restrict time (in milliseconds) during which Jedi can compute types.
584 help="""Experimental: restrict time (in milliseconds) during which Jedi can compute types.
585 Set to 0 to stop computing types. Non-zero value lower than 100ms may hurt
585 Set to 0 to stop computing types. Non-zero value lower than 100ms may hurt
586 performance by preventing jedi to build its cache.
586 performance by preventing jedi to build its cache.
587 """).tag(config=True)
587 """).tag(config=True)
588
588
589 debug = Bool(default_value=False,
589 debug = Bool(default_value=False,
590 help='Enable debug for the Completer. Mostly print extra '
590 help='Enable debug for the Completer. Mostly print extra '
591 'information for experimental jedi integration.')\
591 'information for experimental jedi integration.')\
592 .tag(config=True)
592 .tag(config=True)
593
593
594 backslash_combining_completions = Bool(True,
594 backslash_combining_completions = Bool(True,
595 help="Enable unicode completions, e.g. \\alpha<tab> . "
595 help="Enable unicode completions, e.g. \\alpha<tab> . "
596 "Includes completion of latex commands, unicode names, and expanding "
596 "Includes completion of latex commands, unicode names, and expanding "
597 "unicode characters back to latex commands.").tag(config=True)
597 "unicode characters back to latex commands.").tag(config=True)
598
598
599
599
600
600
601 def __init__(self, namespace=None, global_namespace=None, **kwargs):
601 def __init__(self, namespace=None, global_namespace=None, **kwargs):
602 """Create a new completer for the command line.
602 """Create a new completer for the command line.
603
603
604 Completer(namespace=ns, global_namespace=ns2) -> completer instance.
604 Completer(namespace=ns, global_namespace=ns2) -> completer instance.
605
605
606 If unspecified, the default namespace where completions are performed
606 If unspecified, the default namespace where completions are performed
607 is __main__ (technically, __main__.__dict__). Namespaces should be
607 is __main__ (technically, __main__.__dict__). Namespaces should be
608 given as dictionaries.
608 given as dictionaries.
609
609
610 An optional second namespace can be given. This allows the completer
610 An optional second namespace can be given. This allows the completer
611 to handle cases where both the local and global scopes need to be
611 to handle cases where both the local and global scopes need to be
612 distinguished.
612 distinguished.
613 """
613 """
614
614
615 # Don't bind to namespace quite yet, but flag whether the user wants a
615 # Don't bind to namespace quite yet, but flag whether the user wants a
616 # specific namespace or to use __main__.__dict__. This will allow us
616 # specific namespace or to use __main__.__dict__. This will allow us
617 # to bind to __main__.__dict__ at completion time, not now.
617 # to bind to __main__.__dict__ at completion time, not now.
618 if namespace is None:
618 if namespace is None:
619 self.use_main_ns = True
619 self.use_main_ns = True
620 else:
620 else:
621 self.use_main_ns = False
621 self.use_main_ns = False
622 self.namespace = namespace
622 self.namespace = namespace
623
623
624 # The global namespace, if given, can be bound directly
624 # The global namespace, if given, can be bound directly
625 if global_namespace is None:
625 if global_namespace is None:
626 self.global_namespace = {}
626 self.global_namespace = {}
627 else:
627 else:
628 self.global_namespace = global_namespace
628 self.global_namespace = global_namespace
629
629
630 super(Completer, self).__init__(**kwargs)
630 super(Completer, self).__init__(**kwargs)
631
631
632 def complete(self, text, state):
632 def complete(self, text, state):
633 """Return the next possible completion for 'text'.
633 """Return the next possible completion for 'text'.
634
634
635 This is called successively with state == 0, 1, 2, ... until it
635 This is called successively with state == 0, 1, 2, ... until it
636 returns None. The completion should begin with 'text'.
636 returns None. The completion should begin with 'text'.
637
637
638 """
638 """
639 if self.use_main_ns:
639 if self.use_main_ns:
640 self.namespace = __main__.__dict__
640 self.namespace = __main__.__dict__
641
641
642 if state == 0:
642 if state == 0:
643 if "." in text:
643 if "." in text:
644 self.matches = self.attr_matches(text)
644 self.matches = self.attr_matches(text)
645 else:
645 else:
646 self.matches = self.global_matches(text)
646 self.matches = self.global_matches(text)
647 try:
647 try:
648 return self.matches[state]
648 return self.matches[state]
649 except IndexError:
649 except IndexError:
650 return None
650 return None
651
651
652 def global_matches(self, text):
652 def global_matches(self, text):
653 """Compute matches when text is a simple name.
653 """Compute matches when text is a simple name.
654
654
655 Return a list of all keywords, built-in functions and names currently
655 Return a list of all keywords, built-in functions and names currently
656 defined in self.namespace or self.global_namespace that match.
656 defined in self.namespace or self.global_namespace that match.
657
657
658 """
658 """
659 matches = []
659 matches = []
660 match_append = matches.append
660 match_append = matches.append
661 n = len(text)
661 n = len(text)
662 for lst in [keyword.kwlist,
662 for lst in [keyword.kwlist,
663 builtin_mod.__dict__.keys(),
663 builtin_mod.__dict__.keys(),
664 self.namespace.keys(),
664 self.namespace.keys(),
665 self.global_namespace.keys()]:
665 self.global_namespace.keys()]:
666 for word in lst:
666 for word in lst:
667 if word[:n] == text and word != "__builtins__":
667 if word[:n] == text and word != "__builtins__":
668 match_append(word)
668 match_append(word)
669
669
670 snake_case_re = re.compile(r"[^_]+(_[^_]+)+?\Z")
670 snake_case_re = re.compile(r"[^_]+(_[^_]+)+?\Z")
671 for lst in [self.namespace.keys(),
671 for lst in [self.namespace.keys(),
672 self.global_namespace.keys()]:
672 self.global_namespace.keys()]:
673 shortened = {"_".join([sub[0] for sub in word.split('_')]) : word
673 shortened = {"_".join([sub[0] for sub in word.split('_')]) : word
674 for word in lst if snake_case_re.match(word)}
674 for word in lst if snake_case_re.match(word)}
675 for word in shortened.keys():
675 for word in shortened.keys():
676 if word[:n] == text and word != "__builtins__":
676 if word[:n] == text and word != "__builtins__":
677 match_append(shortened[word])
677 match_append(shortened[word])
678 return matches
678 return matches
679
679
680 def attr_matches(self, text):
680 def attr_matches(self, text):
681 """Compute matches when text contains a dot.
681 """Compute matches when text contains a dot.
682
682
683 Assuming the text is of the form NAME.NAME....[NAME], and is
683 Assuming the text is of the form NAME.NAME....[NAME], and is
684 evaluatable in self.namespace or self.global_namespace, it will be
684 evaluatable in self.namespace or self.global_namespace, it will be
685 evaluated and its attributes (as revealed by dir()) are used as
685 evaluated and its attributes (as revealed by dir()) are used as
686 possible completions. (For class instances, class members are
686 possible completions. (For class instances, class members are
687 also considered.)
687 also considered.)
688
688
689 WARNING: this can still invoke arbitrary C code, if an object
689 WARNING: this can still invoke arbitrary C code, if an object
690 with a __getattr__ hook is evaluated.
690 with a __getattr__ hook is evaluated.
691
691
692 """
692 """
693
693
694 # Another option, seems to work great. Catches things like ''.<tab>
694 # Another option, seems to work great. Catches things like ''.<tab>
695 m = re.match(r"(\S+(\.\w+)*)\.(\w*)$", text)
695 m = re.match(r"(\S+(\.\w+)*)\.(\w*)$", text)
696
696
697 if m:
697 if m:
698 expr, attr = m.group(1, 3)
698 expr, attr = m.group(1, 3)
699 elif self.greedy:
699 elif self.greedy:
700 m2 = re.match(r"(.+)\.(\w*)$", self.line_buffer)
700 m2 = re.match(r"(.+)\.(\w*)$", self.line_buffer)
701 if not m2:
701 if not m2:
702 return []
702 return []
703 expr, attr = m2.group(1,2)
703 expr, attr = m2.group(1,2)
704 else:
704 else:
705 return []
705 return []
706
706
707 try:
707 try:
708 obj = eval(expr, self.namespace)
708 obj = eval(expr, self.namespace)
709 except:
709 except:
710 try:
710 try:
711 obj = eval(expr, self.global_namespace)
711 obj = eval(expr, self.global_namespace)
712 except:
712 except:
713 return []
713 return []
714
714
715 if self.limit_to__all__ and hasattr(obj, '__all__'):
715 if self.limit_to__all__ and hasattr(obj, '__all__'):
716 words = get__all__entries(obj)
716 words = get__all__entries(obj)
717 else:
717 else:
718 words = dir2(obj)
718 words = dir2(obj)
719
719
720 try:
720 try:
721 words = generics.complete_object(obj, words)
721 words = generics.complete_object(obj, words)
722 except TryNext:
722 except TryNext:
723 pass
723 pass
724 except AssertionError:
724 except AssertionError:
725 raise
725 raise
726 except Exception:
726 except Exception:
727 # Silence errors from completion function
727 # Silence errors from completion function
728 #raise # dbg
728 #raise # dbg
729 pass
729 pass
730 # Build match list to return
730 # Build match list to return
731 n = len(attr)
731 n = len(attr)
732 return [u"%s.%s" % (expr, w) for w in words if w[:n] == attr ]
732 return [u"%s.%s" % (expr, w) for w in words if w[:n] == attr ]
733
733
734
734
735 def get__all__entries(obj):
735 def get__all__entries(obj):
736 """returns the strings in the __all__ attribute"""
736 """returns the strings in the __all__ attribute"""
737 try:
737 try:
738 words = getattr(obj, '__all__')
738 words = getattr(obj, '__all__')
739 except:
739 except:
740 return []
740 return []
741
741
742 return [w for w in words if isinstance(w, str)]
742 return [w for w in words if isinstance(w, str)]
743
743
744
744
745 def match_dict_keys(keys: List[str], prefix: str, delims: str):
745 def match_dict_keys(keys: List[str], prefix: str, delims: str):
746 """Used by dict_key_matches, matching the prefix to a list of keys
746 """Used by dict_key_matches, matching the prefix to a list of keys
747
747
748 Parameters
748 Parameters
749 ==========
749 ==========
750 keys:
750 keys:
751 list of keys in dictionary currently being completed.
751 list of keys in dictionary currently being completed.
752 prefix:
752 prefix:
753 Part of the text already typed by the user. e.g. `mydict[b'fo`
753 Part of the text already typed by the user. e.g. `mydict[b'fo`
754 delims:
754 delims:
755 String of delimiters to consider when finding the current key.
755 String of delimiters to consider when finding the current key.
756
756
757 Returns
757 Returns
758 =======
758 =======
759
759
760 A tuple of three elements: ``quote``, ``token_start``, ``matched``, with
760 A tuple of three elements: ``quote``, ``token_start``, ``matched``, with
761 ``quote`` being the quote that need to be used to close current string.
761 ``quote`` being the quote that need to be used to close current string.
762 ``token_start`` the position where the replacement should start occurring,
762 ``token_start`` the position where the replacement should start occurring,
763 ``matches`` a list of replacement/completion
763 ``matches`` a list of replacement/completion
764
764
765 """
765 """
766 if not prefix:
766 if not prefix:
767 return None, 0, [repr(k) for k in keys
767 return None, 0, [repr(k) for k in keys
768 if isinstance(k, (str, bytes))]
768 if isinstance(k, (str, bytes))]
769 quote_match = re.search('["\']', prefix)
769 quote_match = re.search('["\']', prefix)
770 quote = quote_match.group()
770 quote = quote_match.group()
771 try:
771 try:
772 prefix_str = eval(prefix + quote, {})
772 prefix_str = eval(prefix + quote, {})
773 except Exception:
773 except Exception:
774 return None, 0, []
774 return None, 0, []
775
775
776 pattern = '[^' + ''.join('\\' + c for c in delims) + ']*$'
776 pattern = '[^' + ''.join('\\' + c for c in delims) + ']*$'
777 token_match = re.search(pattern, prefix, re.UNICODE)
777 token_match = re.search(pattern, prefix, re.UNICODE)
778 token_start = token_match.start()
778 token_start = token_match.start()
779 token_prefix = token_match.group()
779 token_prefix = token_match.group()
780
780
781 matched = []
781 matched = []
782 for key in keys:
782 for key in keys:
783 try:
783 try:
784 if not key.startswith(prefix_str):
784 if not key.startswith(prefix_str):
785 continue
785 continue
786 except (AttributeError, TypeError, UnicodeError):
786 except (AttributeError, TypeError, UnicodeError):
787 # Python 3+ TypeError on b'a'.startswith('a') or vice-versa
787 # Python 3+ TypeError on b'a'.startswith('a') or vice-versa
788 continue
788 continue
789
789
790 # reformat remainder of key to begin with prefix
790 # reformat remainder of key to begin with prefix
791 rem = key[len(prefix_str):]
791 rem = key[len(prefix_str):]
792 # force repr wrapped in '
792 # force repr wrapped in '
793 rem_repr = repr(rem + '"') if isinstance(rem, str) else repr(rem + b'"')
793 rem_repr = repr(rem + '"') if isinstance(rem, str) else repr(rem + b'"')
794 if rem_repr.startswith('u') and prefix[0] not in 'uU':
794 if rem_repr.startswith('u') and prefix[0] not in 'uU':
795 # Found key is unicode, but prefix is Py2 string.
795 # Found key is unicode, but prefix is Py2 string.
796 # Therefore attempt to interpret key as string.
796 # Therefore attempt to interpret key as string.
797 try:
797 try:
798 rem_repr = repr(rem.encode('ascii') + '"')
798 rem_repr = repr(rem.encode('ascii') + '"')
799 except UnicodeEncodeError:
799 except UnicodeEncodeError:
800 continue
800 continue
801
801
802 rem_repr = rem_repr[1 + rem_repr.index("'"):-2]
802 rem_repr = rem_repr[1 + rem_repr.index("'"):-2]
803 if quote == '"':
803 if quote == '"':
804 # The entered prefix is quoted with ",
804 # The entered prefix is quoted with ",
805 # but the match is quoted with '.
805 # but the match is quoted with '.
806 # A contained " hence needs escaping for comparison:
806 # A contained " hence needs escaping for comparison:
807 rem_repr = rem_repr.replace('"', '\\"')
807 rem_repr = rem_repr.replace('"', '\\"')
808
808
809 # then reinsert prefix from start of token
809 # then reinsert prefix from start of token
810 matched.append('%s%s' % (token_prefix, rem_repr))
810 matched.append('%s%s' % (token_prefix, rem_repr))
811 return quote, token_start, matched
811 return quote, token_start, matched
812
812
813
813
814 def cursor_to_position(text:str, line:int, column:int)->int:
814 def cursor_to_position(text:str, line:int, column:int)->int:
815 """
815 """
816
816
817 Convert the (line,column) position of the cursor in text to an offset in a
817 Convert the (line,column) position of the cursor in text to an offset in a
818 string.
818 string.
819
819
820 Parameters
820 Parameters
821 ----------
821 ----------
822
822
823 text : str
823 text : str
824 The text in which to calculate the cursor offset
824 The text in which to calculate the cursor offset
825 line : int
825 line : int
826 Line of the cursor; 0-indexed
826 Line of the cursor; 0-indexed
827 column : int
827 column : int
828 Column of the cursor 0-indexed
828 Column of the cursor 0-indexed
829
829
830 Return
830 Return
831 ------
831 ------
832 Position of the cursor in ``text``, 0-indexed.
832 Position of the cursor in ``text``, 0-indexed.
833
833
834 See Also
834 See Also
835 --------
835 --------
836 position_to_cursor: reciprocal of this function
836 position_to_cursor: reciprocal of this function
837
837
838 """
838 """
839 lines = text.split('\n')
839 lines = text.split('\n')
840 assert line <= len(lines), '{} <= {}'.format(str(line), str(len(lines)))
840 assert line <= len(lines), '{} <= {}'.format(str(line), str(len(lines)))
841
841
842 return sum(len(l) + 1 for l in lines[:line]) + column
842 return sum(len(l) + 1 for l in lines[:line]) + column
843
843
844 def position_to_cursor(text:str, offset:int)->Tuple[int, int]:
844 def position_to_cursor(text:str, offset:int)->Tuple[int, int]:
845 """
845 """
846 Convert the position of the cursor in text (0 indexed) to a line
846 Convert the position of the cursor in text (0 indexed) to a line
847 number(0-indexed) and a column number (0-indexed) pair
847 number(0-indexed) and a column number (0-indexed) pair
848
848
849 Position should be a valid position in ``text``.
849 Position should be a valid position in ``text``.
850
850
851 Parameters
851 Parameters
852 ----------
852 ----------
853
853
854 text : str
854 text : str
855 The text in which to calculate the cursor offset
855 The text in which to calculate the cursor offset
856 offset : int
856 offset : int
857 Position of the cursor in ``text``, 0-indexed.
857 Position of the cursor in ``text``, 0-indexed.
858
858
859 Return
859 Return
860 ------
860 ------
861 (line, column) : (int, int)
861 (line, column) : (int, int)
862 Line of the cursor; 0-indexed, column of the cursor 0-indexed
862 Line of the cursor; 0-indexed, column of the cursor 0-indexed
863
863
864
864
865 See Also
865 See Also
866 --------
866 --------
867 cursor_to_position : reciprocal of this function
867 cursor_to_position : reciprocal of this function
868
868
869
869
870 """
870 """
871
871
872 assert 0 <= offset <= len(text) , "0 <= %s <= %s" % (offset , len(text))
872 assert 0 <= offset <= len(text) , "0 <= %s <= %s" % (offset , len(text))
873
873
874 before = text[:offset]
874 before = text[:offset]
875 blines = before.split('\n') # ! splitnes trim trailing \n
875 blines = before.split('\n') # ! splitnes trim trailing \n
876 line = before.count('\n')
876 line = before.count('\n')
877 col = len(blines[-1])
877 col = len(blines[-1])
878 return line, col
878 return line, col
879
879
880
880
881 def _safe_isinstance(obj, module, class_name):
881 def _safe_isinstance(obj, module, class_name):
882 """Checks if obj is an instance of module.class_name if loaded
882 """Checks if obj is an instance of module.class_name if loaded
883 """
883 """
884 return (module in sys.modules and
884 return (module in sys.modules and
885 isinstance(obj, getattr(import_module(module), class_name)))
885 isinstance(obj, getattr(import_module(module), class_name)))
886
886
887
887
888 def back_unicode_name_matches(text):
888 def back_unicode_name_matches(text):
889 u"""Match unicode characters back to unicode name
889 u"""Match unicode characters back to unicode name
890
890
891 This does ``β˜ƒ`` -> ``\\snowman``
891 This does ``β˜ƒ`` -> ``\\snowman``
892
892
893 Note that snowman is not a valid python3 combining character but will be expanded.
893 Note that snowman is not a valid python3 combining character but will be expanded.
894 Though it will not recombine back to the snowman character by the completion machinery.
894 Though it will not recombine back to the snowman character by the completion machinery.
895
895
896 This will not either back-complete standard sequences like \\n, \\b ...
896 This will not either back-complete standard sequences like \\n, \\b ...
897
897
898 Used on Python 3 only.
898 Used on Python 3 only.
899 """
899 """
900 if len(text)<2:
900 if len(text)<2:
901 return u'', ()
901 return u'', ()
902 maybe_slash = text[-2]
902 maybe_slash = text[-2]
903 if maybe_slash != '\\':
903 if maybe_slash != '\\':
904 return u'', ()
904 return u'', ()
905
905
906 char = text[-1]
906 char = text[-1]
907 # no expand on quote for completion in strings.
907 # no expand on quote for completion in strings.
908 # nor backcomplete standard ascii keys
908 # nor backcomplete standard ascii keys
909 if char in string.ascii_letters or char in ['"',"'"]:
909 if char in string.ascii_letters or char in ['"',"'"]:
910 return u'', ()
910 return u'', ()
911 try :
911 try :
912 unic = unicodedata.name(char)
912 unic = unicodedata.name(char)
913 return '\\'+char,['\\'+unic]
913 return '\\'+char,['\\'+unic]
914 except KeyError:
914 except KeyError:
915 pass
915 pass
916 return u'', ()
916 return u'', ()
917
917
918 def back_latex_name_matches(text:str):
918 def back_latex_name_matches(text:str):
919 """Match latex characters back to unicode name
919 """Match latex characters back to unicode name
920
920
921 This does ``\\β„΅`` -> ``\\aleph``
921 This does ``\\β„΅`` -> ``\\aleph``
922
922
923 Used on Python 3 only.
923 Used on Python 3 only.
924 """
924 """
925 if len(text)<2:
925 if len(text)<2:
926 return u'', ()
926 return u'', ()
927 maybe_slash = text[-2]
927 maybe_slash = text[-2]
928 if maybe_slash != '\\':
928 if maybe_slash != '\\':
929 return u'', ()
929 return u'', ()
930
930
931
931
932 char = text[-1]
932 char = text[-1]
933 # no expand on quote for completion in strings.
933 # no expand on quote for completion in strings.
934 # nor backcomplete standard ascii keys
934 # nor backcomplete standard ascii keys
935 if char in string.ascii_letters or char in ['"',"'"]:
935 if char in string.ascii_letters or char in ['"',"'"]:
936 return u'', ()
936 return u'', ()
937 try :
937 try :
938 latex = reverse_latex_symbol[char]
938 latex = reverse_latex_symbol[char]
939 # '\\' replace the \ as well
939 # '\\' replace the \ as well
940 return '\\'+char,[latex]
940 return '\\'+char,[latex]
941 except KeyError:
941 except KeyError:
942 pass
942 pass
943 return u'', ()
943 return u'', ()
944
944
945
945
946 def _formatparamchildren(parameter) -> str:
946 def _formatparamchildren(parameter) -> str:
947 """
947 """
948 Get parameter name and value from Jedi Private API
948 Get parameter name and value from Jedi Private API
949
949
950 Jedi does not expose a simple way to get `param=value` from its API.
950 Jedi does not expose a simple way to get `param=value` from its API.
951
951
952 Parameter
952 Parameter
953 =========
953 =========
954
954
955 parameter:
955 parameter:
956 Jedi's function `Param`
956 Jedi's function `Param`
957
957
958 Returns
958 Returns
959 =======
959 =======
960
960
961 A string like 'a', 'b=1', '*args', '**kwargs'
961 A string like 'a', 'b=1', '*args', '**kwargs'
962
962
963
963
964 """
964 """
965 description = parameter.description
965 description = parameter.description
966 if not description.startswith('param '):
966 if not description.startswith('param '):
967 raise ValueError('Jedi function parameter description have change format.'
967 raise ValueError('Jedi function parameter description have change format.'
968 'Expected "param ...", found %r".' % description)
968 'Expected "param ...", found %r".' % description)
969 return description[6:]
969 return description[6:]
970
970
971 def _make_signature(completion)-> str:
971 def _make_signature(completion)-> str:
972 """
972 """
973 Make the signature from a jedi completion
973 Make the signature from a jedi completion
974
974
975 Parameter
975 Parameter
976 =========
976 =========
977
977
978 completion: jedi.Completion
978 completion: jedi.Completion
979 object does not complete a function type
979 object does not complete a function type
980
980
981 Returns
981 Returns
982 =======
982 =======
983
983
984 a string consisting of the function signature, with the parenthesis but
984 a string consisting of the function signature, with the parenthesis but
985 without the function name. example:
985 without the function name. example:
986 `(a, *args, b=1, **kwargs)`
986 `(a, *args, b=1, **kwargs)`
987
987
988 """
988 """
989
989
990 return '(%s)'% ', '.join([f for f in (_formatparamchildren(p) for p in completion.params) if f])
990 return '(%s)'% ', '.join([f for f in (_formatparamchildren(p) for p in completion.params) if f])
991
991
992 class IPCompleter(Completer):
992 class IPCompleter(Completer):
993 """Extension of the completer class with IPython-specific features"""
993 """Extension of the completer class with IPython-specific features"""
994
994
995 @observe('greedy')
995 @observe('greedy')
996 def _greedy_changed(self, change):
996 def _greedy_changed(self, change):
997 """update the splitter and readline delims when greedy is changed"""
997 """update the splitter and readline delims when greedy is changed"""
998 if change['new']:
998 if change['new']:
999 self.splitter.delims = GREEDY_DELIMS
999 self.splitter.delims = GREEDY_DELIMS
1000 else:
1000 else:
1001 self.splitter.delims = DELIMS
1001 self.splitter.delims = DELIMS
1002
1002
1003 dict_keys_only = Bool(False,
1003 dict_keys_only = Bool(False,
1004 help="""Whether to show dict key matches only""")
1004 help="""Whether to show dict key matches only""")
1005
1005
1006 merge_completions = Bool(True,
1006 merge_completions = Bool(True,
1007 help="""Whether to merge completion results into a single list
1007 help="""Whether to merge completion results into a single list
1008
1008
1009 If False, only the completion results from the first non-empty
1009 If False, only the completion results from the first non-empty
1010 completer will be returned.
1010 completer will be returned.
1011 """
1011 """
1012 ).tag(config=True)
1012 ).tag(config=True)
1013 omit__names = Enum((0,1,2), default_value=2,
1013 omit__names = Enum((0,1,2), default_value=2,
1014 help="""Instruct the completer to omit private method names
1014 help="""Instruct the completer to omit private method names
1015
1015
1016 Specifically, when completing on ``object.<tab>``.
1016 Specifically, when completing on ``object.<tab>``.
1017
1017
1018 When 2 [default]: all names that start with '_' will be excluded.
1018 When 2 [default]: all names that start with '_' will be excluded.
1019
1019
1020 When 1: all 'magic' names (``__foo__``) will be excluded.
1020 When 1: all 'magic' names (``__foo__``) will be excluded.
1021
1021
1022 When 0: nothing will be excluded.
1022 When 0: nothing will be excluded.
1023 """
1023 """
1024 ).tag(config=True)
1024 ).tag(config=True)
1025 limit_to__all__ = Bool(False,
1025 limit_to__all__ = Bool(False,
1026 help="""
1026 help="""
1027 DEPRECATED as of version 5.0.
1027 DEPRECATED as of version 5.0.
1028
1028
1029 Instruct the completer to use __all__ for the completion
1029 Instruct the completer to use __all__ for the completion
1030
1030
1031 Specifically, when completing on ``object.<tab>``.
1031 Specifically, when completing on ``object.<tab>``.
1032
1032
1033 When True: only those names in obj.__all__ will be included.
1033 When True: only those names in obj.__all__ will be included.
1034
1034
1035 When False [default]: the __all__ attribute is ignored
1035 When False [default]: the __all__ attribute is ignored
1036 """,
1036 """,
1037 ).tag(config=True)
1037 ).tag(config=True)
1038
1038
1039 @observe('limit_to__all__')
1039 @observe('limit_to__all__')
1040 def _limit_to_all_changed(self, change):
1040 def _limit_to_all_changed(self, change):
1041 warnings.warn('`IPython.core.IPCompleter.limit_to__all__` configuration '
1041 warnings.warn('`IPython.core.IPCompleter.limit_to__all__` configuration '
1042 'value has been deprecated since IPython 5.0, will be made to have '
1042 'value has been deprecated since IPython 5.0, will be made to have '
1043 'no effects and then removed in future version of IPython.',
1043 'no effects and then removed in future version of IPython.',
1044 UserWarning)
1044 UserWarning)
1045
1045
1046 def __init__(self, shell=None, namespace=None, global_namespace=None,
1046 def __init__(self, shell=None, namespace=None, global_namespace=None,
1047 use_readline=_deprecation_readline_sentinel, config=None, **kwargs):
1047 use_readline=_deprecation_readline_sentinel, config=None, **kwargs):
1048 """IPCompleter() -> completer
1048 """IPCompleter() -> completer
1049
1049
1050 Return a completer object.
1050 Return a completer object.
1051
1051
1052 Parameters
1052 Parameters
1053 ----------
1053 ----------
1054
1054
1055 shell
1055 shell
1056 a pointer to the ipython shell itself. This is needed
1056 a pointer to the ipython shell itself. This is needed
1057 because this completer knows about magic functions, and those can
1057 because this completer knows about magic functions, and those can
1058 only be accessed via the ipython instance.
1058 only be accessed via the ipython instance.
1059
1059
1060 namespace : dict, optional
1060 namespace : dict, optional
1061 an optional dict where completions are performed.
1061 an optional dict where completions are performed.
1062
1062
1063 global_namespace : dict, optional
1063 global_namespace : dict, optional
1064 secondary optional dict for completions, to
1064 secondary optional dict for completions, to
1065 handle cases (such as IPython embedded inside functions) where
1065 handle cases (such as IPython embedded inside functions) where
1066 both Python scopes are visible.
1066 both Python scopes are visible.
1067
1067
1068 use_readline : bool, optional
1068 use_readline : bool, optional
1069 DEPRECATED, ignored since IPython 6.0, will have no effects
1069 DEPRECATED, ignored since IPython 6.0, will have no effects
1070 """
1070 """
1071
1071
1072 self.magic_escape = ESC_MAGIC
1072 self.magic_escape = ESC_MAGIC
1073 self.splitter = CompletionSplitter()
1073 self.splitter = CompletionSplitter()
1074
1074
1075 if use_readline is not _deprecation_readline_sentinel:
1075 if use_readline is not _deprecation_readline_sentinel:
1076 warnings.warn('The `use_readline` parameter is deprecated and ignored since IPython 6.0.',
1076 warnings.warn('The `use_readline` parameter is deprecated and ignored since IPython 6.0.',
1077 DeprecationWarning, stacklevel=2)
1077 DeprecationWarning, stacklevel=2)
1078
1078
1079 # _greedy_changed() depends on splitter and readline being defined:
1079 # _greedy_changed() depends on splitter and readline being defined:
1080 Completer.__init__(self, namespace=namespace, global_namespace=global_namespace,
1080 Completer.__init__(self, namespace=namespace, global_namespace=global_namespace,
1081 config=config, **kwargs)
1081 config=config, **kwargs)
1082
1082
1083 # List where completion matches will be stored
1083 # List where completion matches will be stored
1084 self.matches = []
1084 self.matches = []
1085 self.shell = shell
1085 self.shell = shell
1086 # Regexp to split filenames with spaces in them
1086 # Regexp to split filenames with spaces in them
1087 self.space_name_re = re.compile(r'([^\\] )')
1087 self.space_name_re = re.compile(r'([^\\] )')
1088 # Hold a local ref. to glob.glob for speed
1088 # Hold a local ref. to glob.glob for speed
1089 self.glob = glob.glob
1089 self.glob = glob.glob
1090
1090
1091 # Determine if we are running on 'dumb' terminals, like (X)Emacs
1091 # Determine if we are running on 'dumb' terminals, like (X)Emacs
1092 # buffers, to avoid completion problems.
1092 # buffers, to avoid completion problems.
1093 term = os.environ.get('TERM','xterm')
1093 term = os.environ.get('TERM','xterm')
1094 self.dumb_terminal = term in ['dumb','emacs']
1094 self.dumb_terminal = term in ['dumb','emacs']
1095
1095
1096 # Special handling of backslashes needed in win32 platforms
1096 # Special handling of backslashes needed in win32 platforms
1097 if sys.platform == "win32":
1097 if sys.platform == "win32":
1098 self.clean_glob = self._clean_glob_win32
1098 self.clean_glob = self._clean_glob_win32
1099 else:
1099 else:
1100 self.clean_glob = self._clean_glob
1100 self.clean_glob = self._clean_glob
1101
1101
1102 #regexp to parse docstring for function signature
1102 #regexp to parse docstring for function signature
1103 self.docstring_sig_re = re.compile(r'^[\w|\s.]+\(([^)]*)\).*')
1103 self.docstring_sig_re = re.compile(r'^[\w|\s.]+\(([^)]*)\).*')
1104 self.docstring_kwd_re = re.compile(r'[\s|\[]*(\w+)(?:\s*=\s*.*)')
1104 self.docstring_kwd_re = re.compile(r'[\s|\[]*(\w+)(?:\s*=\s*.*)')
1105 #use this if positional argument name is also needed
1105 #use this if positional argument name is also needed
1106 #= re.compile(r'[\s|\[]*(\w+)(?:\s*=?\s*.*)')
1106 #= re.compile(r'[\s|\[]*(\w+)(?:\s*=?\s*.*)')
1107
1107
1108 self.magic_arg_matchers = [
1108 self.magic_arg_matchers = [
1109 self.magic_config_matches,
1109 self.magic_config_matches,
1110 self.magic_color_matches,
1110 self.magic_color_matches,
1111 ]
1111 ]
1112
1112
1113 # This is set externally by InteractiveShell
1113 # This is set externally by InteractiveShell
1114 self.custom_completers = None
1114 self.custom_completers = None
1115
1115
1116 @property
1116 @property
1117 def matchers(self):
1117 def matchers(self):
1118 """All active matcher routines for completion"""
1118 """All active matcher routines for completion"""
1119 if self.dict_keys_only:
1119 if self.dict_keys_only:
1120 return [self.dict_key_matches]
1120 return [self.dict_key_matches]
1121
1121
1122 if self.use_jedi:
1122 if self.use_jedi:
1123 return [
1123 return [
1124 self.file_matches,
1124 self.file_matches,
1125 self.magic_matches,
1125 self.magic_matches,
1126 self.dict_key_matches,
1126 self.dict_key_matches,
1127 ]
1127 ]
1128 else:
1128 else:
1129 return [
1129 return [
1130 self.python_matches,
1130 self.python_matches,
1131 self.file_matches,
1131 self.file_matches,
1132 self.magic_matches,
1132 self.magic_matches,
1133 self.python_func_kw_matches,
1133 self.python_func_kw_matches,
1134 self.dict_key_matches,
1134 self.dict_key_matches,
1135 ]
1135 ]
1136
1136
1137 def all_completions(self, text):
1137 def all_completions(self, text):
1138 """
1138 """
1139 Wrapper around the complete method for the benefit of emacs.
1139 Wrapper around the complete method for the benefit of emacs.
1140 """
1140 """
1141 return self.complete(text)[1]
1141 return self.complete(text)[1]
1142
1142
1143 def _clean_glob(self, text):
1143 def _clean_glob(self, text):
1144 return self.glob("%s*" % text)
1144 return self.glob("%s*" % text)
1145
1145
1146 def _clean_glob_win32(self,text):
1146 def _clean_glob_win32(self,text):
1147 return [f.replace("\\","/")
1147 return [f.replace("\\","/")
1148 for f in self.glob("%s*" % text)]
1148 for f in self.glob("%s*" % text)]
1149
1149
1150 def file_matches(self, text):
1150 def file_matches(self, text):
1151 """Match filenames, expanding ~USER type strings.
1151 """Match filenames, expanding ~USER type strings.
1152
1152
1153 Most of the seemingly convoluted logic in this completer is an
1153 Most of the seemingly convoluted logic in this completer is an
1154 attempt to handle filenames with spaces in them. And yet it's not
1154 attempt to handle filenames with spaces in them. And yet it's not
1155 quite perfect, because Python's readline doesn't expose all of the
1155 quite perfect, because Python's readline doesn't expose all of the
1156 GNU readline details needed for this to be done correctly.
1156 GNU readline details needed for this to be done correctly.
1157
1157
1158 For a filename with a space in it, the printed completions will be
1158 For a filename with a space in it, the printed completions will be
1159 only the parts after what's already been typed (instead of the
1159 only the parts after what's already been typed (instead of the
1160 full completions, as is normally done). I don't think with the
1160 full completions, as is normally done). I don't think with the
1161 current (as of Python 2.3) Python readline it's possible to do
1161 current (as of Python 2.3) Python readline it's possible to do
1162 better."""
1162 better."""
1163
1163
1164 # chars that require escaping with backslash - i.e. chars
1164 # chars that require escaping with backslash - i.e. chars
1165 # that readline treats incorrectly as delimiters, but we
1165 # that readline treats incorrectly as delimiters, but we
1166 # don't want to treat as delimiters in filename matching
1166 # don't want to treat as delimiters in filename matching
1167 # when escaped with backslash
1167 # when escaped with backslash
1168 if text.startswith('!'):
1168 if text.startswith('!'):
1169 text = text[1:]
1169 text = text[1:]
1170 text_prefix = u'!'
1170 text_prefix = u'!'
1171 else:
1171 else:
1172 text_prefix = u''
1172 text_prefix = u''
1173
1173
1174 text_until_cursor = self.text_until_cursor
1174 text_until_cursor = self.text_until_cursor
1175 # track strings with open quotes
1175 # track strings with open quotes
1176 open_quotes = has_open_quotes(text_until_cursor)
1176 open_quotes = has_open_quotes(text_until_cursor)
1177
1177
1178 if '(' in text_until_cursor or '[' in text_until_cursor:
1178 if '(' in text_until_cursor or '[' in text_until_cursor:
1179 lsplit = text
1179 lsplit = text
1180 else:
1180 else:
1181 try:
1181 try:
1182 # arg_split ~ shlex.split, but with unicode bugs fixed by us
1182 # arg_split ~ shlex.split, but with unicode bugs fixed by us
1183 lsplit = arg_split(text_until_cursor)[-1]
1183 lsplit = arg_split(text_until_cursor)[-1]
1184 except ValueError:
1184 except ValueError:
1185 # typically an unmatched ", or backslash without escaped char.
1185 # typically an unmatched ", or backslash without escaped char.
1186 if open_quotes:
1186 if open_quotes:
1187 lsplit = text_until_cursor.split(open_quotes)[-1]
1187 lsplit = text_until_cursor.split(open_quotes)[-1]
1188 else:
1188 else:
1189 return []
1189 return []
1190 except IndexError:
1190 except IndexError:
1191 # tab pressed on empty line
1191 # tab pressed on empty line
1192 lsplit = ""
1192 lsplit = ""
1193
1193
1194 if not open_quotes and lsplit != protect_filename(lsplit):
1194 if not open_quotes and lsplit != protect_filename(lsplit):
1195 # if protectables are found, do matching on the whole escaped name
1195 # if protectables are found, do matching on the whole escaped name
1196 has_protectables = True
1196 has_protectables = True
1197 text0,text = text,lsplit
1197 text0,text = text,lsplit
1198 else:
1198 else:
1199 has_protectables = False
1199 has_protectables = False
1200 text = os.path.expanduser(text)
1200 text = os.path.expanduser(text)
1201
1201
1202 if text == "":
1202 if text == "":
1203 return [text_prefix + protect_filename(f) for f in self.glob("*")]
1203 return [text_prefix + protect_filename(f) for f in self.glob("*")]
1204
1204
1205 # Compute the matches from the filesystem
1205 # Compute the matches from the filesystem
1206 if sys.platform == 'win32':
1206 if sys.platform == 'win32':
1207 m0 = self.clean_glob(text)
1207 m0 = self.clean_glob(text)
1208 else:
1208 else:
1209 m0 = self.clean_glob(text.replace('\\', ''))
1209 m0 = self.clean_glob(text.replace('\\', ''))
1210
1210
1211 if has_protectables:
1211 if has_protectables:
1212 # If we had protectables, we need to revert our changes to the
1212 # If we had protectables, we need to revert our changes to the
1213 # beginning of filename so that we don't double-write the part
1213 # beginning of filename so that we don't double-write the part
1214 # of the filename we have so far
1214 # of the filename we have so far
1215 len_lsplit = len(lsplit)
1215 len_lsplit = len(lsplit)
1216 matches = [text_prefix + text0 +
1216 matches = [text_prefix + text0 +
1217 protect_filename(f[len_lsplit:]) for f in m0]
1217 protect_filename(f[len_lsplit:]) for f in m0]
1218 else:
1218 else:
1219 if open_quotes:
1219 if open_quotes:
1220 # if we have a string with an open quote, we don't need to
1220 # if we have a string with an open quote, we don't need to
1221 # protect the names beyond the quote (and we _shouldn't_, as
1221 # protect the names beyond the quote (and we _shouldn't_, as
1222 # it would cause bugs when the filesystem call is made).
1222 # it would cause bugs when the filesystem call is made).
1223 matches = m0 if sys.platform == "win32" else\
1223 matches = m0 if sys.platform == "win32" else\
1224 [protect_filename(f, open_quotes) for f in m0]
1224 [protect_filename(f, open_quotes) for f in m0]
1225 else:
1225 else:
1226 matches = [text_prefix +
1226 matches = [text_prefix +
1227 protect_filename(f) for f in m0]
1227 protect_filename(f) for f in m0]
1228
1228
1229 # Mark directories in input list by appending '/' to their names.
1229 # Mark directories in input list by appending '/' to their names.
1230 return [x+'/' if os.path.isdir(x) else x for x in matches]
1230 return [x+'/' if os.path.isdir(x) else x for x in matches]
1231
1231
1232 def magic_matches(self, text):
1232 def magic_matches(self, text):
1233 """Match magics"""
1233 """Match magics"""
1234 # Get all shell magics now rather than statically, so magics loaded at
1234 # Get all shell magics now rather than statically, so magics loaded at
1235 # runtime show up too.
1235 # runtime show up too.
1236 lsm = self.shell.magics_manager.lsmagic()
1236 lsm = self.shell.magics_manager.lsmagic()
1237 line_magics = lsm['line']
1237 line_magics = lsm['line']
1238 cell_magics = lsm['cell']
1238 cell_magics = lsm['cell']
1239 pre = self.magic_escape
1239 pre = self.magic_escape
1240 pre2 = pre+pre
1240 pre2 = pre+pre
1241
1241
1242 explicit_magic = text.startswith(pre)
1242 explicit_magic = text.startswith(pre)
1243
1243
1244 # Completion logic:
1244 # Completion logic:
1245 # - user gives %%: only do cell magics
1245 # - user gives %%: only do cell magics
1246 # - user gives %: do both line and cell magics
1246 # - user gives %: do both line and cell magics
1247 # - no prefix: do both
1247 # - no prefix: do both
1248 # In other words, line magics are skipped if the user gives %% explicitly
1248 # In other words, line magics are skipped if the user gives %% explicitly
1249 #
1249 #
1250 # We also exclude magics that match any currently visible names:
1250 # We also exclude magics that match any currently visible names:
1251 # https://github.com/ipython/ipython/issues/4877, unless the user has
1251 # https://github.com/ipython/ipython/issues/4877, unless the user has
1252 # typed a %:
1252 # typed a %:
1253 # https://github.com/ipython/ipython/issues/10754
1253 # https://github.com/ipython/ipython/issues/10754
1254 bare_text = text.lstrip(pre)
1254 bare_text = text.lstrip(pre)
1255 global_matches = self.global_matches(bare_text)
1255 global_matches = self.global_matches(bare_text)
1256 if not explicit_magic:
1256 if not explicit_magic:
1257 def matches(magic):
1257 def matches(magic):
1258 """
1258 """
1259 Filter magics, in particular remove magics that match
1259 Filter magics, in particular remove magics that match
1260 a name present in global namespace.
1260 a name present in global namespace.
1261 """
1261 """
1262 return ( magic.startswith(bare_text) and
1262 return ( magic.startswith(bare_text) and
1263 magic not in global_matches )
1263 magic not in global_matches )
1264 else:
1264 else:
1265 def matches(magic):
1265 def matches(magic):
1266 return magic.startswith(bare_text)
1266 return magic.startswith(bare_text)
1267
1267
1268 comp = [ pre2+m for m in cell_magics if matches(m)]
1268 comp = [ pre2+m for m in cell_magics if matches(m)]
1269 if not text.startswith(pre2):
1269 if not text.startswith(pre2):
1270 comp += [ pre+m for m in line_magics if matches(m)]
1270 comp += [ pre+m for m in line_magics if matches(m)]
1271
1271
1272 return comp
1272 return comp
1273
1273
1274 def magic_config_matches(self, text:str) -> List[str]:
1274 def magic_config_matches(self, text:str) -> List[str]:
1275 """ Match class names and attributes for %config magic """
1275 """ Match class names and attributes for %config magic """
1276 texts = text.strip().split()
1276 texts = text.strip().split()
1277
1277
1278 if len(texts) > 0 and (texts[0] == 'config' or texts[0] == '%config'):
1278 if len(texts) > 0 and (texts[0] == 'config' or texts[0] == '%config'):
1279 # get all configuration classes
1279 # get all configuration classes
1280 classes = sorted(set([ c for c in self.shell.configurables
1280 classes = sorted(set([ c for c in self.shell.configurables
1281 if c.__class__.class_traits(config=True)
1281 if c.__class__.class_traits(config=True)
1282 ]), key=lambda x: x.__class__.__name__)
1282 ]), key=lambda x: x.__class__.__name__)
1283 classnames = [ c.__class__.__name__ for c in classes ]
1283 classnames = [ c.__class__.__name__ for c in classes ]
1284
1284
1285 # return all classnames if config or %config is given
1285 # return all classnames if config or %config is given
1286 if len(texts) == 1:
1286 if len(texts) == 1:
1287 return classnames
1287 return classnames
1288
1288
1289 # match classname
1289 # match classname
1290 classname_texts = texts[1].split('.')
1290 classname_texts = texts[1].split('.')
1291 classname = classname_texts[0]
1291 classname = classname_texts[0]
1292 classname_matches = [ c for c in classnames
1292 classname_matches = [ c for c in classnames
1293 if c.startswith(classname) ]
1293 if c.startswith(classname) ]
1294
1294
1295 # return matched classes or the matched class with attributes
1295 # return matched classes or the matched class with attributes
1296 if texts[1].find('.') < 0:
1296 if texts[1].find('.') < 0:
1297 return classname_matches
1297 return classname_matches
1298 elif len(classname_matches) == 1 and \
1298 elif len(classname_matches) == 1 and \
1299 classname_matches[0] == classname:
1299 classname_matches[0] == classname:
1300 cls = classes[classnames.index(classname)].__class__
1300 cls = classes[classnames.index(classname)].__class__
1301 help = cls.class_get_help()
1301 help = cls.class_get_help()
1302 # strip leading '--' from cl-args:
1302 # strip leading '--' from cl-args:
1303 help = re.sub(re.compile(r'^--', re.MULTILINE), '', help)
1303 help = re.sub(re.compile(r'^--', re.MULTILINE), '', help)
1304 return [ attr.split('=')[0]
1304 return [ attr.split('=')[0]
1305 for attr in help.strip().splitlines()
1305 for attr in help.strip().splitlines()
1306 if attr.startswith(texts[1]) ]
1306 if attr.startswith(texts[1]) ]
1307 return []
1307 return []
1308
1308
1309 def magic_color_matches(self, text:str) -> List[str] :
1309 def magic_color_matches(self, text:str) -> List[str] :
1310 """ Match color schemes for %colors magic"""
1310 """ Match color schemes for %colors magic"""
1311 texts = text.split()
1311 texts = text.split()
1312 if text.endswith(' '):
1312 if text.endswith(' '):
1313 # .split() strips off the trailing whitespace. Add '' back
1313 # .split() strips off the trailing whitespace. Add '' back
1314 # so that: '%colors ' -> ['%colors', '']
1314 # so that: '%colors ' -> ['%colors', '']
1315 texts.append('')
1315 texts.append('')
1316
1316
1317 if len(texts) == 2 and (texts[0] == 'colors' or texts[0] == '%colors'):
1317 if len(texts) == 2 and (texts[0] == 'colors' or texts[0] == '%colors'):
1318 prefix = texts[1]
1318 prefix = texts[1]
1319 return [ color for color in InspectColors.keys()
1319 return [ color for color in InspectColors.keys()
1320 if color.startswith(prefix) ]
1320 if color.startswith(prefix) ]
1321 return []
1321 return []
1322
1322
1323 def _jedi_matches(self, cursor_column:int, cursor_line:int, text:str):
1323 def _jedi_matches(self, cursor_column:int, cursor_line:int, text:str):
1324 """
1324 """
1325
1325
1326 Return a list of :any:`jedi.api.Completions` object from a ``text`` and
1326 Return a list of :any:`jedi.api.Completions` object from a ``text`` and
1327 cursor position.
1327 cursor position.
1328
1328
1329 Parameters
1329 Parameters
1330 ----------
1330 ----------
1331 cursor_column : int
1331 cursor_column : int
1332 column position of the cursor in ``text``, 0-indexed.
1332 column position of the cursor in ``text``, 0-indexed.
1333 cursor_line : int
1333 cursor_line : int
1334 line position of the cursor in ``text``, 0-indexed
1334 line position of the cursor in ``text``, 0-indexed
1335 text : str
1335 text : str
1336 text to complete
1336 text to complete
1337
1337
1338 Debugging
1338 Debugging
1339 ---------
1339 ---------
1340
1340
1341 If ``IPCompleter.debug`` is ``True`` may return a :any:`_FakeJediCompletion`
1341 If ``IPCompleter.debug`` is ``True`` may return a :any:`_FakeJediCompletion`
1342 object containing a string with the Jedi debug information attached.
1342 object containing a string with the Jedi debug information attached.
1343 """
1343 """
1344 namespaces = [self.namespace]
1344 namespaces = [self.namespace]
1345 if self.global_namespace is not None:
1345 if self.global_namespace is not None:
1346 namespaces.append(self.global_namespace)
1346 namespaces.append(self.global_namespace)
1347
1347
1348 completion_filter = lambda x:x
1348 completion_filter = lambda x:x
1349 offset = cursor_to_position(text, cursor_line, cursor_column)
1349 offset = cursor_to_position(text, cursor_line, cursor_column)
1350 # filter output if we are completing for object members
1350 # filter output if we are completing for object members
1351 if offset:
1351 if offset:
1352 pre = text[offset-1]
1352 pre = text[offset-1]
1353 if pre == '.':
1353 if pre == '.':
1354 if self.omit__names == 2:
1354 if self.omit__names == 2:
1355 completion_filter = lambda c:not c.name.startswith('_')
1355 completion_filter = lambda c:not c.name.startswith('_')
1356 elif self.omit__names == 1:
1356 elif self.omit__names == 1:
1357 completion_filter = lambda c:not (c.name.startswith('__') and c.name.endswith('__'))
1357 completion_filter = lambda c:not (c.name.startswith('__') and c.name.endswith('__'))
1358 elif self.omit__names == 0:
1358 elif self.omit__names == 0:
1359 completion_filter = lambda x:x
1359 completion_filter = lambda x:x
1360 else:
1360 else:
1361 raise ValueError("Don't understand self.omit__names == {}".format(self.omit__names))
1361 raise ValueError("Don't understand self.omit__names == {}".format(self.omit__names))
1362
1362
1363 interpreter = jedi.Interpreter(
1363 interpreter = jedi.Interpreter(
1364 text[:offset], namespaces, column=cursor_column, line=cursor_line + 1)
1364 text[:offset], namespaces, column=cursor_column, line=cursor_line + 1)
1365 try_jedi = True
1365 try_jedi = True
1366
1366
1367 try:
1367 try:
1368 # should we check the type of the node is Error ?
1368 # should we check the type of the node is Error ?
1369 try:
1369 try:
1370 # jedi < 0.11
1370 # jedi < 0.11
1371 from jedi.parser.tree import ErrorLeaf
1371 from jedi.parser.tree import ErrorLeaf
1372 except ImportError:
1372 except ImportError:
1373 # jedi >= 0.11
1373 # jedi >= 0.11
1374 from parso.tree import ErrorLeaf
1374 from parso.tree import ErrorLeaf
1375
1375
1376 next_to_last_tree = interpreter._get_module().tree_node.children[-2]
1376 next_to_last_tree = interpreter._get_module().tree_node.children[-2]
1377 completing_string = False
1377 completing_string = False
1378 if isinstance(next_to_last_tree, ErrorLeaf):
1378 if isinstance(next_to_last_tree, ErrorLeaf):
1379 completing_string = next_to_last_tree.value.lstrip()[0] in {'"', "'"}
1379 completing_string = next_to_last_tree.value.lstrip()[0] in {'"', "'"}
1380 # if we are in a string jedi is likely not the right candidate for
1380 # if we are in a string jedi is likely not the right candidate for
1381 # now. Skip it.
1381 # now. Skip it.
1382 try_jedi = not completing_string
1382 try_jedi = not completing_string
1383 except Exception as e:
1383 except Exception as e:
1384 # many of things can go wrong, we are using private API just don't crash.
1384 # many of things can go wrong, we are using private API just don't crash.
1385 if self.debug:
1385 if self.debug:
1386 print("Error detecting if completing a non-finished string :", e, '|')
1386 print("Error detecting if completing a non-finished string :", e, '|')
1387
1387
1388 if not try_jedi:
1388 if not try_jedi:
1389 return []
1389 return []
1390 try:
1390 try:
1391 return filter(completion_filter, interpreter.completions())
1391 return filter(completion_filter, interpreter.completions())
1392 except Exception as e:
1392 except Exception as e:
1393 if self.debug:
1393 if self.debug:
1394 return [_FakeJediCompletion('Oops Jedi has crashed, please report a bug with the following:\n"""\n%s\ns"""' % (e))]
1394 return [_FakeJediCompletion('Oops Jedi has crashed, please report a bug with the following:\n"""\n%s\ns"""' % (e))]
1395 else:
1395 else:
1396 return []
1396 return []
1397
1397
1398 def python_matches(self, text):
1398 def python_matches(self, text):
1399 """Match attributes or global python names"""
1399 """Match attributes or global python names"""
1400 if "." in text:
1400 if "." in text:
1401 try:
1401 try:
1402 matches = self.attr_matches(text)
1402 matches = self.attr_matches(text)
1403 if text.endswith('.') and self.omit__names:
1403 if text.endswith('.') and self.omit__names:
1404 if self.omit__names == 1:
1404 if self.omit__names == 1:
1405 # true if txt is _not_ a __ name, false otherwise:
1405 # true if txt is _not_ a __ name, false otherwise:
1406 no__name = (lambda txt:
1406 no__name = (lambda txt:
1407 re.match(r'.*\.__.*?__',txt) is None)
1407 re.match(r'.*\.__.*?__',txt) is None)
1408 else:
1408 else:
1409 # true if txt is _not_ a _ name, false otherwise:
1409 # true if txt is _not_ a _ name, false otherwise:
1410 no__name = (lambda txt:
1410 no__name = (lambda txt:
1411 re.match(r'\._.*?',txt[txt.rindex('.'):]) is None)
1411 re.match(r'\._.*?',txt[txt.rindex('.'):]) is None)
1412 matches = filter(no__name, matches)
1412 matches = filter(no__name, matches)
1413 except NameError:
1413 except NameError:
1414 # catches <undefined attributes>.<tab>
1414 # catches <undefined attributes>.<tab>
1415 matches = []
1415 matches = []
1416 else:
1416 else:
1417 matches = self.global_matches(text)
1417 matches = self.global_matches(text)
1418 return matches
1418 return matches
1419
1419
1420 def _default_arguments_from_docstring(self, doc):
1420 def _default_arguments_from_docstring(self, doc):
1421 """Parse the first line of docstring for call signature.
1421 """Parse the first line of docstring for call signature.
1422
1422
1423 Docstring should be of the form 'min(iterable[, key=func])\n'.
1423 Docstring should be of the form 'min(iterable[, key=func])\n'.
1424 It can also parse cython docstring of the form
1424 It can also parse cython docstring of the form
1425 'Minuit.migrad(self, int ncall=10000, resume=True, int nsplit=1)'.
1425 'Minuit.migrad(self, int ncall=10000, resume=True, int nsplit=1)'.
1426 """
1426 """
1427 if doc is None:
1427 if doc is None:
1428 return []
1428 return []
1429
1429
1430 #care only the firstline
1430 #care only the firstline
1431 line = doc.lstrip().splitlines()[0]
1431 line = doc.lstrip().splitlines()[0]
1432
1432
1433 #p = re.compile(r'^[\w|\s.]+\(([^)]*)\).*')
1433 #p = re.compile(r'^[\w|\s.]+\(([^)]*)\).*')
1434 #'min(iterable[, key=func])\n' -> 'iterable[, key=func]'
1434 #'min(iterable[, key=func])\n' -> 'iterable[, key=func]'
1435 sig = self.docstring_sig_re.search(line)
1435 sig = self.docstring_sig_re.search(line)
1436 if sig is None:
1436 if sig is None:
1437 return []
1437 return []
1438 # iterable[, key=func]' -> ['iterable[' ,' key=func]']
1438 # iterable[, key=func]' -> ['iterable[' ,' key=func]']
1439 sig = sig.groups()[0].split(',')
1439 sig = sig.groups()[0].split(',')
1440 ret = []
1440 ret = []
1441 for s in sig:
1441 for s in sig:
1442 #re.compile(r'[\s|\[]*(\w+)(?:\s*=\s*.*)')
1442 #re.compile(r'[\s|\[]*(\w+)(?:\s*=\s*.*)')
1443 ret += self.docstring_kwd_re.findall(s)
1443 ret += self.docstring_kwd_re.findall(s)
1444 return ret
1444 return ret
1445
1445
1446 def _default_arguments(self, obj):
1446 def _default_arguments(self, obj):
1447 """Return the list of default arguments of obj if it is callable,
1447 """Return the list of default arguments of obj if it is callable,
1448 or empty list otherwise."""
1448 or empty list otherwise."""
1449 call_obj = obj
1449 call_obj = obj
1450 ret = []
1450 ret = []
1451 if inspect.isbuiltin(obj):
1451 if inspect.isbuiltin(obj):
1452 pass
1452 pass
1453 elif not (inspect.isfunction(obj) or inspect.ismethod(obj)):
1453 elif not (inspect.isfunction(obj) or inspect.ismethod(obj)):
1454 if inspect.isclass(obj):
1454 if inspect.isclass(obj):
1455 #for cython embedsignature=True the constructor docstring
1455 #for cython embedsignature=True the constructor docstring
1456 #belongs to the object itself not __init__
1456 #belongs to the object itself not __init__
1457 ret += self._default_arguments_from_docstring(
1457 ret += self._default_arguments_from_docstring(
1458 getattr(obj, '__doc__', ''))
1458 getattr(obj, '__doc__', ''))
1459 # for classes, check for __init__,__new__
1459 # for classes, check for __init__,__new__
1460 call_obj = (getattr(obj, '__init__', None) or
1460 call_obj = (getattr(obj, '__init__', None) or
1461 getattr(obj, '__new__', None))
1461 getattr(obj, '__new__', None))
1462 # for all others, check if they are __call__able
1462 # for all others, check if they are __call__able
1463 elif hasattr(obj, '__call__'):
1463 elif hasattr(obj, '__call__'):
1464 call_obj = obj.__call__
1464 call_obj = obj.__call__
1465 ret += self._default_arguments_from_docstring(
1465 ret += self._default_arguments_from_docstring(
1466 getattr(call_obj, '__doc__', ''))
1466 getattr(call_obj, '__doc__', ''))
1467
1467
1468 _keeps = (inspect.Parameter.KEYWORD_ONLY,
1468 _keeps = (inspect.Parameter.KEYWORD_ONLY,
1469 inspect.Parameter.POSITIONAL_OR_KEYWORD)
1469 inspect.Parameter.POSITIONAL_OR_KEYWORD)
1470
1470
1471 try:
1471 try:
1472 sig = inspect.signature(call_obj)
1472 sig = inspect.signature(call_obj)
1473 ret.extend(k for k, v in sig.parameters.items() if
1473 ret.extend(k for k, v in sig.parameters.items() if
1474 v.kind in _keeps)
1474 v.kind in _keeps)
1475 except ValueError:
1475 except ValueError:
1476 pass
1476 pass
1477
1477
1478 return list(set(ret))
1478 return list(set(ret))
1479
1479
1480 def python_func_kw_matches(self,text):
1480 def python_func_kw_matches(self,text):
1481 """Match named parameters (kwargs) of the last open function"""
1481 """Match named parameters (kwargs) of the last open function"""
1482
1482
1483 if "." in text: # a parameter cannot be dotted
1483 if "." in text: # a parameter cannot be dotted
1484 return []
1484 return []
1485 try: regexp = self.__funcParamsRegex
1485 try: regexp = self.__funcParamsRegex
1486 except AttributeError:
1486 except AttributeError:
1487 regexp = self.__funcParamsRegex = re.compile(r'''
1487 regexp = self.__funcParamsRegex = re.compile(r'''
1488 '.*?(?<!\\)' | # single quoted strings or
1488 '.*?(?<!\\)' | # single quoted strings or
1489 ".*?(?<!\\)" | # double quoted strings or
1489 ".*?(?<!\\)" | # double quoted strings or
1490 \w+ | # identifier
1490 \w+ | # identifier
1491 \S # other characters
1491 \S # other characters
1492 ''', re.VERBOSE | re.DOTALL)
1492 ''', re.VERBOSE | re.DOTALL)
1493 # 1. find the nearest identifier that comes before an unclosed
1493 # 1. find the nearest identifier that comes before an unclosed
1494 # parenthesis before the cursor
1494 # parenthesis before the cursor
1495 # e.g. for "foo (1+bar(x), pa<cursor>,a=1)", the candidate is "foo"
1495 # e.g. for "foo (1+bar(x), pa<cursor>,a=1)", the candidate is "foo"
1496 tokens = regexp.findall(self.text_until_cursor)
1496 tokens = regexp.findall(self.text_until_cursor)
1497 iterTokens = reversed(tokens); openPar = 0
1497 iterTokens = reversed(tokens); openPar = 0
1498
1498
1499 for token in iterTokens:
1499 for token in iterTokens:
1500 if token == ')':
1500 if token == ')':
1501 openPar -= 1
1501 openPar -= 1
1502 elif token == '(':
1502 elif token == '(':
1503 openPar += 1
1503 openPar += 1
1504 if openPar > 0:
1504 if openPar > 0:
1505 # found the last unclosed parenthesis
1505 # found the last unclosed parenthesis
1506 break
1506 break
1507 else:
1507 else:
1508 return []
1508 return []
1509 # 2. Concatenate dotted names ("foo.bar" for "foo.bar(x, pa" )
1509 # 2. Concatenate dotted names ("foo.bar" for "foo.bar(x, pa" )
1510 ids = []
1510 ids = []
1511 isId = re.compile(r'\w+$').match
1511 isId = re.compile(r'\w+$').match
1512
1512
1513 while True:
1513 while True:
1514 try:
1514 try:
1515 ids.append(next(iterTokens))
1515 ids.append(next(iterTokens))
1516 if not isId(ids[-1]):
1516 if not isId(ids[-1]):
1517 ids.pop(); break
1517 ids.pop(); break
1518 if not next(iterTokens) == '.':
1518 if not next(iterTokens) == '.':
1519 break
1519 break
1520 except StopIteration:
1520 except StopIteration:
1521 break
1521 break
1522
1522
1523 # Find all named arguments already assigned to, as to avoid suggesting
1523 # Find all named arguments already assigned to, as to avoid suggesting
1524 # them again
1524 # them again
1525 usedNamedArgs = set()
1525 usedNamedArgs = set()
1526 par_level = -1
1526 par_level = -1
1527 for token, next_token in zip(tokens, tokens[1:]):
1527 for token, next_token in zip(tokens, tokens[1:]):
1528 if token == '(':
1528 if token == '(':
1529 par_level += 1
1529 par_level += 1
1530 elif token == ')':
1530 elif token == ')':
1531 par_level -= 1
1531 par_level -= 1
1532
1532
1533 if par_level != 0:
1533 if par_level != 0:
1534 continue
1534 continue
1535
1535
1536 if next_token != '=':
1536 if next_token != '=':
1537 continue
1537 continue
1538
1538
1539 usedNamedArgs.add(token)
1539 usedNamedArgs.add(token)
1540
1540
1541 # lookup the candidate callable matches either using global_matches
1542 # or attr_matches for dotted names
1543 if len(ids) == 1:
1544 callableMatches = self.global_matches(ids[0])
1545 else:
1546 callableMatches = self.attr_matches('.'.join(ids[::-1]))
1547 argMatches = []
1541 argMatches = []
1548 for callableMatch in callableMatches:
1542 try:
1549 try:
1543 callableObj = '.'.join(ids[::-1])
1550 namedArgs = self._default_arguments(eval(callableMatch,
1544 namedArgs = self._default_arguments(eval(callableObj,
1551 self.namespace))
1545 self.namespace))
1552 except:
1553 continue
1554
1546
1555 # Remove used named arguments from the list, no need to show twice
1547 # Remove used named arguments from the list, no need to show twice
1556 for namedArg in set(namedArgs) - usedNamedArgs:
1548 for namedArg in set(namedArgs) - usedNamedArgs:
1557 if namedArg.startswith(text):
1549 if namedArg.startswith(text):
1558 argMatches.append(u"%s=" %namedArg)
1550 argMatches.append(u"%s=" %namedArg)
1551 except:
1552 pass
1553
1559 return argMatches
1554 return argMatches
1560
1555
1561 def dict_key_matches(self, text):
1556 def dict_key_matches(self, text):
1562 "Match string keys in a dictionary, after e.g. 'foo[' "
1557 "Match string keys in a dictionary, after e.g. 'foo[' "
1563 def get_keys(obj):
1558 def get_keys(obj):
1564 # Objects can define their own completions by defining an
1559 # Objects can define their own completions by defining an
1565 # _ipy_key_completions_() method.
1560 # _ipy_key_completions_() method.
1566 method = get_real_method(obj, '_ipython_key_completions_')
1561 method = get_real_method(obj, '_ipython_key_completions_')
1567 if method is not None:
1562 if method is not None:
1568 return method()
1563 return method()
1569
1564
1570 # Special case some common in-memory dict-like types
1565 # Special case some common in-memory dict-like types
1571 if isinstance(obj, dict) or\
1566 if isinstance(obj, dict) or\
1572 _safe_isinstance(obj, 'pandas', 'DataFrame'):
1567 _safe_isinstance(obj, 'pandas', 'DataFrame'):
1573 try:
1568 try:
1574 return list(obj.keys())
1569 return list(obj.keys())
1575 except Exception:
1570 except Exception:
1576 return []
1571 return []
1577 elif _safe_isinstance(obj, 'numpy', 'ndarray') or\
1572 elif _safe_isinstance(obj, 'numpy', 'ndarray') or\
1578 _safe_isinstance(obj, 'numpy', 'void'):
1573 _safe_isinstance(obj, 'numpy', 'void'):
1579 return obj.dtype.names or []
1574 return obj.dtype.names or []
1580 return []
1575 return []
1581
1576
1582 try:
1577 try:
1583 regexps = self.__dict_key_regexps
1578 regexps = self.__dict_key_regexps
1584 except AttributeError:
1579 except AttributeError:
1585 dict_key_re_fmt = r'''(?x)
1580 dict_key_re_fmt = r'''(?x)
1586 ( # match dict-referring expression wrt greedy setting
1581 ( # match dict-referring expression wrt greedy setting
1587 %s
1582 %s
1588 )
1583 )
1589 \[ # open bracket
1584 \[ # open bracket
1590 \s* # and optional whitespace
1585 \s* # and optional whitespace
1591 ([uUbB]? # string prefix (r not handled)
1586 ([uUbB]? # string prefix (r not handled)
1592 (?: # unclosed string
1587 (?: # unclosed string
1593 '(?:[^']|(?<!\\)\\')*
1588 '(?:[^']|(?<!\\)\\')*
1594 |
1589 |
1595 "(?:[^"]|(?<!\\)\\")*
1590 "(?:[^"]|(?<!\\)\\")*
1596 )
1591 )
1597 )?
1592 )?
1598 $
1593 $
1599 '''
1594 '''
1600 regexps = self.__dict_key_regexps = {
1595 regexps = self.__dict_key_regexps = {
1601 False: re.compile(dict_key_re_fmt % r'''
1596 False: re.compile(dict_key_re_fmt % r'''
1602 # identifiers separated by .
1597 # identifiers separated by .
1603 (?!\d)\w+
1598 (?!\d)\w+
1604 (?:\.(?!\d)\w+)*
1599 (?:\.(?!\d)\w+)*
1605 '''),
1600 '''),
1606 True: re.compile(dict_key_re_fmt % '''
1601 True: re.compile(dict_key_re_fmt % '''
1607 .+
1602 .+
1608 ''')
1603 ''')
1609 }
1604 }
1610
1605
1611 match = regexps[self.greedy].search(self.text_until_cursor)
1606 match = regexps[self.greedy].search(self.text_until_cursor)
1612 if match is None:
1607 if match is None:
1613 return []
1608 return []
1614
1609
1615 expr, prefix = match.groups()
1610 expr, prefix = match.groups()
1616 try:
1611 try:
1617 obj = eval(expr, self.namespace)
1612 obj = eval(expr, self.namespace)
1618 except Exception:
1613 except Exception:
1619 try:
1614 try:
1620 obj = eval(expr, self.global_namespace)
1615 obj = eval(expr, self.global_namespace)
1621 except Exception:
1616 except Exception:
1622 return []
1617 return []
1623
1618
1624 keys = get_keys(obj)
1619 keys = get_keys(obj)
1625 if not keys:
1620 if not keys:
1626 return keys
1621 return keys
1627 closing_quote, token_offset, matches = match_dict_keys(keys, prefix, self.splitter.delims)
1622 closing_quote, token_offset, matches = match_dict_keys(keys, prefix, self.splitter.delims)
1628 if not matches:
1623 if not matches:
1629 return matches
1624 return matches
1630
1625
1631 # get the cursor position of
1626 # get the cursor position of
1632 # - the text being completed
1627 # - the text being completed
1633 # - the start of the key text
1628 # - the start of the key text
1634 # - the start of the completion
1629 # - the start of the completion
1635 text_start = len(self.text_until_cursor) - len(text)
1630 text_start = len(self.text_until_cursor) - len(text)
1636 if prefix:
1631 if prefix:
1637 key_start = match.start(2)
1632 key_start = match.start(2)
1638 completion_start = key_start + token_offset
1633 completion_start = key_start + token_offset
1639 else:
1634 else:
1640 key_start = completion_start = match.end()
1635 key_start = completion_start = match.end()
1641
1636
1642 # grab the leading prefix, to make sure all completions start with `text`
1637 # grab the leading prefix, to make sure all completions start with `text`
1643 if text_start > key_start:
1638 if text_start > key_start:
1644 leading = ''
1639 leading = ''
1645 else:
1640 else:
1646 leading = text[text_start:completion_start]
1641 leading = text[text_start:completion_start]
1647
1642
1648 # the index of the `[` character
1643 # the index of the `[` character
1649 bracket_idx = match.end(1)
1644 bracket_idx = match.end(1)
1650
1645
1651 # append closing quote and bracket as appropriate
1646 # append closing quote and bracket as appropriate
1652 # this is *not* appropriate if the opening quote or bracket is outside
1647 # this is *not* appropriate if the opening quote or bracket is outside
1653 # the text given to this method
1648 # the text given to this method
1654 suf = ''
1649 suf = ''
1655 continuation = self.line_buffer[len(self.text_until_cursor):]
1650 continuation = self.line_buffer[len(self.text_until_cursor):]
1656 if key_start > text_start and closing_quote:
1651 if key_start > text_start and closing_quote:
1657 # quotes were opened inside text, maybe close them
1652 # quotes were opened inside text, maybe close them
1658 if continuation.startswith(closing_quote):
1653 if continuation.startswith(closing_quote):
1659 continuation = continuation[len(closing_quote):]
1654 continuation = continuation[len(closing_quote):]
1660 else:
1655 else:
1661 suf += closing_quote
1656 suf += closing_quote
1662 if bracket_idx > text_start:
1657 if bracket_idx > text_start:
1663 # brackets were opened inside text, maybe close them
1658 # brackets were opened inside text, maybe close them
1664 if not continuation.startswith(']'):
1659 if not continuation.startswith(']'):
1665 suf += ']'
1660 suf += ']'
1666
1661
1667 return [leading + k + suf for k in matches]
1662 return [leading + k + suf for k in matches]
1668
1663
1669 def unicode_name_matches(self, text):
1664 def unicode_name_matches(self, text):
1670 u"""Match Latex-like syntax for unicode characters base
1665 u"""Match Latex-like syntax for unicode characters base
1671 on the name of the character.
1666 on the name of the character.
1672
1667
1673 This does ``\\GREEK SMALL LETTER ETA`` -> ``Ξ·``
1668 This does ``\\GREEK SMALL LETTER ETA`` -> ``Ξ·``
1674
1669
1675 Works only on valid python 3 identifier, or on combining characters that
1670 Works only on valid python 3 identifier, or on combining characters that
1676 will combine to form a valid identifier.
1671 will combine to form a valid identifier.
1677
1672
1678 Used on Python 3 only.
1673 Used on Python 3 only.
1679 """
1674 """
1680 slashpos = text.rfind('\\')
1675 slashpos = text.rfind('\\')
1681 if slashpos > -1:
1676 if slashpos > -1:
1682 s = text[slashpos+1:]
1677 s = text[slashpos+1:]
1683 try :
1678 try :
1684 unic = unicodedata.lookup(s)
1679 unic = unicodedata.lookup(s)
1685 # allow combining chars
1680 # allow combining chars
1686 if ('a'+unic).isidentifier():
1681 if ('a'+unic).isidentifier():
1687 return '\\'+s,[unic]
1682 return '\\'+s,[unic]
1688 except KeyError:
1683 except KeyError:
1689 pass
1684 pass
1690 return u'', []
1685 return u'', []
1691
1686
1692
1687
1693 def latex_matches(self, text):
1688 def latex_matches(self, text):
1694 u"""Match Latex syntax for unicode characters.
1689 u"""Match Latex syntax for unicode characters.
1695
1690
1696 This does both ``\\alp`` -> ``\\alpha`` and ``\\alpha`` -> ``Ξ±``
1691 This does both ``\\alp`` -> ``\\alpha`` and ``\\alpha`` -> ``Ξ±``
1697
1692
1698 Used on Python 3 only.
1693 Used on Python 3 only.
1699 """
1694 """
1700 slashpos = text.rfind('\\')
1695 slashpos = text.rfind('\\')
1701 if slashpos > -1:
1696 if slashpos > -1:
1702 s = text[slashpos:]
1697 s = text[slashpos:]
1703 if s in latex_symbols:
1698 if s in latex_symbols:
1704 # Try to complete a full latex symbol to unicode
1699 # Try to complete a full latex symbol to unicode
1705 # \\alpha -> Ξ±
1700 # \\alpha -> Ξ±
1706 return s, [latex_symbols[s]]
1701 return s, [latex_symbols[s]]
1707 else:
1702 else:
1708 # If a user has partially typed a latex symbol, give them
1703 # If a user has partially typed a latex symbol, give them
1709 # a full list of options \al -> [\aleph, \alpha]
1704 # a full list of options \al -> [\aleph, \alpha]
1710 matches = [k for k in latex_symbols if k.startswith(s)]
1705 matches = [k for k in latex_symbols if k.startswith(s)]
1711 return s, matches
1706 return s, matches
1712 return u'', []
1707 return u'', []
1713
1708
1714 def dispatch_custom_completer(self, text):
1709 def dispatch_custom_completer(self, text):
1715 if not self.custom_completers:
1710 if not self.custom_completers:
1716 return
1711 return
1717
1712
1718 line = self.line_buffer
1713 line = self.line_buffer
1719 if not line.strip():
1714 if not line.strip():
1720 return None
1715 return None
1721
1716
1722 # Create a little structure to pass all the relevant information about
1717 # Create a little structure to pass all the relevant information about
1723 # the current completion to any custom completer.
1718 # the current completion to any custom completer.
1724 event = SimpleNamespace()
1719 event = SimpleNamespace()
1725 event.line = line
1720 event.line = line
1726 event.symbol = text
1721 event.symbol = text
1727 cmd = line.split(None,1)[0]
1722 cmd = line.split(None,1)[0]
1728 event.command = cmd
1723 event.command = cmd
1729 event.text_until_cursor = self.text_until_cursor
1724 event.text_until_cursor = self.text_until_cursor
1730
1725
1731 # for foo etc, try also to find completer for %foo
1726 # for foo etc, try also to find completer for %foo
1732 if not cmd.startswith(self.magic_escape):
1727 if not cmd.startswith(self.magic_escape):
1733 try_magic = self.custom_completers.s_matches(
1728 try_magic = self.custom_completers.s_matches(
1734 self.magic_escape + cmd)
1729 self.magic_escape + cmd)
1735 else:
1730 else:
1736 try_magic = []
1731 try_magic = []
1737
1732
1738 for c in itertools.chain(self.custom_completers.s_matches(cmd),
1733 for c in itertools.chain(self.custom_completers.s_matches(cmd),
1739 try_magic,
1734 try_magic,
1740 self.custom_completers.flat_matches(self.text_until_cursor)):
1735 self.custom_completers.flat_matches(self.text_until_cursor)):
1741 try:
1736 try:
1742 res = c(event)
1737 res = c(event)
1743 if res:
1738 if res:
1744 # first, try case sensitive match
1739 # first, try case sensitive match
1745 withcase = [r for r in res if r.startswith(text)]
1740 withcase = [r for r in res if r.startswith(text)]
1746 if withcase:
1741 if withcase:
1747 return withcase
1742 return withcase
1748 # if none, then case insensitive ones are ok too
1743 # if none, then case insensitive ones are ok too
1749 text_low = text.lower()
1744 text_low = text.lower()
1750 return [r for r in res if r.lower().startswith(text_low)]
1745 return [r for r in res if r.lower().startswith(text_low)]
1751 except TryNext:
1746 except TryNext:
1752 pass
1747 pass
1753 except KeyboardInterrupt:
1748 except KeyboardInterrupt:
1754 """
1749 """
1755 If custom completer take too long,
1750 If custom completer take too long,
1756 let keyboard interrupt abort and return nothing.
1751 let keyboard interrupt abort and return nothing.
1757 """
1752 """
1758 break
1753 break
1759
1754
1760 return None
1755 return None
1761
1756
1762 def completions(self, text: str, offset: int)->Iterator[Completion]:
1757 def completions(self, text: str, offset: int)->Iterator[Completion]:
1763 """
1758 """
1764 Returns an iterator over the possible completions
1759 Returns an iterator over the possible completions
1765
1760
1766 .. warning:: Unstable
1761 .. warning:: Unstable
1767
1762
1768 This function is unstable, API may change without warning.
1763 This function is unstable, API may change without warning.
1769 It will also raise unless use in proper context manager.
1764 It will also raise unless use in proper context manager.
1770
1765
1771 Parameters
1766 Parameters
1772 ----------
1767 ----------
1773
1768
1774 text:str
1769 text:str
1775 Full text of the current input, multi line string.
1770 Full text of the current input, multi line string.
1776 offset:int
1771 offset:int
1777 Integer representing the position of the cursor in ``text``. Offset
1772 Integer representing the position of the cursor in ``text``. Offset
1778 is 0-based indexed.
1773 is 0-based indexed.
1779
1774
1780 Yields
1775 Yields
1781 ------
1776 ------
1782 :any:`Completion` object
1777 :any:`Completion` object
1783
1778
1784
1779
1785 The cursor on a text can either be seen as being "in between"
1780 The cursor on a text can either be seen as being "in between"
1786 characters or "On" a character depending on the interface visible to
1781 characters or "On" a character depending on the interface visible to
1787 the user. For consistency the cursor being on "in between" characters X
1782 the user. For consistency the cursor being on "in between" characters X
1788 and Y is equivalent to the cursor being "on" character Y, that is to say
1783 and Y is equivalent to the cursor being "on" character Y, that is to say
1789 the character the cursor is on is considered as being after the cursor.
1784 the character the cursor is on is considered as being after the cursor.
1790
1785
1791 Combining characters may span more that one position in the
1786 Combining characters may span more that one position in the
1792 text.
1787 text.
1793
1788
1794
1789
1795 .. note::
1790 .. note::
1796
1791
1797 If ``IPCompleter.debug`` is :any:`True` will yield a ``--jedi/ipython--``
1792 If ``IPCompleter.debug`` is :any:`True` will yield a ``--jedi/ipython--``
1798 fake Completion token to distinguish completion returned by Jedi
1793 fake Completion token to distinguish completion returned by Jedi
1799 and usual IPython completion.
1794 and usual IPython completion.
1800
1795
1801 .. note::
1796 .. note::
1802
1797
1803 Completions are not completely deduplicated yet. If identical
1798 Completions are not completely deduplicated yet. If identical
1804 completions are coming from different sources this function does not
1799 completions are coming from different sources this function does not
1805 ensure that each completion object will only be present once.
1800 ensure that each completion object will only be present once.
1806 """
1801 """
1807 warnings.warn("_complete is a provisional API (as of IPython 6.0). "
1802 warnings.warn("_complete is a provisional API (as of IPython 6.0). "
1808 "It may change without warnings. "
1803 "It may change without warnings. "
1809 "Use in corresponding context manager.",
1804 "Use in corresponding context manager.",
1810 category=ProvisionalCompleterWarning, stacklevel=2)
1805 category=ProvisionalCompleterWarning, stacklevel=2)
1811
1806
1812 seen = set()
1807 seen = set()
1813 try:
1808 try:
1814 for c in self._completions(text, offset, _timeout=self.jedi_compute_type_timeout/1000):
1809 for c in self._completions(text, offset, _timeout=self.jedi_compute_type_timeout/1000):
1815 if c and (c in seen):
1810 if c and (c in seen):
1816 continue
1811 continue
1817 yield c
1812 yield c
1818 seen.add(c)
1813 seen.add(c)
1819 except KeyboardInterrupt:
1814 except KeyboardInterrupt:
1820 """if completions take too long and users send keyboard interrupt,
1815 """if completions take too long and users send keyboard interrupt,
1821 do not crash and return ASAP. """
1816 do not crash and return ASAP. """
1822 pass
1817 pass
1823
1818
1824 def _completions(self, full_text: str, offset: int, *, _timeout)->Iterator[Completion]:
1819 def _completions(self, full_text: str, offset: int, *, _timeout)->Iterator[Completion]:
1825 """
1820 """
1826 Core completion module.Same signature as :any:`completions`, with the
1821 Core completion module.Same signature as :any:`completions`, with the
1827 extra `timeout` parameter (in seconds).
1822 extra `timeout` parameter (in seconds).
1828
1823
1829
1824
1830 Computing jedi's completion ``.type`` can be quite expensive (it is a
1825 Computing jedi's completion ``.type`` can be quite expensive (it is a
1831 lazy property) and can require some warm-up, more warm up than just
1826 lazy property) and can require some warm-up, more warm up than just
1832 computing the ``name`` of a completion. The warm-up can be :
1827 computing the ``name`` of a completion. The warm-up can be :
1833
1828
1834 - Long warm-up the first time a module is encountered after
1829 - Long warm-up the first time a module is encountered after
1835 install/update: actually build parse/inference tree.
1830 install/update: actually build parse/inference tree.
1836
1831
1837 - first time the module is encountered in a session: load tree from
1832 - first time the module is encountered in a session: load tree from
1838 disk.
1833 disk.
1839
1834
1840 We don't want to block completions for tens of seconds so we give the
1835 We don't want to block completions for tens of seconds so we give the
1841 completer a "budget" of ``_timeout`` seconds per invocation to compute
1836 completer a "budget" of ``_timeout`` seconds per invocation to compute
1842 completions types, the completions that have not yet been computed will
1837 completions types, the completions that have not yet been computed will
1843 be marked as "unknown" an will have a chance to be computed next round
1838 be marked as "unknown" an will have a chance to be computed next round
1844 are things get cached.
1839 are things get cached.
1845
1840
1846 Keep in mind that Jedi is not the only thing treating the completion so
1841 Keep in mind that Jedi is not the only thing treating the completion so
1847 keep the timeout short-ish as if we take more than 0.3 second we still
1842 keep the timeout short-ish as if we take more than 0.3 second we still
1848 have lots of processing to do.
1843 have lots of processing to do.
1849
1844
1850 """
1845 """
1851 deadline = time.monotonic() + _timeout
1846 deadline = time.monotonic() + _timeout
1852
1847
1853
1848
1854 before = full_text[:offset]
1849 before = full_text[:offset]
1855 cursor_line, cursor_column = position_to_cursor(full_text, offset)
1850 cursor_line, cursor_column = position_to_cursor(full_text, offset)
1856
1851
1857 matched_text, matches, matches_origin, jedi_matches = self._complete(
1852 matched_text, matches, matches_origin, jedi_matches = self._complete(
1858 full_text=full_text, cursor_line=cursor_line, cursor_pos=cursor_column)
1853 full_text=full_text, cursor_line=cursor_line, cursor_pos=cursor_column)
1859
1854
1860 iter_jm = iter(jedi_matches)
1855 iter_jm = iter(jedi_matches)
1861 if _timeout:
1856 if _timeout:
1862 for jm in iter_jm:
1857 for jm in iter_jm:
1863 try:
1858 try:
1864 type_ = jm.type
1859 type_ = jm.type
1865 except Exception:
1860 except Exception:
1866 if self.debug:
1861 if self.debug:
1867 print("Error in Jedi getting type of ", jm)
1862 print("Error in Jedi getting type of ", jm)
1868 type_ = None
1863 type_ = None
1869 delta = len(jm.name_with_symbols) - len(jm.complete)
1864 delta = len(jm.name_with_symbols) - len(jm.complete)
1870 if type_ == 'function':
1865 if type_ == 'function':
1871 signature = _make_signature(jm)
1866 signature = _make_signature(jm)
1872 else:
1867 else:
1873 signature = ''
1868 signature = ''
1874 yield Completion(start=offset - delta,
1869 yield Completion(start=offset - delta,
1875 end=offset,
1870 end=offset,
1876 text=jm.name_with_symbols,
1871 text=jm.name_with_symbols,
1877 type=type_,
1872 type=type_,
1878 signature=signature,
1873 signature=signature,
1879 _origin='jedi')
1874 _origin='jedi')
1880
1875
1881 if time.monotonic() > deadline:
1876 if time.monotonic() > deadline:
1882 break
1877 break
1883
1878
1884 for jm in iter_jm:
1879 for jm in iter_jm:
1885 delta = len(jm.name_with_symbols) - len(jm.complete)
1880 delta = len(jm.name_with_symbols) - len(jm.complete)
1886 yield Completion(start=offset - delta,
1881 yield Completion(start=offset - delta,
1887 end=offset,
1882 end=offset,
1888 text=jm.name_with_symbols,
1883 text=jm.name_with_symbols,
1889 type='<unknown>', # don't compute type for speed
1884 type='<unknown>', # don't compute type for speed
1890 _origin='jedi',
1885 _origin='jedi',
1891 signature='')
1886 signature='')
1892
1887
1893
1888
1894 start_offset = before.rfind(matched_text)
1889 start_offset = before.rfind(matched_text)
1895
1890
1896 # TODO:
1891 # TODO:
1897 # Suppress this, right now just for debug.
1892 # Suppress this, right now just for debug.
1898 if jedi_matches and matches and self.debug:
1893 if jedi_matches and matches and self.debug:
1899 yield Completion(start=start_offset, end=offset, text='--jedi/ipython--',
1894 yield Completion(start=start_offset, end=offset, text='--jedi/ipython--',
1900 _origin='debug', type='none', signature='')
1895 _origin='debug', type='none', signature='')
1901
1896
1902 # I'm unsure if this is always true, so let's assert and see if it
1897 # I'm unsure if this is always true, so let's assert and see if it
1903 # crash
1898 # crash
1904 assert before.endswith(matched_text)
1899 assert before.endswith(matched_text)
1905 for m, t in zip(matches, matches_origin):
1900 for m, t in zip(matches, matches_origin):
1906 yield Completion(start=start_offset, end=offset, text=m, _origin=t, signature='', type='<unknown>')
1901 yield Completion(start=start_offset, end=offset, text=m, _origin=t, signature='', type='<unknown>')
1907
1902
1908
1903
1909 def complete(self, text=None, line_buffer=None, cursor_pos=None):
1904 def complete(self, text=None, line_buffer=None, cursor_pos=None):
1910 """Find completions for the given text and line context.
1905 """Find completions for the given text and line context.
1911
1906
1912 Note that both the text and the line_buffer are optional, but at least
1907 Note that both the text and the line_buffer are optional, but at least
1913 one of them must be given.
1908 one of them must be given.
1914
1909
1915 Parameters
1910 Parameters
1916 ----------
1911 ----------
1917 text : string, optional
1912 text : string, optional
1918 Text to perform the completion on. If not given, the line buffer
1913 Text to perform the completion on. If not given, the line buffer
1919 is split using the instance's CompletionSplitter object.
1914 is split using the instance's CompletionSplitter object.
1920
1915
1921 line_buffer : string, optional
1916 line_buffer : string, optional
1922 If not given, the completer attempts to obtain the current line
1917 If not given, the completer attempts to obtain the current line
1923 buffer via readline. This keyword allows clients which are
1918 buffer via readline. This keyword allows clients which are
1924 requesting for text completions in non-readline contexts to inform
1919 requesting for text completions in non-readline contexts to inform
1925 the completer of the entire text.
1920 the completer of the entire text.
1926
1921
1927 cursor_pos : int, optional
1922 cursor_pos : int, optional
1928 Index of the cursor in the full line buffer. Should be provided by
1923 Index of the cursor in the full line buffer. Should be provided by
1929 remote frontends where kernel has no access to frontend state.
1924 remote frontends where kernel has no access to frontend state.
1930
1925
1931 Returns
1926 Returns
1932 -------
1927 -------
1933 text : str
1928 text : str
1934 Text that was actually used in the completion.
1929 Text that was actually used in the completion.
1935
1930
1936 matches : list
1931 matches : list
1937 A list of completion matches.
1932 A list of completion matches.
1938
1933
1939
1934
1940 .. note::
1935 .. note::
1941
1936
1942 This API is likely to be deprecated and replaced by
1937 This API is likely to be deprecated and replaced by
1943 :any:`IPCompleter.completions` in the future.
1938 :any:`IPCompleter.completions` in the future.
1944
1939
1945
1940
1946 """
1941 """
1947 warnings.warn('`Completer.complete` is pending deprecation since '
1942 warnings.warn('`Completer.complete` is pending deprecation since '
1948 'IPython 6.0 and will be replaced by `Completer.completions`.',
1943 'IPython 6.0 and will be replaced by `Completer.completions`.',
1949 PendingDeprecationWarning)
1944 PendingDeprecationWarning)
1950 # potential todo, FOLD the 3rd throw away argument of _complete
1945 # potential todo, FOLD the 3rd throw away argument of _complete
1951 # into the first 2 one.
1946 # into the first 2 one.
1952 return self._complete(line_buffer=line_buffer, cursor_pos=cursor_pos, text=text, cursor_line=0)[:2]
1947 return self._complete(line_buffer=line_buffer, cursor_pos=cursor_pos, text=text, cursor_line=0)[:2]
1953
1948
1954 def _complete(self, *, cursor_line, cursor_pos, line_buffer=None, text=None,
1949 def _complete(self, *, cursor_line, cursor_pos, line_buffer=None, text=None,
1955 full_text=None) -> Tuple[str, List[str], List[str], Iterable[_FakeJediCompletion]]:
1950 full_text=None) -> Tuple[str, List[str], List[str], Iterable[_FakeJediCompletion]]:
1956 """
1951 """
1957
1952
1958 Like complete but can also returns raw jedi completions as well as the
1953 Like complete but can also returns raw jedi completions as well as the
1959 origin of the completion text. This could (and should) be made much
1954 origin of the completion text. This could (and should) be made much
1960 cleaner but that will be simpler once we drop the old (and stateful)
1955 cleaner but that will be simpler once we drop the old (and stateful)
1961 :any:`complete` API.
1956 :any:`complete` API.
1962
1957
1963
1958
1964 With current provisional API, cursor_pos act both (depending on the
1959 With current provisional API, cursor_pos act both (depending on the
1965 caller) as the offset in the ``text`` or ``line_buffer``, or as the
1960 caller) as the offset in the ``text`` or ``line_buffer``, or as the
1966 ``column`` when passing multiline strings this could/should be renamed
1961 ``column`` when passing multiline strings this could/should be renamed
1967 but would add extra noise.
1962 but would add extra noise.
1968 """
1963 """
1969
1964
1970 # if the cursor position isn't given, the only sane assumption we can
1965 # if the cursor position isn't given, the only sane assumption we can
1971 # make is that it's at the end of the line (the common case)
1966 # make is that it's at the end of the line (the common case)
1972 if cursor_pos is None:
1967 if cursor_pos is None:
1973 cursor_pos = len(line_buffer) if text is None else len(text)
1968 cursor_pos = len(line_buffer) if text is None else len(text)
1974
1969
1975 if self.use_main_ns:
1970 if self.use_main_ns:
1976 self.namespace = __main__.__dict__
1971 self.namespace = __main__.__dict__
1977
1972
1978 # if text is either None or an empty string, rely on the line buffer
1973 # if text is either None or an empty string, rely on the line buffer
1979 if (not line_buffer) and full_text:
1974 if (not line_buffer) and full_text:
1980 line_buffer = full_text.split('\n')[cursor_line]
1975 line_buffer = full_text.split('\n')[cursor_line]
1981 if not text:
1976 if not text:
1982 text = self.splitter.split_line(line_buffer, cursor_pos)
1977 text = self.splitter.split_line(line_buffer, cursor_pos)
1983
1978
1984 if self.backslash_combining_completions:
1979 if self.backslash_combining_completions:
1985 # allow deactivation of these on windows.
1980 # allow deactivation of these on windows.
1986 base_text = text if not line_buffer else line_buffer[:cursor_pos]
1981 base_text = text if not line_buffer else line_buffer[:cursor_pos]
1987 latex_text, latex_matches = self.latex_matches(base_text)
1982 latex_text, latex_matches = self.latex_matches(base_text)
1988 if latex_matches:
1983 if latex_matches:
1989 return latex_text, latex_matches, ['latex_matches']*len(latex_matches), ()
1984 return latex_text, latex_matches, ['latex_matches']*len(latex_matches), ()
1990 name_text = ''
1985 name_text = ''
1991 name_matches = []
1986 name_matches = []
1992 for meth in (self.unicode_name_matches, back_latex_name_matches, back_unicode_name_matches):
1987 for meth in (self.unicode_name_matches, back_latex_name_matches, back_unicode_name_matches):
1993 name_text, name_matches = meth(base_text)
1988 name_text, name_matches = meth(base_text)
1994 if name_text:
1989 if name_text:
1995 return name_text, name_matches[:MATCHES_LIMIT], \
1990 return name_text, name_matches[:MATCHES_LIMIT], \
1996 [meth.__qualname__]*min(len(name_matches), MATCHES_LIMIT), ()
1991 [meth.__qualname__]*min(len(name_matches), MATCHES_LIMIT), ()
1997
1992
1998
1993
1999 # If no line buffer is given, assume the input text is all there was
1994 # If no line buffer is given, assume the input text is all there was
2000 if line_buffer is None:
1995 if line_buffer is None:
2001 line_buffer = text
1996 line_buffer = text
2002
1997
2003 self.line_buffer = line_buffer
1998 self.line_buffer = line_buffer
2004 self.text_until_cursor = self.line_buffer[:cursor_pos]
1999 self.text_until_cursor = self.line_buffer[:cursor_pos]
2005
2000
2006 # Do magic arg matches
2001 # Do magic arg matches
2007 for matcher in self.magic_arg_matchers:
2002 for matcher in self.magic_arg_matchers:
2008 matches = list(matcher(line_buffer))[:MATCHES_LIMIT]
2003 matches = list(matcher(line_buffer))[:MATCHES_LIMIT]
2009 if matches:
2004 if matches:
2010 origins = [matcher.__qualname__] * len(matches)
2005 origins = [matcher.__qualname__] * len(matches)
2011 return text, matches, origins, ()
2006 return text, matches, origins, ()
2012
2007
2013 # Start with a clean slate of completions
2008 # Start with a clean slate of completions
2014 matches = []
2009 matches = []
2015 custom_res = self.dispatch_custom_completer(text)
2010 custom_res = self.dispatch_custom_completer(text)
2016 # FIXME: we should extend our api to return a dict with completions for
2011 # FIXME: we should extend our api to return a dict with completions for
2017 # different types of objects. The rlcomplete() method could then
2012 # different types of objects. The rlcomplete() method could then
2018 # simply collapse the dict into a list for readline, but we'd have
2013 # simply collapse the dict into a list for readline, but we'd have
2019 # richer completion semantics in other environments.
2014 # richer completion semantics in other environments.
2020 completions = ()
2015 completions = ()
2021 if self.use_jedi:
2016 if self.use_jedi:
2022 if not full_text:
2017 if not full_text:
2023 full_text = line_buffer
2018 full_text = line_buffer
2024 completions = self._jedi_matches(
2019 completions = self._jedi_matches(
2025 cursor_pos, cursor_line, full_text)
2020 cursor_pos, cursor_line, full_text)
2026 if custom_res is not None:
2021 if custom_res is not None:
2027 # did custom completers produce something?
2022 # did custom completers produce something?
2028 matches = [(m, 'custom') for m in custom_res]
2023 matches = [(m, 'custom') for m in custom_res]
2029 else:
2024 else:
2030 # Extend the list of completions with the results of each
2025 # Extend the list of completions with the results of each
2031 # matcher, so we return results to the user from all
2026 # matcher, so we return results to the user from all
2032 # namespaces.
2027 # namespaces.
2033 if self.merge_completions:
2028 if self.merge_completions:
2034 matches = []
2029 matches = []
2035 for matcher in self.matchers:
2030 for matcher in self.matchers:
2036 try:
2031 try:
2037 matches.extend([(m, matcher.__qualname__)
2032 matches.extend([(m, matcher.__qualname__)
2038 for m in matcher(text)])
2033 for m in matcher(text)])
2039 except:
2034 except:
2040 # Show the ugly traceback if the matcher causes an
2035 # Show the ugly traceback if the matcher causes an
2041 # exception, but do NOT crash the kernel!
2036 # exception, but do NOT crash the kernel!
2042 sys.excepthook(*sys.exc_info())
2037 sys.excepthook(*sys.exc_info())
2043 else:
2038 else:
2044 for matcher in self.matchers:
2039 for matcher in self.matchers:
2045 matches = [(m, matcher.__qualname__)
2040 matches = [(m, matcher.__qualname__)
2046 for m in matcher(text)]
2041 for m in matcher(text)]
2047 if matches:
2042 if matches:
2048 break
2043 break
2049 seen = set()
2044 seen = set()
2050 filtered_matches = set()
2045 filtered_matches = set()
2051 for m in matches:
2046 for m in matches:
2052 t, c = m
2047 t, c = m
2053 if t not in seen:
2048 if t not in seen:
2054 filtered_matches.add(m)
2049 filtered_matches.add(m)
2055 seen.add(t)
2050 seen.add(t)
2056
2051
2057 _filtered_matches = sorted(
2052 _filtered_matches = sorted(
2058 set(filtered_matches), key=lambda x: completions_sorting_key(x[0]))\
2053 set(filtered_matches), key=lambda x: completions_sorting_key(x[0]))\
2059 [:MATCHES_LIMIT]
2054 [:MATCHES_LIMIT]
2060
2055
2061 _matches = [m[0] for m in _filtered_matches]
2056 _matches = [m[0] for m in _filtered_matches]
2062 origins = [m[1] for m in _filtered_matches]
2057 origins = [m[1] for m in _filtered_matches]
2063
2058
2064 self.matches = _matches
2059 self.matches = _matches
2065
2060
2066 return text, _matches, origins, completions
2061 return text, _matches, origins, completions
@@ -1,1029 +1,1049 b''
1 # encoding: utf-8
1 # encoding: utf-8
2 """Tests for the IPython tab-completion machinery."""
2 """Tests for the IPython tab-completion machinery."""
3
3
4 # Copyright (c) IPython Development Team.
4 # Copyright (c) IPython Development Team.
5 # Distributed under the terms of the Modified BSD License.
5 # Distributed under the terms of the Modified BSD License.
6
6
7 import os
7 import os
8 import sys
8 import sys
9 import textwrap
9 import textwrap
10 import unittest
10 import unittest
11
11
12 from contextlib import contextmanager
12 from contextlib import contextmanager
13
13
14 import nose.tools as nt
14 import nose.tools as nt
15
15
16 from traitlets.config.loader import Config
16 from traitlets.config.loader import Config
17 from IPython import get_ipython
17 from IPython import get_ipython
18 from IPython.core import completer
18 from IPython.core import completer
19 from IPython.external.decorators import knownfailureif
19 from IPython.external.decorators import knownfailureif
20 from IPython.utils.tempdir import TemporaryDirectory, TemporaryWorkingDirectory
20 from IPython.utils.tempdir import TemporaryDirectory, TemporaryWorkingDirectory
21 from IPython.utils.generics import complete_object
21 from IPython.utils.generics import complete_object
22 from IPython.testing import decorators as dec
22 from IPython.testing import decorators as dec
23
23
24 from IPython.core.completer import (
24 from IPython.core.completer import (
25 Completion, provisionalcompleter, match_dict_keys, _deduplicate_completions)
25 Completion, provisionalcompleter, match_dict_keys, _deduplicate_completions)
26 from nose.tools import assert_in, assert_not_in
26 from nose.tools import assert_in, assert_not_in
27
27
28 #-----------------------------------------------------------------------------
28 #-----------------------------------------------------------------------------
29 # Test functions
29 # Test functions
30 #-----------------------------------------------------------------------------
30 #-----------------------------------------------------------------------------
31
31
32 @contextmanager
32 @contextmanager
33 def greedy_completion():
33 def greedy_completion():
34 ip = get_ipython()
34 ip = get_ipython()
35 greedy_original = ip.Completer.greedy
35 greedy_original = ip.Completer.greedy
36 try:
36 try:
37 ip.Completer.greedy = True
37 ip.Completer.greedy = True
38 yield
38 yield
39 finally:
39 finally:
40 ip.Completer.greedy = greedy_original
40 ip.Completer.greedy = greedy_original
41
41
42 def test_protect_filename():
42 def test_protect_filename():
43 if sys.platform == 'win32':
43 if sys.platform == 'win32':
44 pairs = [('abc','abc'),
44 pairs = [('abc','abc'),
45 (' abc','" abc"'),
45 (' abc','" abc"'),
46 ('a bc','"a bc"'),
46 ('a bc','"a bc"'),
47 ('a bc','"a bc"'),
47 ('a bc','"a bc"'),
48 (' bc','" bc"'),
48 (' bc','" bc"'),
49 ]
49 ]
50 else:
50 else:
51 pairs = [('abc','abc'),
51 pairs = [('abc','abc'),
52 (' abc',r'\ abc'),
52 (' abc',r'\ abc'),
53 ('a bc',r'a\ bc'),
53 ('a bc',r'a\ bc'),
54 ('a bc',r'a\ \ bc'),
54 ('a bc',r'a\ \ bc'),
55 (' bc',r'\ \ bc'),
55 (' bc',r'\ \ bc'),
56 # On posix, we also protect parens and other special characters.
56 # On posix, we also protect parens and other special characters.
57 ('a(bc',r'a\(bc'),
57 ('a(bc',r'a\(bc'),
58 ('a)bc',r'a\)bc'),
58 ('a)bc',r'a\)bc'),
59 ('a( )bc',r'a\(\ \)bc'),
59 ('a( )bc',r'a\(\ \)bc'),
60 ('a[1]bc', r'a\[1\]bc'),
60 ('a[1]bc', r'a\[1\]bc'),
61 ('a{1}bc', r'a\{1\}bc'),
61 ('a{1}bc', r'a\{1\}bc'),
62 ('a#bc', r'a\#bc'),
62 ('a#bc', r'a\#bc'),
63 ('a?bc', r'a\?bc'),
63 ('a?bc', r'a\?bc'),
64 ('a=bc', r'a\=bc'),
64 ('a=bc', r'a\=bc'),
65 ('a\\bc', r'a\\bc'),
65 ('a\\bc', r'a\\bc'),
66 ('a|bc', r'a\|bc'),
66 ('a|bc', r'a\|bc'),
67 ('a;bc', r'a\;bc'),
67 ('a;bc', r'a\;bc'),
68 ('a:bc', r'a\:bc'),
68 ('a:bc', r'a\:bc'),
69 ("a'bc", r"a\'bc"),
69 ("a'bc", r"a\'bc"),
70 ('a*bc', r'a\*bc'),
70 ('a*bc', r'a\*bc'),
71 ('a"bc', r'a\"bc'),
71 ('a"bc', r'a\"bc'),
72 ('a^bc', r'a\^bc'),
72 ('a^bc', r'a\^bc'),
73 ('a&bc', r'a\&bc'),
73 ('a&bc', r'a\&bc'),
74 ]
74 ]
75 # run the actual tests
75 # run the actual tests
76 for s1, s2 in pairs:
76 for s1, s2 in pairs:
77 s1p = completer.protect_filename(s1)
77 s1p = completer.protect_filename(s1)
78 nt.assert_equal(s1p, s2)
78 nt.assert_equal(s1p, s2)
79
79
80
80
81 def check_line_split(splitter, test_specs):
81 def check_line_split(splitter, test_specs):
82 for part1, part2, split in test_specs:
82 for part1, part2, split in test_specs:
83 cursor_pos = len(part1)
83 cursor_pos = len(part1)
84 line = part1+part2
84 line = part1+part2
85 out = splitter.split_line(line, cursor_pos)
85 out = splitter.split_line(line, cursor_pos)
86 nt.assert_equal(out, split)
86 nt.assert_equal(out, split)
87
87
88
88
89 def test_line_split():
89 def test_line_split():
90 """Basic line splitter test with default specs."""
90 """Basic line splitter test with default specs."""
91 sp = completer.CompletionSplitter()
91 sp = completer.CompletionSplitter()
92 # The format of the test specs is: part1, part2, expected answer. Parts 1
92 # The format of the test specs is: part1, part2, expected answer. Parts 1
93 # and 2 are joined into the 'line' sent to the splitter, as if the cursor
93 # and 2 are joined into the 'line' sent to the splitter, as if the cursor
94 # was at the end of part1. So an empty part2 represents someone hitting
94 # was at the end of part1. So an empty part2 represents someone hitting
95 # tab at the end of the line, the most common case.
95 # tab at the end of the line, the most common case.
96 t = [('run some/scrip', '', 'some/scrip'),
96 t = [('run some/scrip', '', 'some/scrip'),
97 ('run scripts/er', 'ror.py foo', 'scripts/er'),
97 ('run scripts/er', 'ror.py foo', 'scripts/er'),
98 ('echo $HOM', '', 'HOM'),
98 ('echo $HOM', '', 'HOM'),
99 ('print sys.pa', '', 'sys.pa'),
99 ('print sys.pa', '', 'sys.pa'),
100 ('print(sys.pa', '', 'sys.pa'),
100 ('print(sys.pa', '', 'sys.pa'),
101 ("execfile('scripts/er", '', 'scripts/er'),
101 ("execfile('scripts/er", '', 'scripts/er'),
102 ('a[x.', '', 'x.'),
102 ('a[x.', '', 'x.'),
103 ('a[x.', 'y', 'x.'),
103 ('a[x.', 'y', 'x.'),
104 ('cd "some_file/', '', 'some_file/'),
104 ('cd "some_file/', '', 'some_file/'),
105 ]
105 ]
106 check_line_split(sp, t)
106 check_line_split(sp, t)
107 # Ensure splitting works OK with unicode by re-running the tests with
107 # Ensure splitting works OK with unicode by re-running the tests with
108 # all inputs turned into unicode
108 # all inputs turned into unicode
109 check_line_split(sp, [ map(str, p) for p in t] )
109 check_line_split(sp, [ map(str, p) for p in t] )
110
110
111
111
112 def test_custom_completion_error():
112 def test_custom_completion_error():
113 """Test that errors from custom attribute completers are silenced."""
113 """Test that errors from custom attribute completers are silenced."""
114 ip = get_ipython()
114 ip = get_ipython()
115 class A(object): pass
115 class A(object): pass
116 ip.user_ns['a'] = A()
116 ip.user_ns['a'] = A()
117
117
118 @complete_object.register(A)
118 @complete_object.register(A)
119 def complete_A(a, existing_completions):
119 def complete_A(a, existing_completions):
120 raise TypeError("this should be silenced")
120 raise TypeError("this should be silenced")
121
121
122 ip.complete("a.")
122 ip.complete("a.")
123
123
124
124
125 def test_unicode_completions():
125 def test_unicode_completions():
126 ip = get_ipython()
126 ip = get_ipython()
127 # Some strings that trigger different types of completion. Check them both
127 # Some strings that trigger different types of completion. Check them both
128 # in str and unicode forms
128 # in str and unicode forms
129 s = ['ru', '%ru', 'cd /', 'floa', 'float(x)/']
129 s = ['ru', '%ru', 'cd /', 'floa', 'float(x)/']
130 for t in s + list(map(str, s)):
130 for t in s + list(map(str, s)):
131 # We don't need to check exact completion values (they may change
131 # We don't need to check exact completion values (they may change
132 # depending on the state of the namespace, but at least no exceptions
132 # depending on the state of the namespace, but at least no exceptions
133 # should be thrown and the return value should be a pair of text, list
133 # should be thrown and the return value should be a pair of text, list
134 # values.
134 # values.
135 text, matches = ip.complete(t)
135 text, matches = ip.complete(t)
136 nt.assert_true(isinstance(text, str))
136 nt.assert_true(isinstance(text, str))
137 nt.assert_true(isinstance(matches, list))
137 nt.assert_true(isinstance(matches, list))
138
138
139 def test_latex_completions():
139 def test_latex_completions():
140 from IPython.core.latex_symbols import latex_symbols
140 from IPython.core.latex_symbols import latex_symbols
141 import random
141 import random
142 ip = get_ipython()
142 ip = get_ipython()
143 # Test some random unicode symbols
143 # Test some random unicode symbols
144 keys = random.sample(latex_symbols.keys(), 10)
144 keys = random.sample(latex_symbols.keys(), 10)
145 for k in keys:
145 for k in keys:
146 text, matches = ip.complete(k)
146 text, matches = ip.complete(k)
147 nt.assert_equal(len(matches),1)
147 nt.assert_equal(len(matches),1)
148 nt.assert_equal(text, k)
148 nt.assert_equal(text, k)
149 nt.assert_equal(matches[0], latex_symbols[k])
149 nt.assert_equal(matches[0], latex_symbols[k])
150 # Test a more complex line
150 # Test a more complex line
151 text, matches = ip.complete(u'print(\\alpha')
151 text, matches = ip.complete(u'print(\\alpha')
152 nt.assert_equal(text, u'\\alpha')
152 nt.assert_equal(text, u'\\alpha')
153 nt.assert_equal(matches[0], latex_symbols['\\alpha'])
153 nt.assert_equal(matches[0], latex_symbols['\\alpha'])
154 # Test multiple matching latex symbols
154 # Test multiple matching latex symbols
155 text, matches = ip.complete(u'\\al')
155 text, matches = ip.complete(u'\\al')
156 nt.assert_in('\\alpha', matches)
156 nt.assert_in('\\alpha', matches)
157 nt.assert_in('\\aleph', matches)
157 nt.assert_in('\\aleph', matches)
158
158
159
159
160
160
161
161
162 def test_back_latex_completion():
162 def test_back_latex_completion():
163 ip = get_ipython()
163 ip = get_ipython()
164
164
165 # do not return more than 1 matches fro \beta, only the latex one.
165 # do not return more than 1 matches fro \beta, only the latex one.
166 name, matches = ip.complete('\\Ξ²')
166 name, matches = ip.complete('\\Ξ²')
167 nt.assert_equal(len(matches), 1)
167 nt.assert_equal(len(matches), 1)
168 nt.assert_equal(matches[0], '\\beta')
168 nt.assert_equal(matches[0], '\\beta')
169
169
170 def test_back_unicode_completion():
170 def test_back_unicode_completion():
171 ip = get_ipython()
171 ip = get_ipython()
172
172
173 name, matches = ip.complete('\\β…€')
173 name, matches = ip.complete('\\β…€')
174 nt.assert_equal(len(matches), 1)
174 nt.assert_equal(len(matches), 1)
175 nt.assert_equal(matches[0], '\\ROMAN NUMERAL FIVE')
175 nt.assert_equal(matches[0], '\\ROMAN NUMERAL FIVE')
176
176
177
177
178 def test_forward_unicode_completion():
178 def test_forward_unicode_completion():
179 ip = get_ipython()
179 ip = get_ipython()
180
180
181 name, matches = ip.complete('\\ROMAN NUMERAL FIVE')
181 name, matches = ip.complete('\\ROMAN NUMERAL FIVE')
182 nt.assert_equal(len(matches), 1)
182 nt.assert_equal(len(matches), 1)
183 nt.assert_equal(matches[0], 'β…€')
183 nt.assert_equal(matches[0], 'β…€')
184
184
185 @dec.knownfailureif(sys.platform == 'win32', 'Fails if there is a C:\\j... path')
185 @dec.knownfailureif(sys.platform == 'win32', 'Fails if there is a C:\\j... path')
186 def test_no_ascii_back_completion():
186 def test_no_ascii_back_completion():
187 ip = get_ipython()
187 ip = get_ipython()
188 with TemporaryWorkingDirectory(): # Avoid any filename completions
188 with TemporaryWorkingDirectory(): # Avoid any filename completions
189 # single ascii letter that don't have yet completions
189 # single ascii letter that don't have yet completions
190 for letter in 'jJ' :
190 for letter in 'jJ' :
191 name, matches = ip.complete('\\'+letter)
191 name, matches = ip.complete('\\'+letter)
192 nt.assert_equal(matches, [])
192 nt.assert_equal(matches, [])
193
193
194
194
195
195
196
196
197 class CompletionSplitterTestCase(unittest.TestCase):
197 class CompletionSplitterTestCase(unittest.TestCase):
198 def setUp(self):
198 def setUp(self):
199 self.sp = completer.CompletionSplitter()
199 self.sp = completer.CompletionSplitter()
200
200
201 def test_delim_setting(self):
201 def test_delim_setting(self):
202 self.sp.delims = ' '
202 self.sp.delims = ' '
203 nt.assert_equal(self.sp.delims, ' ')
203 nt.assert_equal(self.sp.delims, ' ')
204 nt.assert_equal(self.sp._delim_expr, '[\ ]')
204 nt.assert_equal(self.sp._delim_expr, '[\ ]')
205
205
206 def test_spaces(self):
206 def test_spaces(self):
207 """Test with only spaces as split chars."""
207 """Test with only spaces as split chars."""
208 self.sp.delims = ' '
208 self.sp.delims = ' '
209 t = [('foo', '', 'foo'),
209 t = [('foo', '', 'foo'),
210 ('run foo', '', 'foo'),
210 ('run foo', '', 'foo'),
211 ('run foo', 'bar', 'foo'),
211 ('run foo', 'bar', 'foo'),
212 ]
212 ]
213 check_line_split(self.sp, t)
213 check_line_split(self.sp, t)
214
214
215
215
216 def test_has_open_quotes1():
216 def test_has_open_quotes1():
217 for s in ["'", "'''", "'hi' '"]:
217 for s in ["'", "'''", "'hi' '"]:
218 nt.assert_equal(completer.has_open_quotes(s), "'")
218 nt.assert_equal(completer.has_open_quotes(s), "'")
219
219
220
220
221 def test_has_open_quotes2():
221 def test_has_open_quotes2():
222 for s in ['"', '"""', '"hi" "']:
222 for s in ['"', '"""', '"hi" "']:
223 nt.assert_equal(completer.has_open_quotes(s), '"')
223 nt.assert_equal(completer.has_open_quotes(s), '"')
224
224
225
225
226 def test_has_open_quotes3():
226 def test_has_open_quotes3():
227 for s in ["''", "''' '''", "'hi' 'ipython'"]:
227 for s in ["''", "''' '''", "'hi' 'ipython'"]:
228 nt.assert_false(completer.has_open_quotes(s))
228 nt.assert_false(completer.has_open_quotes(s))
229
229
230
230
231 def test_has_open_quotes4():
231 def test_has_open_quotes4():
232 for s in ['""', '""" """', '"hi" "ipython"']:
232 for s in ['""', '""" """', '"hi" "ipython"']:
233 nt.assert_false(completer.has_open_quotes(s))
233 nt.assert_false(completer.has_open_quotes(s))
234
234
235
235
236 @knownfailureif(sys.platform == 'win32', "abspath completions fail on Windows")
236 @knownfailureif(sys.platform == 'win32', "abspath completions fail on Windows")
237 def test_abspath_file_completions():
237 def test_abspath_file_completions():
238 ip = get_ipython()
238 ip = get_ipython()
239 with TemporaryDirectory() as tmpdir:
239 with TemporaryDirectory() as tmpdir:
240 prefix = os.path.join(tmpdir, 'foo')
240 prefix = os.path.join(tmpdir, 'foo')
241 suffixes = ['1', '2']
241 suffixes = ['1', '2']
242 names = [prefix+s for s in suffixes]
242 names = [prefix+s for s in suffixes]
243 for n in names:
243 for n in names:
244 open(n, 'w').close()
244 open(n, 'w').close()
245
245
246 # Check simple completion
246 # Check simple completion
247 c = ip.complete(prefix)[1]
247 c = ip.complete(prefix)[1]
248 nt.assert_equal(c, names)
248 nt.assert_equal(c, names)
249
249
250 # Now check with a function call
250 # Now check with a function call
251 cmd = 'a = f("%s' % prefix
251 cmd = 'a = f("%s' % prefix
252 c = ip.complete(prefix, cmd)[1]
252 c = ip.complete(prefix, cmd)[1]
253 comp = [prefix+s for s in suffixes]
253 comp = [prefix+s for s in suffixes]
254 nt.assert_equal(c, comp)
254 nt.assert_equal(c, comp)
255
255
256
256
257 def test_local_file_completions():
257 def test_local_file_completions():
258 ip = get_ipython()
258 ip = get_ipython()
259 with TemporaryWorkingDirectory():
259 with TemporaryWorkingDirectory():
260 prefix = './foo'
260 prefix = './foo'
261 suffixes = ['1', '2']
261 suffixes = ['1', '2']
262 names = [prefix+s for s in suffixes]
262 names = [prefix+s for s in suffixes]
263 for n in names:
263 for n in names:
264 open(n, 'w').close()
264 open(n, 'w').close()
265
265
266 # Check simple completion
266 # Check simple completion
267 c = ip.complete(prefix)[1]
267 c = ip.complete(prefix)[1]
268 nt.assert_equal(c, names)
268 nt.assert_equal(c, names)
269
269
270 # Now check with a function call
270 # Now check with a function call
271 cmd = 'a = f("%s' % prefix
271 cmd = 'a = f("%s' % prefix
272 c = ip.complete(prefix, cmd)[1]
272 c = ip.complete(prefix, cmd)[1]
273 comp = set(prefix+s for s in suffixes)
273 comp = set(prefix+s for s in suffixes)
274 nt.assert_true(comp.issubset(set(c)))
274 nt.assert_true(comp.issubset(set(c)))
275
275
276
276
277 def test_quoted_file_completions():
277 def test_quoted_file_completions():
278 ip = get_ipython()
278 ip = get_ipython()
279 with TemporaryWorkingDirectory():
279 with TemporaryWorkingDirectory():
280 name = "foo'bar"
280 name = "foo'bar"
281 open(name, 'w').close()
281 open(name, 'w').close()
282
282
283 # Don't escape Windows
283 # Don't escape Windows
284 escaped = name if sys.platform == "win32" else "foo\\'bar"
284 escaped = name if sys.platform == "win32" else "foo\\'bar"
285
285
286 # Single quote matches embedded single quote
286 # Single quote matches embedded single quote
287 text = "open('foo"
287 text = "open('foo"
288 c = ip.Completer._complete(cursor_line=0,
288 c = ip.Completer._complete(cursor_line=0,
289 cursor_pos=len(text),
289 cursor_pos=len(text),
290 full_text=text)[1]
290 full_text=text)[1]
291 nt.assert_equal(c, [escaped])
291 nt.assert_equal(c, [escaped])
292
292
293 # Double quote requires no escape
293 # Double quote requires no escape
294 text = 'open("foo'
294 text = 'open("foo'
295 c = ip.Completer._complete(cursor_line=0,
295 c = ip.Completer._complete(cursor_line=0,
296 cursor_pos=len(text),
296 cursor_pos=len(text),
297 full_text=text)[1]
297 full_text=text)[1]
298 nt.assert_equal(c, [name])
298 nt.assert_equal(c, [name])
299
299
300 # No quote requires an escape
300 # No quote requires an escape
301 text = '%ls foo'
301 text = '%ls foo'
302 c = ip.Completer._complete(cursor_line=0,
302 c = ip.Completer._complete(cursor_line=0,
303 cursor_pos=len(text),
303 cursor_pos=len(text),
304 full_text=text)[1]
304 full_text=text)[1]
305 nt.assert_equal(c, [escaped])
305 nt.assert_equal(c, [escaped])
306
306
307
307
308 def test_jedi():
308 def test_jedi():
309 """
309 """
310 A couple of issue we had with Jedi
310 A couple of issue we had with Jedi
311 """
311 """
312 ip = get_ipython()
312 ip = get_ipython()
313
313
314 def _test_complete(reason, s, comp, start=None, end=None):
314 def _test_complete(reason, s, comp, start=None, end=None):
315 l = len(s)
315 l = len(s)
316 start = start if start is not None else l
316 start = start if start is not None else l
317 end = end if end is not None else l
317 end = end if end is not None else l
318 with provisionalcompleter():
318 with provisionalcompleter():
319 ip.Completer.use_jedi = True
319 ip.Completer.use_jedi = True
320 completions = set(ip.Completer.completions(s, l))
320 completions = set(ip.Completer.completions(s, l))
321 ip.Completer.use_jedi = False
321 ip.Completer.use_jedi = False
322 assert_in(Completion(start, end, comp), completions, reason)
322 assert_in(Completion(start, end, comp), completions, reason)
323
323
324 def _test_not_complete(reason, s, comp):
324 def _test_not_complete(reason, s, comp):
325 l = len(s)
325 l = len(s)
326 with provisionalcompleter():
326 with provisionalcompleter():
327 ip.Completer.use_jedi = True
327 ip.Completer.use_jedi = True
328 completions = set(ip.Completer.completions(s, l))
328 completions = set(ip.Completer.completions(s, l))
329 ip.Completer.use_jedi = False
329 ip.Completer.use_jedi = False
330 assert_not_in(Completion(l, l, comp), completions, reason)
330 assert_not_in(Completion(l, l, comp), completions, reason)
331
331
332 import jedi
332 import jedi
333 jedi_version = tuple(int(i) for i in jedi.__version__.split('.')[:3])
333 jedi_version = tuple(int(i) for i in jedi.__version__.split('.')[:3])
334 if jedi_version > (0, 10):
334 if jedi_version > (0, 10):
335 yield _test_complete, 'jedi >0.9 should complete and not crash', 'a=1;a.', 'real'
335 yield _test_complete, 'jedi >0.9 should complete and not crash', 'a=1;a.', 'real'
336 yield _test_complete, 'can infer first argument', 'a=(1,"foo");a[0].', 'real'
336 yield _test_complete, 'can infer first argument', 'a=(1,"foo");a[0].', 'real'
337 yield _test_complete, 'can infer second argument', 'a=(1,"foo");a[1].', 'capitalize'
337 yield _test_complete, 'can infer second argument', 'a=(1,"foo");a[1].', 'capitalize'
338 yield _test_complete, 'cover duplicate completions', 'im', 'import', 0, 2
338 yield _test_complete, 'cover duplicate completions', 'im', 'import', 0, 2
339
339
340 yield _test_not_complete, 'does not mix types', 'a=(1,"foo");a[0].', 'capitalize'
340 yield _test_not_complete, 'does not mix types', 'a=(1,"foo");a[0].', 'capitalize'
341
341
342 def test_completion_have_signature():
342 def test_completion_have_signature():
343 """
343 """
344 Lets make sure jedi is capable of pulling out the signature of the function we are completing.
344 Lets make sure jedi is capable of pulling out the signature of the function we are completing.
345 """
345 """
346 ip = get_ipython()
346 ip = get_ipython()
347 with provisionalcompleter():
347 with provisionalcompleter():
348 ip.Completer.use_jedi = True
348 ip.Completer.use_jedi = True
349 completions = ip.Completer.completions('ope', 3)
349 completions = ip.Completer.completions('ope', 3)
350 c = next(completions) # should be `open`
350 c = next(completions) # should be `open`
351 ip.Completer.use_jedi = False
351 ip.Completer.use_jedi = False
352 assert 'file' in c.signature, "Signature of function was not found by completer"
352 assert 'file' in c.signature, "Signature of function was not found by completer"
353 assert 'encoding' in c.signature, "Signature of function was not found by completer"
353 assert 'encoding' in c.signature, "Signature of function was not found by completer"
354
354
355
355
356 def test_deduplicate_completions():
356 def test_deduplicate_completions():
357 """
357 """
358 Test that completions are correctly deduplicated (even if ranges are not the same)
358 Test that completions are correctly deduplicated (even if ranges are not the same)
359 """
359 """
360 ip = get_ipython()
360 ip = get_ipython()
361 ip.ex(textwrap.dedent('''
361 ip.ex(textwrap.dedent('''
362 class Z:
362 class Z:
363 zoo = 1
363 zoo = 1
364 '''))
364 '''))
365 with provisionalcompleter():
365 with provisionalcompleter():
366 ip.Completer.use_jedi = True
366 ip.Completer.use_jedi = True
367 l = list(_deduplicate_completions('Z.z', ip.Completer.completions('Z.z', 3)))
367 l = list(_deduplicate_completions('Z.z', ip.Completer.completions('Z.z', 3)))
368 ip.Completer.use_jedi = False
368 ip.Completer.use_jedi = False
369
369
370 assert len(l) == 1, 'Completions (Z.z<tab>) correctly deduplicate: %s ' % l
370 assert len(l) == 1, 'Completions (Z.z<tab>) correctly deduplicate: %s ' % l
371 assert l[0].text == 'zoo' # and not `it.accumulate`
371 assert l[0].text == 'zoo' # and not `it.accumulate`
372
372
373
373
374 def test_greedy_completions():
374 def test_greedy_completions():
375 """
375 """
376 Test the capability of the Greedy completer.
376 Test the capability of the Greedy completer.
377
377
378 Most of the test here does not really show off the greedy completer, for proof
378 Most of the test here does not really show off the greedy completer, for proof
379 each of the text below now pass with Jedi. The greedy completer is capable of more.
379 each of the text below now pass with Jedi. The greedy completer is capable of more.
380
380
381 See the :any:`test_dict_key_completion_contexts`
381 See the :any:`test_dict_key_completion_contexts`
382
382
383 """
383 """
384 ip = get_ipython()
384 ip = get_ipython()
385 ip.ex('a=list(range(5))')
385 ip.ex('a=list(range(5))')
386 _,c = ip.complete('.',line='a[0].')
386 _,c = ip.complete('.',line='a[0].')
387 nt.assert_false('.real' in c,
387 nt.assert_false('.real' in c,
388 "Shouldn't have completed on a[0]: %s"%c)
388 "Shouldn't have completed on a[0]: %s"%c)
389 with greedy_completion(), provisionalcompleter():
389 with greedy_completion(), provisionalcompleter():
390 def _(line, cursor_pos, expect, message, completion):
390 def _(line, cursor_pos, expect, message, completion):
391 ip.Completer.use_jedi = False
391 ip.Completer.use_jedi = False
392 _,c = ip.complete('.', line=line, cursor_pos=cursor_pos)
392 _,c = ip.complete('.', line=line, cursor_pos=cursor_pos)
393 nt.assert_in(expect, c, message % c)
393 nt.assert_in(expect, c, message % c)
394
394
395 ip.Completer.use_jedi = True
395 ip.Completer.use_jedi = True
396 with provisionalcompleter():
396 with provisionalcompleter():
397 completions = ip.Completer.completions(line, cursor_pos)
397 completions = ip.Completer.completions(line, cursor_pos)
398 nt.assert_in(completion, completions)
398 nt.assert_in(completion, completions)
399
399
400 yield _, 'a[0].', 5, 'a[0].real', "Should have completed on a[0].: %s", Completion(5,5, 'real')
400 yield _, 'a[0].', 5, 'a[0].real', "Should have completed on a[0].: %s", Completion(5,5, 'real')
401 yield _, 'a[0].r', 6, 'a[0].real', "Should have completed on a[0].r: %s", Completion(5,6, 'real')
401 yield _, 'a[0].r', 6, 'a[0].real', "Should have completed on a[0].r: %s", Completion(5,6, 'real')
402
402
403 if sys.version_info > (3, 4):
403 if sys.version_info > (3, 4):
404 yield _, 'a[0].from_', 10, 'a[0].from_bytes', "Should have completed on a[0].from_: %s", Completion(5, 10, 'from_bytes')
404 yield _, 'a[0].from_', 10, 'a[0].from_bytes', "Should have completed on a[0].from_: %s", Completion(5, 10, 'from_bytes')
405
405
406
406
407 def test_omit__names():
407 def test_omit__names():
408 # also happens to test IPCompleter as a configurable
408 # also happens to test IPCompleter as a configurable
409 ip = get_ipython()
409 ip = get_ipython()
410 ip._hidden_attr = 1
410 ip._hidden_attr = 1
411 ip._x = {}
411 ip._x = {}
412 c = ip.Completer
412 c = ip.Completer
413 ip.ex('ip=get_ipython()')
413 ip.ex('ip=get_ipython()')
414 cfg = Config()
414 cfg = Config()
415 cfg.IPCompleter.omit__names = 0
415 cfg.IPCompleter.omit__names = 0
416 c.update_config(cfg)
416 c.update_config(cfg)
417 with provisionalcompleter():
417 with provisionalcompleter():
418 c.use_jedi = False
418 c.use_jedi = False
419 s,matches = c.complete('ip.')
419 s,matches = c.complete('ip.')
420 nt.assert_in('ip.__str__', matches)
420 nt.assert_in('ip.__str__', matches)
421 nt.assert_in('ip._hidden_attr', matches)
421 nt.assert_in('ip._hidden_attr', matches)
422
422
423 # c.use_jedi = True
423 # c.use_jedi = True
424 # completions = set(c.completions('ip.', 3))
424 # completions = set(c.completions('ip.', 3))
425 # nt.assert_in(Completion(3, 3, '__str__'), completions)
425 # nt.assert_in(Completion(3, 3, '__str__'), completions)
426 # nt.assert_in(Completion(3,3, "_hidden_attr"), completions)
426 # nt.assert_in(Completion(3,3, "_hidden_attr"), completions)
427
427
428
428
429 cfg = Config()
429 cfg = Config()
430 cfg.IPCompleter.omit__names = 1
430 cfg.IPCompleter.omit__names = 1
431 c.update_config(cfg)
431 c.update_config(cfg)
432 with provisionalcompleter():
432 with provisionalcompleter():
433 c.use_jedi = False
433 c.use_jedi = False
434 s,matches = c.complete('ip.')
434 s,matches = c.complete('ip.')
435 nt.assert_not_in('ip.__str__', matches)
435 nt.assert_not_in('ip.__str__', matches)
436 # nt.assert_in('ip._hidden_attr', matches)
436 # nt.assert_in('ip._hidden_attr', matches)
437
437
438 # c.use_jedi = True
438 # c.use_jedi = True
439 # completions = set(c.completions('ip.', 3))
439 # completions = set(c.completions('ip.', 3))
440 # nt.assert_not_in(Completion(3,3,'__str__'), completions)
440 # nt.assert_not_in(Completion(3,3,'__str__'), completions)
441 # nt.assert_in(Completion(3,3, "_hidden_attr"), completions)
441 # nt.assert_in(Completion(3,3, "_hidden_attr"), completions)
442
442
443 cfg = Config()
443 cfg = Config()
444 cfg.IPCompleter.omit__names = 2
444 cfg.IPCompleter.omit__names = 2
445 c.update_config(cfg)
445 c.update_config(cfg)
446 with provisionalcompleter():
446 with provisionalcompleter():
447 c.use_jedi = False
447 c.use_jedi = False
448 s,matches = c.complete('ip.')
448 s,matches = c.complete('ip.')
449 nt.assert_not_in('ip.__str__', matches)
449 nt.assert_not_in('ip.__str__', matches)
450 nt.assert_not_in('ip._hidden_attr', matches)
450 nt.assert_not_in('ip._hidden_attr', matches)
451
451
452 # c.use_jedi = True
452 # c.use_jedi = True
453 # completions = set(c.completions('ip.', 3))
453 # completions = set(c.completions('ip.', 3))
454 # nt.assert_not_in(Completion(3,3,'__str__'), completions)
454 # nt.assert_not_in(Completion(3,3,'__str__'), completions)
455 # nt.assert_not_in(Completion(3,3, "_hidden_attr"), completions)
455 # nt.assert_not_in(Completion(3,3, "_hidden_attr"), completions)
456
456
457 with provisionalcompleter():
457 with provisionalcompleter():
458 c.use_jedi = False
458 c.use_jedi = False
459 s,matches = c.complete('ip._x.')
459 s,matches = c.complete('ip._x.')
460 nt.assert_in('ip._x.keys', matches)
460 nt.assert_in('ip._x.keys', matches)
461
461
462 # c.use_jedi = True
462 # c.use_jedi = True
463 # completions = set(c.completions('ip._x.', 6))
463 # completions = set(c.completions('ip._x.', 6))
464 # nt.assert_in(Completion(6,6, "keys"), completions)
464 # nt.assert_in(Completion(6,6, "keys"), completions)
465
465
466 del ip._hidden_attr
466 del ip._hidden_attr
467 del ip._x
467 del ip._x
468
468
469
469
470 def test_limit_to__all__False_ok():
470 def test_limit_to__all__False_ok():
471 """
471 """
472 Limit to all is deprecated, once we remove it this test can go away.
472 Limit to all is deprecated, once we remove it this test can go away.
473 """
473 """
474 ip = get_ipython()
474 ip = get_ipython()
475 c = ip.Completer
475 c = ip.Completer
476 c.use_jedi = False
476 c.use_jedi = False
477 ip.ex('class D: x=24')
477 ip.ex('class D: x=24')
478 ip.ex('d=D()')
478 ip.ex('d=D()')
479 cfg = Config()
479 cfg = Config()
480 cfg.IPCompleter.limit_to__all__ = False
480 cfg.IPCompleter.limit_to__all__ = False
481 c.update_config(cfg)
481 c.update_config(cfg)
482 s, matches = c.complete('d.')
482 s, matches = c.complete('d.')
483 nt.assert_in('d.x', matches)
483 nt.assert_in('d.x', matches)
484
484
485
485
486 def test_get__all__entries_ok():
486 def test_get__all__entries_ok():
487 class A(object):
487 class A(object):
488 __all__ = ['x', 1]
488 __all__ = ['x', 1]
489 words = completer.get__all__entries(A())
489 words = completer.get__all__entries(A())
490 nt.assert_equal(words, ['x'])
490 nt.assert_equal(words, ['x'])
491
491
492
492
493 def test_get__all__entries_no__all__ok():
493 def test_get__all__entries_no__all__ok():
494 class A(object):
494 class A(object):
495 pass
495 pass
496 words = completer.get__all__entries(A())
496 words = completer.get__all__entries(A())
497 nt.assert_equal(words, [])
497 nt.assert_equal(words, [])
498
498
499
499
500 def test_func_kw_completions():
500 def test_func_kw_completions():
501 ip = get_ipython()
501 ip = get_ipython()
502 c = ip.Completer
502 c = ip.Completer
503 c.use_jedi = False
503 c.use_jedi = False
504 ip.ex('def myfunc(a=1,b=2): return a+b')
504 ip.ex('def myfunc(a=1,b=2): return a+b')
505 s, matches = c.complete(None, 'myfunc(1,b')
505 s, matches = c.complete(None, 'myfunc(1,b')
506 nt.assert_in('b=', matches)
506 nt.assert_in('b=', matches)
507 # Simulate completing with cursor right after b (pos==10):
507 # Simulate completing with cursor right after b (pos==10):
508 s, matches = c.complete(None, 'myfunc(1,b)', 10)
508 s, matches = c.complete(None, 'myfunc(1,b)', 10)
509 nt.assert_in('b=', matches)
509 nt.assert_in('b=', matches)
510 s, matches = c.complete(None, 'myfunc(a="escaped\\")string",b')
510 s, matches = c.complete(None, 'myfunc(a="escaped\\")string",b')
511 nt.assert_in('b=', matches)
511 nt.assert_in('b=', matches)
512 #builtin function
512 #builtin function
513 s, matches = c.complete(None, 'min(k, k')
513 s, matches = c.complete(None, 'min(k, k')
514 nt.assert_in('key=', matches)
514 nt.assert_in('key=', matches)
515
515
516
516
517 def test_default_arguments_from_docstring():
517 def test_default_arguments_from_docstring():
518 ip = get_ipython()
518 ip = get_ipython()
519 c = ip.Completer
519 c = ip.Completer
520 kwd = c._default_arguments_from_docstring(
520 kwd = c._default_arguments_from_docstring(
521 'min(iterable[, key=func]) -> value')
521 'min(iterable[, key=func]) -> value')
522 nt.assert_equal(kwd, ['key'])
522 nt.assert_equal(kwd, ['key'])
523 #with cython type etc
523 #with cython type etc
524 kwd = c._default_arguments_from_docstring(
524 kwd = c._default_arguments_from_docstring(
525 'Minuit.migrad(self, int ncall=10000, resume=True, int nsplit=1)\n')
525 'Minuit.migrad(self, int ncall=10000, resume=True, int nsplit=1)\n')
526 nt.assert_equal(kwd, ['ncall', 'resume', 'nsplit'])
526 nt.assert_equal(kwd, ['ncall', 'resume', 'nsplit'])
527 #white spaces
527 #white spaces
528 kwd = c._default_arguments_from_docstring(
528 kwd = c._default_arguments_from_docstring(
529 '\n Minuit.migrad(self, int ncall=10000, resume=True, int nsplit=1)\n')
529 '\n Minuit.migrad(self, int ncall=10000, resume=True, int nsplit=1)\n')
530 nt.assert_equal(kwd, ['ncall', 'resume', 'nsplit'])
530 nt.assert_equal(kwd, ['ncall', 'resume', 'nsplit'])
531
531
532 def test_line_magics():
532 def test_line_magics():
533 ip = get_ipython()
533 ip = get_ipython()
534 c = ip.Completer
534 c = ip.Completer
535 s, matches = c.complete(None, 'lsmag')
535 s, matches = c.complete(None, 'lsmag')
536 nt.assert_in('%lsmagic', matches)
536 nt.assert_in('%lsmagic', matches)
537 s, matches = c.complete(None, '%lsmag')
537 s, matches = c.complete(None, '%lsmag')
538 nt.assert_in('%lsmagic', matches)
538 nt.assert_in('%lsmagic', matches)
539
539
540
540
541 def test_cell_magics():
541 def test_cell_magics():
542 from IPython.core.magic import register_cell_magic
542 from IPython.core.magic import register_cell_magic
543
543
544 @register_cell_magic
544 @register_cell_magic
545 def _foo_cellm(line, cell):
545 def _foo_cellm(line, cell):
546 pass
546 pass
547
547
548 ip = get_ipython()
548 ip = get_ipython()
549 c = ip.Completer
549 c = ip.Completer
550
550
551 s, matches = c.complete(None, '_foo_ce')
551 s, matches = c.complete(None, '_foo_ce')
552 nt.assert_in('%%_foo_cellm', matches)
552 nt.assert_in('%%_foo_cellm', matches)
553 s, matches = c.complete(None, '%%_foo_ce')
553 s, matches = c.complete(None, '%%_foo_ce')
554 nt.assert_in('%%_foo_cellm', matches)
554 nt.assert_in('%%_foo_cellm', matches)
555
555
556
556
557 def test_line_cell_magics():
557 def test_line_cell_magics():
558 from IPython.core.magic import register_line_cell_magic
558 from IPython.core.magic import register_line_cell_magic
559
559
560 @register_line_cell_magic
560 @register_line_cell_magic
561 def _bar_cellm(line, cell):
561 def _bar_cellm(line, cell):
562 pass
562 pass
563
563
564 ip = get_ipython()
564 ip = get_ipython()
565 c = ip.Completer
565 c = ip.Completer
566
566
567 # The policy here is trickier, see comments in completion code. The
567 # The policy here is trickier, see comments in completion code. The
568 # returned values depend on whether the user passes %% or not explicitly,
568 # returned values depend on whether the user passes %% or not explicitly,
569 # and this will show a difference if the same name is both a line and cell
569 # and this will show a difference if the same name is both a line and cell
570 # magic.
570 # magic.
571 s, matches = c.complete(None, '_bar_ce')
571 s, matches = c.complete(None, '_bar_ce')
572 nt.assert_in('%_bar_cellm', matches)
572 nt.assert_in('%_bar_cellm', matches)
573 nt.assert_in('%%_bar_cellm', matches)
573 nt.assert_in('%%_bar_cellm', matches)
574 s, matches = c.complete(None, '%_bar_ce')
574 s, matches = c.complete(None, '%_bar_ce')
575 nt.assert_in('%_bar_cellm', matches)
575 nt.assert_in('%_bar_cellm', matches)
576 nt.assert_in('%%_bar_cellm', matches)
576 nt.assert_in('%%_bar_cellm', matches)
577 s, matches = c.complete(None, '%%_bar_ce')
577 s, matches = c.complete(None, '%%_bar_ce')
578 nt.assert_not_in('%_bar_cellm', matches)
578 nt.assert_not_in('%_bar_cellm', matches)
579 nt.assert_in('%%_bar_cellm', matches)
579 nt.assert_in('%%_bar_cellm', matches)
580
580
581
581
582 def test_magic_completion_order():
582 def test_magic_completion_order():
583 ip = get_ipython()
583 ip = get_ipython()
584 c = ip.Completer
584 c = ip.Completer
585
585
586 # Test ordering of line and cell magics.
586 # Test ordering of line and cell magics.
587 text, matches = c.complete("timeit")
587 text, matches = c.complete("timeit")
588 nt.assert_equal(matches, ["%timeit", "%%timeit"])
588 nt.assert_equal(matches, ["%timeit", "%%timeit"])
589
589
590
590
591 def test_magic_completion_shadowing():
591 def test_magic_completion_shadowing():
592 ip = get_ipython()
592 ip = get_ipython()
593 c = ip.Completer
593 c = ip.Completer
594 c.use_jedi = False
594 c.use_jedi = False
595
595
596 # Before importing matplotlib, %matplotlib magic should be the only option.
596 # Before importing matplotlib, %matplotlib magic should be the only option.
597 text, matches = c.complete("mat")
597 text, matches = c.complete("mat")
598 nt.assert_equal(matches, ["%matplotlib"])
598 nt.assert_equal(matches, ["%matplotlib"])
599
599
600 # The newly introduced name should shadow the magic.
600 # The newly introduced name should shadow the magic.
601 ip.run_cell("matplotlib = 1")
601 ip.run_cell("matplotlib = 1")
602 text, matches = c.complete("mat")
602 text, matches = c.complete("mat")
603 nt.assert_equal(matches, ["matplotlib"])
603 nt.assert_equal(matches, ["matplotlib"])
604
604
605 # After removing matplotlib from namespace, the magic should again be
605 # After removing matplotlib from namespace, the magic should again be
606 # the only option.
606 # the only option.
607 del ip.user_ns["matplotlib"]
607 del ip.user_ns["matplotlib"]
608 text, matches = c.complete("mat")
608 text, matches = c.complete("mat")
609 nt.assert_equal(matches, ["%matplotlib"])
609 nt.assert_equal(matches, ["%matplotlib"])
610
610
611 def test_magic_completion_shadowing_explicit():
611 def test_magic_completion_shadowing_explicit():
612 """
612 """
613 If the user try to complete a shadowed magic, and explicit % start should
613 If the user try to complete a shadowed magic, and explicit % start should
614 still return the completions.
614 still return the completions.
615 """
615 """
616 ip = get_ipython()
616 ip = get_ipython()
617 c = ip.Completer
617 c = ip.Completer
618
618
619 # Before importing matplotlib, %matplotlib magic should be the only option.
619 # Before importing matplotlib, %matplotlib magic should be the only option.
620 text, matches = c.complete("%mat")
620 text, matches = c.complete("%mat")
621 nt.assert_equal(matches, ["%matplotlib"])
621 nt.assert_equal(matches, ["%matplotlib"])
622
622
623 ip.run_cell("matplotlib = 1")
623 ip.run_cell("matplotlib = 1")
624
624
625 # After removing matplotlib from namespace, the magic should still be
625 # After removing matplotlib from namespace, the magic should still be
626 # the only option.
626 # the only option.
627 text, matches = c.complete("%mat")
627 text, matches = c.complete("%mat")
628 nt.assert_equal(matches, ["%matplotlib"])
628 nt.assert_equal(matches, ["%matplotlib"])
629
629
630 def test_magic_config():
630 def test_magic_config():
631 ip = get_ipython()
631 ip = get_ipython()
632 c = ip.Completer
632 c = ip.Completer
633
633
634 s, matches = c.complete(None, 'conf')
634 s, matches = c.complete(None, 'conf')
635 nt.assert_in('%config', matches)
635 nt.assert_in('%config', matches)
636 s, matches = c.complete(None, 'conf')
636 s, matches = c.complete(None, 'conf')
637 nt.assert_not_in('AliasManager', matches)
637 nt.assert_not_in('AliasManager', matches)
638 s, matches = c.complete(None, 'config ')
638 s, matches = c.complete(None, 'config ')
639 nt.assert_in('AliasManager', matches)
639 nt.assert_in('AliasManager', matches)
640 s, matches = c.complete(None, '%config ')
640 s, matches = c.complete(None, '%config ')
641 nt.assert_in('AliasManager', matches)
641 nt.assert_in('AliasManager', matches)
642 s, matches = c.complete(None, 'config Ali')
642 s, matches = c.complete(None, 'config Ali')
643 nt.assert_list_equal(['AliasManager'], matches)
643 nt.assert_list_equal(['AliasManager'], matches)
644 s, matches = c.complete(None, '%config Ali')
644 s, matches = c.complete(None, '%config Ali')
645 nt.assert_list_equal(['AliasManager'], matches)
645 nt.assert_list_equal(['AliasManager'], matches)
646 s, matches = c.complete(None, 'config AliasManager')
646 s, matches = c.complete(None, 'config AliasManager')
647 nt.assert_list_equal(['AliasManager'], matches)
647 nt.assert_list_equal(['AliasManager'], matches)
648 s, matches = c.complete(None, '%config AliasManager')
648 s, matches = c.complete(None, '%config AliasManager')
649 nt.assert_list_equal(['AliasManager'], matches)
649 nt.assert_list_equal(['AliasManager'], matches)
650 s, matches = c.complete(None, 'config AliasManager.')
650 s, matches = c.complete(None, 'config AliasManager.')
651 nt.assert_in('AliasManager.default_aliases', matches)
651 nt.assert_in('AliasManager.default_aliases', matches)
652 s, matches = c.complete(None, '%config AliasManager.')
652 s, matches = c.complete(None, '%config AliasManager.')
653 nt.assert_in('AliasManager.default_aliases', matches)
653 nt.assert_in('AliasManager.default_aliases', matches)
654 s, matches = c.complete(None, 'config AliasManager.de')
654 s, matches = c.complete(None, 'config AliasManager.de')
655 nt.assert_list_equal(['AliasManager.default_aliases'], matches)
655 nt.assert_list_equal(['AliasManager.default_aliases'], matches)
656 s, matches = c.complete(None, 'config AliasManager.de')
656 s, matches = c.complete(None, 'config AliasManager.de')
657 nt.assert_list_equal(['AliasManager.default_aliases'], matches)
657 nt.assert_list_equal(['AliasManager.default_aliases'], matches)
658
658
659
659
660 def test_magic_color():
660 def test_magic_color():
661 ip = get_ipython()
661 ip = get_ipython()
662 c = ip.Completer
662 c = ip.Completer
663
663
664 s, matches = c.complete(None, 'colo')
664 s, matches = c.complete(None, 'colo')
665 nt.assert_in('%colors', matches)
665 nt.assert_in('%colors', matches)
666 s, matches = c.complete(None, 'colo')
666 s, matches = c.complete(None, 'colo')
667 nt.assert_not_in('NoColor', matches)
667 nt.assert_not_in('NoColor', matches)
668 s, matches = c.complete(None, '%colors') # No trailing space
668 s, matches = c.complete(None, '%colors') # No trailing space
669 nt.assert_not_in('NoColor', matches)
669 nt.assert_not_in('NoColor', matches)
670 s, matches = c.complete(None, 'colors ')
670 s, matches = c.complete(None, 'colors ')
671 nt.assert_in('NoColor', matches)
671 nt.assert_in('NoColor', matches)
672 s, matches = c.complete(None, '%colors ')
672 s, matches = c.complete(None, '%colors ')
673 nt.assert_in('NoColor', matches)
673 nt.assert_in('NoColor', matches)
674 s, matches = c.complete(None, 'colors NoCo')
674 s, matches = c.complete(None, 'colors NoCo')
675 nt.assert_list_equal(['NoColor'], matches)
675 nt.assert_list_equal(['NoColor'], matches)
676 s, matches = c.complete(None, '%colors NoCo')
676 s, matches = c.complete(None, '%colors NoCo')
677 nt.assert_list_equal(['NoColor'], matches)
677 nt.assert_list_equal(['NoColor'], matches)
678
678
679
679
680 def test_match_dict_keys():
680 def test_match_dict_keys():
681 """
681 """
682 Test that match_dict_keys works on a couple of use case does return what
682 Test that match_dict_keys works on a couple of use case does return what
683 expected, and does not crash
683 expected, and does not crash
684 """
684 """
685 delims = ' \t\n`!@#$^&*()=+[{]}\\|;:\'",<>?'
685 delims = ' \t\n`!@#$^&*()=+[{]}\\|;:\'",<>?'
686
686
687
687
688 keys = ['foo', b'far']
688 keys = ['foo', b'far']
689 assert match_dict_keys(keys, "b'", delims=delims) == ("'", 2 ,['far'])
689 assert match_dict_keys(keys, "b'", delims=delims) == ("'", 2 ,['far'])
690 assert match_dict_keys(keys, "b'f", delims=delims) == ("'", 2 ,['far'])
690 assert match_dict_keys(keys, "b'f", delims=delims) == ("'", 2 ,['far'])
691 assert match_dict_keys(keys, 'b"', delims=delims) == ('"', 2 ,['far'])
691 assert match_dict_keys(keys, 'b"', delims=delims) == ('"', 2 ,['far'])
692 assert match_dict_keys(keys, 'b"f', delims=delims) == ('"', 2 ,['far'])
692 assert match_dict_keys(keys, 'b"f', delims=delims) == ('"', 2 ,['far'])
693
693
694 assert match_dict_keys(keys, "'", delims=delims) == ("'", 1 ,['foo'])
694 assert match_dict_keys(keys, "'", delims=delims) == ("'", 1 ,['foo'])
695 assert match_dict_keys(keys, "'f", delims=delims) == ("'", 1 ,['foo'])
695 assert match_dict_keys(keys, "'f", delims=delims) == ("'", 1 ,['foo'])
696 assert match_dict_keys(keys, '"', delims=delims) == ('"', 1 ,['foo'])
696 assert match_dict_keys(keys, '"', delims=delims) == ('"', 1 ,['foo'])
697 assert match_dict_keys(keys, '"f', delims=delims) == ('"', 1 ,['foo'])
697 assert match_dict_keys(keys, '"f', delims=delims) == ('"', 1 ,['foo'])
698
698
699 match_dict_keys
699 match_dict_keys
700
700
701
701
702 def test_dict_key_completion_string():
702 def test_dict_key_completion_string():
703 """Test dictionary key completion for string keys"""
703 """Test dictionary key completion for string keys"""
704 ip = get_ipython()
704 ip = get_ipython()
705 complete = ip.Completer.complete
705 complete = ip.Completer.complete
706
706
707 ip.user_ns['d'] = {'abc': None}
707 ip.user_ns['d'] = {'abc': None}
708
708
709 # check completion at different stages
709 # check completion at different stages
710 _, matches = complete(line_buffer="d[")
710 _, matches = complete(line_buffer="d[")
711 nt.assert_in("'abc'", matches)
711 nt.assert_in("'abc'", matches)
712 nt.assert_not_in("'abc']", matches)
712 nt.assert_not_in("'abc']", matches)
713
713
714 _, matches = complete(line_buffer="d['")
714 _, matches = complete(line_buffer="d['")
715 nt.assert_in("abc", matches)
715 nt.assert_in("abc", matches)
716 nt.assert_not_in("abc']", matches)
716 nt.assert_not_in("abc']", matches)
717
717
718 _, matches = complete(line_buffer="d['a")
718 _, matches = complete(line_buffer="d['a")
719 nt.assert_in("abc", matches)
719 nt.assert_in("abc", matches)
720 nt.assert_not_in("abc']", matches)
720 nt.assert_not_in("abc']", matches)
721
721
722 # check use of different quoting
722 # check use of different quoting
723 _, matches = complete(line_buffer="d[\"")
723 _, matches = complete(line_buffer="d[\"")
724 nt.assert_in("abc", matches)
724 nt.assert_in("abc", matches)
725 nt.assert_not_in('abc\"]', matches)
725 nt.assert_not_in('abc\"]', matches)
726
726
727 _, matches = complete(line_buffer="d[\"a")
727 _, matches = complete(line_buffer="d[\"a")
728 nt.assert_in("abc", matches)
728 nt.assert_in("abc", matches)
729 nt.assert_not_in('abc\"]', matches)
729 nt.assert_not_in('abc\"]', matches)
730
730
731 # check sensitivity to following context
731 # check sensitivity to following context
732 _, matches = complete(line_buffer="d[]", cursor_pos=2)
732 _, matches = complete(line_buffer="d[]", cursor_pos=2)
733 nt.assert_in("'abc'", matches)
733 nt.assert_in("'abc'", matches)
734
734
735 _, matches = complete(line_buffer="d['']", cursor_pos=3)
735 _, matches = complete(line_buffer="d['']", cursor_pos=3)
736 nt.assert_in("abc", matches)
736 nt.assert_in("abc", matches)
737 nt.assert_not_in("abc'", matches)
737 nt.assert_not_in("abc'", matches)
738 nt.assert_not_in("abc']", matches)
738 nt.assert_not_in("abc']", matches)
739
739
740 # check multiple solutions are correctly returned and that noise is not
740 # check multiple solutions are correctly returned and that noise is not
741 ip.user_ns['d'] = {'abc': None, 'abd': None, 'bad': None, object(): None,
741 ip.user_ns['d'] = {'abc': None, 'abd': None, 'bad': None, object(): None,
742 5: None}
742 5: None}
743
743
744 _, matches = complete(line_buffer="d['a")
744 _, matches = complete(line_buffer="d['a")
745 nt.assert_in("abc", matches)
745 nt.assert_in("abc", matches)
746 nt.assert_in("abd", matches)
746 nt.assert_in("abd", matches)
747 nt.assert_not_in("bad", matches)
747 nt.assert_not_in("bad", matches)
748 assert not any(m.endswith((']', '"', "'")) for m in matches), matches
748 assert not any(m.endswith((']', '"', "'")) for m in matches), matches
749
749
750 # check escaping and whitespace
750 # check escaping and whitespace
751 ip.user_ns['d'] = {'a\nb': None, 'a\'b': None, 'a"b': None, 'a word': None}
751 ip.user_ns['d'] = {'a\nb': None, 'a\'b': None, 'a"b': None, 'a word': None}
752 _, matches = complete(line_buffer="d['a")
752 _, matches = complete(line_buffer="d['a")
753 nt.assert_in("a\\nb", matches)
753 nt.assert_in("a\\nb", matches)
754 nt.assert_in("a\\'b", matches)
754 nt.assert_in("a\\'b", matches)
755 nt.assert_in("a\"b", matches)
755 nt.assert_in("a\"b", matches)
756 nt.assert_in("a word", matches)
756 nt.assert_in("a word", matches)
757 assert not any(m.endswith((']', '"', "'")) for m in matches), matches
757 assert not any(m.endswith((']', '"', "'")) for m in matches), matches
758
758
759 # - can complete on non-initial word of the string
759 # - can complete on non-initial word of the string
760 _, matches = complete(line_buffer="d['a w")
760 _, matches = complete(line_buffer="d['a w")
761 nt.assert_in("word", matches)
761 nt.assert_in("word", matches)
762
762
763 # - understands quote escaping
763 # - understands quote escaping
764 _, matches = complete(line_buffer="d['a\\'")
764 _, matches = complete(line_buffer="d['a\\'")
765 nt.assert_in("b", matches)
765 nt.assert_in("b", matches)
766
766
767 # - default quoting should work like repr
767 # - default quoting should work like repr
768 _, matches = complete(line_buffer="d[")
768 _, matches = complete(line_buffer="d[")
769 nt.assert_in("\"a'b\"", matches)
769 nt.assert_in("\"a'b\"", matches)
770
770
771 # - when opening quote with ", possible to match with unescaped apostrophe
771 # - when opening quote with ", possible to match with unescaped apostrophe
772 _, matches = complete(line_buffer="d[\"a'")
772 _, matches = complete(line_buffer="d[\"a'")
773 nt.assert_in("b", matches)
773 nt.assert_in("b", matches)
774
774
775 # need to not split at delims that readline won't split at
775 # need to not split at delims that readline won't split at
776 if '-' not in ip.Completer.splitter.delims:
776 if '-' not in ip.Completer.splitter.delims:
777 ip.user_ns['d'] = {'before-after': None}
777 ip.user_ns['d'] = {'before-after': None}
778 _, matches = complete(line_buffer="d['before-af")
778 _, matches = complete(line_buffer="d['before-af")
779 nt.assert_in('before-after', matches)
779 nt.assert_in('before-after', matches)
780
780
781 def test_dict_key_completion_contexts():
781 def test_dict_key_completion_contexts():
782 """Test expression contexts in which dict key completion occurs"""
782 """Test expression contexts in which dict key completion occurs"""
783 ip = get_ipython()
783 ip = get_ipython()
784 complete = ip.Completer.complete
784 complete = ip.Completer.complete
785 d = {'abc': None}
785 d = {'abc': None}
786 ip.user_ns['d'] = d
786 ip.user_ns['d'] = d
787
787
788 class C:
788 class C:
789 data = d
789 data = d
790 ip.user_ns['C'] = C
790 ip.user_ns['C'] = C
791 ip.user_ns['get'] = lambda: d
791 ip.user_ns['get'] = lambda: d
792
792
793 def assert_no_completion(**kwargs):
793 def assert_no_completion(**kwargs):
794 _, matches = complete(**kwargs)
794 _, matches = complete(**kwargs)
795 nt.assert_not_in('abc', matches)
795 nt.assert_not_in('abc', matches)
796 nt.assert_not_in('abc\'', matches)
796 nt.assert_not_in('abc\'', matches)
797 nt.assert_not_in('abc\']', matches)
797 nt.assert_not_in('abc\']', matches)
798 nt.assert_not_in('\'abc\'', matches)
798 nt.assert_not_in('\'abc\'', matches)
799 nt.assert_not_in('\'abc\']', matches)
799 nt.assert_not_in('\'abc\']', matches)
800
800
801 def assert_completion(**kwargs):
801 def assert_completion(**kwargs):
802 _, matches = complete(**kwargs)
802 _, matches = complete(**kwargs)
803 nt.assert_in("'abc'", matches)
803 nt.assert_in("'abc'", matches)
804 nt.assert_not_in("'abc']", matches)
804 nt.assert_not_in("'abc']", matches)
805
805
806 # no completion after string closed, even if reopened
806 # no completion after string closed, even if reopened
807 assert_no_completion(line_buffer="d['a'")
807 assert_no_completion(line_buffer="d['a'")
808 assert_no_completion(line_buffer="d[\"a\"")
808 assert_no_completion(line_buffer="d[\"a\"")
809 assert_no_completion(line_buffer="d['a' + ")
809 assert_no_completion(line_buffer="d['a' + ")
810 assert_no_completion(line_buffer="d['a' + '")
810 assert_no_completion(line_buffer="d['a' + '")
811
811
812 # completion in non-trivial expressions
812 # completion in non-trivial expressions
813 assert_completion(line_buffer="+ d[")
813 assert_completion(line_buffer="+ d[")
814 assert_completion(line_buffer="(d[")
814 assert_completion(line_buffer="(d[")
815 assert_completion(line_buffer="C.data[")
815 assert_completion(line_buffer="C.data[")
816
816
817 # greedy flag
817 # greedy flag
818 def assert_completion(**kwargs):
818 def assert_completion(**kwargs):
819 _, matches = complete(**kwargs)
819 _, matches = complete(**kwargs)
820 nt.assert_in("get()['abc']", matches)
820 nt.assert_in("get()['abc']", matches)
821
821
822 assert_no_completion(line_buffer="get()[")
822 assert_no_completion(line_buffer="get()[")
823 with greedy_completion():
823 with greedy_completion():
824 assert_completion(line_buffer="get()[")
824 assert_completion(line_buffer="get()[")
825 assert_completion(line_buffer="get()['")
825 assert_completion(line_buffer="get()['")
826 assert_completion(line_buffer="get()['a")
826 assert_completion(line_buffer="get()['a")
827 assert_completion(line_buffer="get()['ab")
827 assert_completion(line_buffer="get()['ab")
828 assert_completion(line_buffer="get()['abc")
828 assert_completion(line_buffer="get()['abc")
829
829
830
830
831
831
832 def test_dict_key_completion_bytes():
832 def test_dict_key_completion_bytes():
833 """Test handling of bytes in dict key completion"""
833 """Test handling of bytes in dict key completion"""
834 ip = get_ipython()
834 ip = get_ipython()
835 complete = ip.Completer.complete
835 complete = ip.Completer.complete
836
836
837 ip.user_ns['d'] = {'abc': None, b'abd': None}
837 ip.user_ns['d'] = {'abc': None, b'abd': None}
838
838
839 _, matches = complete(line_buffer="d[")
839 _, matches = complete(line_buffer="d[")
840 nt.assert_in("'abc'", matches)
840 nt.assert_in("'abc'", matches)
841 nt.assert_in("b'abd'", matches)
841 nt.assert_in("b'abd'", matches)
842
842
843 if False: # not currently implemented
843 if False: # not currently implemented
844 _, matches = complete(line_buffer="d[b")
844 _, matches = complete(line_buffer="d[b")
845 nt.assert_in("b'abd'", matches)
845 nt.assert_in("b'abd'", matches)
846 nt.assert_not_in("b'abc'", matches)
846 nt.assert_not_in("b'abc'", matches)
847
847
848 _, matches = complete(line_buffer="d[b'")
848 _, matches = complete(line_buffer="d[b'")
849 nt.assert_in("abd", matches)
849 nt.assert_in("abd", matches)
850 nt.assert_not_in("abc", matches)
850 nt.assert_not_in("abc", matches)
851
851
852 _, matches = complete(line_buffer="d[B'")
852 _, matches = complete(line_buffer="d[B'")
853 nt.assert_in("abd", matches)
853 nt.assert_in("abd", matches)
854 nt.assert_not_in("abc", matches)
854 nt.assert_not_in("abc", matches)
855
855
856 _, matches = complete(line_buffer="d['")
856 _, matches = complete(line_buffer="d['")
857 nt.assert_in("abc", matches)
857 nt.assert_in("abc", matches)
858 nt.assert_not_in("abd", matches)
858 nt.assert_not_in("abd", matches)
859
859
860
860
861 def test_dict_key_completion_unicode_py3():
861 def test_dict_key_completion_unicode_py3():
862 """Test handling of unicode in dict key completion"""
862 """Test handling of unicode in dict key completion"""
863 ip = get_ipython()
863 ip = get_ipython()
864 complete = ip.Completer.complete
864 complete = ip.Completer.complete
865
865
866 ip.user_ns['d'] = {u'a\u05d0': None}
866 ip.user_ns['d'] = {u'a\u05d0': None}
867
867
868 # query using escape
868 # query using escape
869 if sys.platform != 'win32':
869 if sys.platform != 'win32':
870 # Known failure on Windows
870 # Known failure on Windows
871 _, matches = complete(line_buffer="d['a\\u05d0")
871 _, matches = complete(line_buffer="d['a\\u05d0")
872 nt.assert_in("u05d0", matches) # tokenized after \\
872 nt.assert_in("u05d0", matches) # tokenized after \\
873
873
874 # query using character
874 # query using character
875 _, matches = complete(line_buffer="d['a\u05d0")
875 _, matches = complete(line_buffer="d['a\u05d0")
876 nt.assert_in(u"a\u05d0", matches)
876 nt.assert_in(u"a\u05d0", matches)
877
877
878 with greedy_completion():
878 with greedy_completion():
879 # query using escape
879 # query using escape
880 _, matches = complete(line_buffer="d['a\\u05d0")
880 _, matches = complete(line_buffer="d['a\\u05d0")
881 nt.assert_in("d['a\\u05d0']", matches) # tokenized after \\
881 nt.assert_in("d['a\\u05d0']", matches) # tokenized after \\
882
882
883 # query using character
883 # query using character
884 _, matches = complete(line_buffer="d['a\u05d0")
884 _, matches = complete(line_buffer="d['a\u05d0")
885 nt.assert_in(u"d['a\u05d0']", matches)
885 nt.assert_in(u"d['a\u05d0']", matches)
886
886
887
887
888
888
889 @dec.skip_without('numpy')
889 @dec.skip_without('numpy')
890 def test_struct_array_key_completion():
890 def test_struct_array_key_completion():
891 """Test dict key completion applies to numpy struct arrays"""
891 """Test dict key completion applies to numpy struct arrays"""
892 import numpy
892 import numpy
893 ip = get_ipython()
893 ip = get_ipython()
894 complete = ip.Completer.complete
894 complete = ip.Completer.complete
895 ip.user_ns['d'] = numpy.array([], dtype=[('hello', 'f'), ('world', 'f')])
895 ip.user_ns['d'] = numpy.array([], dtype=[('hello', 'f'), ('world', 'f')])
896 _, matches = complete(line_buffer="d['")
896 _, matches = complete(line_buffer="d['")
897 nt.assert_in("hello", matches)
897 nt.assert_in("hello", matches)
898 nt.assert_in("world", matches)
898 nt.assert_in("world", matches)
899 # complete on the numpy struct itself
899 # complete on the numpy struct itself
900 dt = numpy.dtype([('my_head', [('my_dt', '>u4'), ('my_df', '>u4')]),
900 dt = numpy.dtype([('my_head', [('my_dt', '>u4'), ('my_df', '>u4')]),
901 ('my_data', '>f4', 5)])
901 ('my_data', '>f4', 5)])
902 x = numpy.zeros(2, dtype=dt)
902 x = numpy.zeros(2, dtype=dt)
903 ip.user_ns['d'] = x[1]
903 ip.user_ns['d'] = x[1]
904 _, matches = complete(line_buffer="d['")
904 _, matches = complete(line_buffer="d['")
905 nt.assert_in("my_head", matches)
905 nt.assert_in("my_head", matches)
906 nt.assert_in("my_data", matches)
906 nt.assert_in("my_data", matches)
907 # complete on a nested level
907 # complete on a nested level
908 with greedy_completion():
908 with greedy_completion():
909 ip.user_ns['d'] = numpy.zeros(2, dtype=dt)
909 ip.user_ns['d'] = numpy.zeros(2, dtype=dt)
910 _, matches = complete(line_buffer="d[1]['my_head']['")
910 _, matches = complete(line_buffer="d[1]['my_head']['")
911 nt.assert_true(any(["my_dt" in m for m in matches]))
911 nt.assert_true(any(["my_dt" in m for m in matches]))
912 nt.assert_true(any(["my_df" in m for m in matches]))
912 nt.assert_true(any(["my_df" in m for m in matches]))
913
913
914
914
915 @dec.skip_without('pandas')
915 @dec.skip_without('pandas')
916 def test_dataframe_key_completion():
916 def test_dataframe_key_completion():
917 """Test dict key completion applies to pandas DataFrames"""
917 """Test dict key completion applies to pandas DataFrames"""
918 import pandas
918 import pandas
919 ip = get_ipython()
919 ip = get_ipython()
920 complete = ip.Completer.complete
920 complete = ip.Completer.complete
921 ip.user_ns['d'] = pandas.DataFrame({'hello': [1], 'world': [2]})
921 ip.user_ns['d'] = pandas.DataFrame({'hello': [1], 'world': [2]})
922 _, matches = complete(line_buffer="d['")
922 _, matches = complete(line_buffer="d['")
923 nt.assert_in("hello", matches)
923 nt.assert_in("hello", matches)
924 nt.assert_in("world", matches)
924 nt.assert_in("world", matches)
925
925
926
926
927 def test_dict_key_completion_invalids():
927 def test_dict_key_completion_invalids():
928 """Smoke test cases dict key completion can't handle"""
928 """Smoke test cases dict key completion can't handle"""
929 ip = get_ipython()
929 ip = get_ipython()
930 complete = ip.Completer.complete
930 complete = ip.Completer.complete
931
931
932 ip.user_ns['no_getitem'] = None
932 ip.user_ns['no_getitem'] = None
933 ip.user_ns['no_keys'] = []
933 ip.user_ns['no_keys'] = []
934 ip.user_ns['cant_call_keys'] = dict
934 ip.user_ns['cant_call_keys'] = dict
935 ip.user_ns['empty'] = {}
935 ip.user_ns['empty'] = {}
936 ip.user_ns['d'] = {'abc': 5}
936 ip.user_ns['d'] = {'abc': 5}
937
937
938 _, matches = complete(line_buffer="no_getitem['")
938 _, matches = complete(line_buffer="no_getitem['")
939 _, matches = complete(line_buffer="no_keys['")
939 _, matches = complete(line_buffer="no_keys['")
940 _, matches = complete(line_buffer="cant_call_keys['")
940 _, matches = complete(line_buffer="cant_call_keys['")
941 _, matches = complete(line_buffer="empty['")
941 _, matches = complete(line_buffer="empty['")
942 _, matches = complete(line_buffer="name_error['")
942 _, matches = complete(line_buffer="name_error['")
943 _, matches = complete(line_buffer="d['\\") # incomplete escape
943 _, matches = complete(line_buffer="d['\\") # incomplete escape
944
944
945 class KeyCompletable(object):
945 class KeyCompletable(object):
946 def __init__(self, things=()):
946 def __init__(self, things=()):
947 self.things = things
947 self.things = things
948
948
949 def _ipython_key_completions_(self):
949 def _ipython_key_completions_(self):
950 return list(self.things)
950 return list(self.things)
951
951
952 def test_object_key_completion():
952 def test_object_key_completion():
953 ip = get_ipython()
953 ip = get_ipython()
954 ip.user_ns['key_completable'] = KeyCompletable(['qwerty', 'qwick'])
954 ip.user_ns['key_completable'] = KeyCompletable(['qwerty', 'qwick'])
955
955
956 _, matches = ip.Completer.complete(line_buffer="key_completable['qw")
956 _, matches = ip.Completer.complete(line_buffer="key_completable['qw")
957 nt.assert_in('qwerty', matches)
957 nt.assert_in('qwerty', matches)
958 nt.assert_in('qwick', matches)
958 nt.assert_in('qwick', matches)
959
959
960
960
961 class NamedInstanceMetaclass(type):
961 class NamedInstanceMetaclass(type):
962 def __getitem__(cls, item):
962 def __getitem__(cls, item):
963 return cls.get_instance(item)
963 return cls.get_instance(item)
964
964
965 class NamedInstanceClass(object, metaclass=NamedInstanceMetaclass):
965 class NamedInstanceClass(object, metaclass=NamedInstanceMetaclass):
966 def __init__(self, name):
966 def __init__(self, name):
967 if not hasattr(self.__class__, 'instances'):
967 if not hasattr(self.__class__, 'instances'):
968 self.__class__.instances = {}
968 self.__class__.instances = {}
969 self.__class__.instances[name] = self
969 self.__class__.instances[name] = self
970
970
971 @classmethod
971 @classmethod
972 def _ipython_key_completions_(cls):
972 def _ipython_key_completions_(cls):
973 return cls.instances.keys()
973 return cls.instances.keys()
974
974
975 @classmethod
975 @classmethod
976 def get_instance(cls, name):
976 def get_instance(cls, name):
977 return cls.instances[name]
977 return cls.instances[name]
978
978
979 def test_class_key_completion():
979 def test_class_key_completion():
980 ip = get_ipython()
980 ip = get_ipython()
981 NamedInstanceClass('qwerty')
981 NamedInstanceClass('qwerty')
982 NamedInstanceClass('qwick')
982 NamedInstanceClass('qwick')
983 ip.user_ns['named_instance_class'] = NamedInstanceClass
983 ip.user_ns['named_instance_class'] = NamedInstanceClass
984
984
985 _, matches = ip.Completer.complete(line_buffer="named_instance_class['qw")
985 _, matches = ip.Completer.complete(line_buffer="named_instance_class['qw")
986 nt.assert_in('qwerty', matches)
986 nt.assert_in('qwerty', matches)
987 nt.assert_in('qwick', matches)
987 nt.assert_in('qwick', matches)
988
988
989 def test_tryimport():
989 def test_tryimport():
990 """
990 """
991 Test that try-import don't crash on trailing dot, and import modules before
991 Test that try-import don't crash on trailing dot, and import modules before
992 """
992 """
993 from IPython.core.completerlib import try_import
993 from IPython.core.completerlib import try_import
994 assert(try_import("IPython."))
994 assert(try_import("IPython."))
995
995
996
996
997 def test_aimport_module_completer():
997 def test_aimport_module_completer():
998 ip = get_ipython()
998 ip = get_ipython()
999 _, matches = ip.complete('i', '%aimport i')
999 _, matches = ip.complete('i', '%aimport i')
1000 nt.assert_in('io', matches)
1000 nt.assert_in('io', matches)
1001 nt.assert_not_in('int', matches)
1001 nt.assert_not_in('int', matches)
1002
1002
1003 def test_nested_import_module_completer():
1003 def test_nested_import_module_completer():
1004 ip = get_ipython()
1004 ip = get_ipython()
1005 _, matches = ip.complete(None, 'import IPython.co', 17)
1005 _, matches = ip.complete(None, 'import IPython.co', 17)
1006 nt.assert_in('IPython.core', matches)
1006 nt.assert_in('IPython.core', matches)
1007 nt.assert_not_in('import IPython.core', matches)
1007 nt.assert_not_in('import IPython.core', matches)
1008 nt.assert_not_in('IPython.display', matches)
1008 nt.assert_not_in('IPython.display', matches)
1009
1009
1010 def test_import_module_completer():
1010 def test_import_module_completer():
1011 ip = get_ipython()
1011 ip = get_ipython()
1012 _, matches = ip.complete('i', 'import i')
1012 _, matches = ip.complete('i', 'import i')
1013 nt.assert_in('io', matches)
1013 nt.assert_in('io', matches)
1014 nt.assert_not_in('int', matches)
1014 nt.assert_not_in('int', matches)
1015
1015
1016 def test_from_module_completer():
1016 def test_from_module_completer():
1017 ip = get_ipython()
1017 ip = get_ipython()
1018 _, matches = ip.complete('B', 'from io import B', 16)
1018 _, matches = ip.complete('B', 'from io import B', 16)
1019 nt.assert_in('BytesIO', matches)
1019 nt.assert_in('BytesIO', matches)
1020 nt.assert_not_in('BaseException', matches)
1020 nt.assert_not_in('BaseException', matches)
1021
1021
1022 def test_snake_case_completion():
1022 def test_snake_case_completion():
1023 ip = get_ipython()
1023 ip = get_ipython()
1024 ip.Completer.use_jedi = False
1024 ip.Completer.use_jedi = False
1025 ip.user_ns['some_three'] = 3
1025 ip.user_ns['some_three'] = 3
1026 ip.user_ns['some_four'] = 4
1026 ip.user_ns['some_four'] = 4
1027 _, matches = ip.complete("s_", "print(s_f")
1027 _, matches = ip.complete("s_", "print(s_f")
1028 nt.assert_in('some_three', matches)
1028 nt.assert_in('some_three', matches)
1029 nt.assert_in('some_four', matches)
1029 nt.assert_in('some_four', matches)
1030
1031 def test_mix_terms():
1032 ip = get_ipython()
1033 from textwrap import dedent
1034 ip.Completer.use_jedi = False
1035 ip.ex(dedent("""
1036 class Test:
1037 def meth(self, meth_arg1):
1038 print("meth")
1039
1040 def meth_1(self, meth1_arg1, meth1_arg2):
1041 print("meth1")
1042
1043 def meth_2(self, meth2_arg1, meth2_arg2):
1044 print("meth2")
1045 test = Test()
1046 """))
1047 _, matches = ip.complete(None, "test.meth(")
1048 nt.assert_in('meth_arg1=', matches)
1049 nt.assert_not_in('meth2_arg1=', matches)
General Comments 0
You need to be logged in to leave comments. Login now