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