Merge branch 'master' into comment-cache
[emacs.git] / lisp / progmodes / python.el
blob90b5e4e0dc67e1dcc2663cbe49a4c7e167680d16
1 ;;; python.el --- Python's flying circus support for Emacs -*- lexical-binding: t -*-
3 ;; Copyright (C) 2003-2017 Free Software Foundation, Inc.
5 ;; Author: Fabián E. Gallina <fgallina@gnu.org>
6 ;; URL: https://github.com/fgallina/python.el
7 ;; Version: 0.25.2
8 ;; Package-Requires: ((emacs "24.1") (cl-lib "1.0"))
9 ;; Maintainer: emacs-devel@gnu.org
10 ;; Created: Jul 2010
11 ;; Keywords: languages
13 ;; This file is part of GNU Emacs.
15 ;; GNU Emacs is free software: you can redistribute it and/or modify
16 ;; it under the terms of the GNU General Public License as published
17 ;; by the Free Software Foundation, either version 3 of the License,
18 ;; or (at your option) any later version.
20 ;; GNU Emacs is distributed in the hope that it will be useful, but
21 ;; WITHOUT ANY WARRANTY; without even the implied warranty of
22 ;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
23 ;; General Public License for more details.
25 ;; You should have received a copy of the GNU General Public License
26 ;; along with GNU Emacs. If not, see <http://www.gnu.org/licenses/>.
28 ;;; Commentary:
30 ;; Major mode for editing Python files with some fontification and
31 ;; indentation bits extracted from original Dave Love's python.el
32 ;; found in GNU/Emacs.
34 ;; Implements Syntax highlighting, Indentation, Movement, Shell
35 ;; interaction, Shell completion, Shell virtualenv support, Shell
36 ;; package support, Shell syntax highlighting, Pdb tracking, Symbol
37 ;; completion, Skeletons, FFAP, Code Check, Eldoc, Imenu.
39 ;; Syntax highlighting: Fontification of code is provided and supports
40 ;; python's triple quoted strings properly.
42 ;; Indentation: Automatic indentation with indentation cycling is
43 ;; provided, it allows you to navigate different available levels of
44 ;; indentation by hitting <tab> several times. Also electric-indent-mode
45 ;; is supported such that when inserting a colon the current line is
46 ;; dedented automatically if needed.
48 ;; Movement: `beginning-of-defun' and `end-of-defun' functions are
49 ;; properly implemented. There are also specialized
50 ;; `forward-sentence' and `backward-sentence' replacements called
51 ;; `python-nav-forward-block', `python-nav-backward-block'
52 ;; respectively which navigate between beginning of blocks of code.
53 ;; Extra functions `python-nav-forward-statement',
54 ;; `python-nav-backward-statement',
55 ;; `python-nav-beginning-of-statement', `python-nav-end-of-statement',
56 ;; `python-nav-beginning-of-block', `python-nav-end-of-block' and
57 ;; `python-nav-if-name-main' are included but no bound to any key. At
58 ;; last but not least the specialized `python-nav-forward-sexp' allows
59 ;; easy navigation between code blocks. If you prefer `cc-mode'-like
60 ;; `forward-sexp' movement, setting `forward-sexp-function' to nil is
61 ;; enough, You can do that using the `python-mode-hook':
63 ;; (add-hook 'python-mode-hook
64 ;; (lambda () (setq forward-sexp-function nil)))
66 ;; Shell interaction: is provided and allows opening Python shells
67 ;; inside Emacs and executing any block of code of your current buffer
68 ;; in that inferior Python process.
70 ;; Besides that only the standard CPython (2.x and 3.x) shell and
71 ;; IPython are officially supported out of the box, the interaction
72 ;; should support any other readline based Python shells as well
73 ;; (e.g. Jython and PyPy have been reported to work). You can change
74 ;; your default interpreter and commandline arguments by setting the
75 ;; `python-shell-interpreter' and `python-shell-interpreter-args'
76 ;; variables. This example enables IPython globally:
78 ;; (setq python-shell-interpreter "ipython"
79 ;; python-shell-interpreter-args "-i")
81 ;; Using the "console" subcommand to start IPython in server-client
82 ;; mode is known to fail intermittently due a bug on IPython itself
83 ;; (see URL `http://debbugs.gnu.org/cgi/bugreport.cgi?bug=18052#27').
84 ;; There seems to be a race condition in the IPython server (A.K.A
85 ;; kernel) when code is sent while it is still initializing, sometimes
86 ;; causing the shell to get stalled. With that said, if an IPython
87 ;; kernel is already running, "console --existing" seems to work fine.
89 ;; Running IPython on Windows needs more tweaking. The way you should
90 ;; set `python-shell-interpreter' and `python-shell-interpreter-args'
91 ;; is as follows (of course you need to modify the paths according to
92 ;; your system):
94 ;; (setq python-shell-interpreter "C:\\Python27\\python.exe"
95 ;; python-shell-interpreter-args
96 ;; "-i C:\\Python27\\Scripts\\ipython-script.py")
98 ;; Missing or delayed output used to happen due to differences between
99 ;; Operating Systems' pipe buffering (e.g. CPython 3.3.4 in Windows 7.
100 ;; See URL `http://debbugs.gnu.org/cgi/bugreport.cgi?bug=17304'). To
101 ;; avoid this, the `python-shell-unbuffered' defaults to non-nil and
102 ;; controls whether `python-shell-calculate-process-environment'
103 ;; should set the "PYTHONUNBUFFERED" environment variable on startup:
104 ;; See URL `https://docs.python.org/3/using/cmdline.html#cmdoption-u'.
106 ;; The interaction relies upon having prompts for input (e.g. ">>> "
107 ;; and "... " in standard Python shell) and output (e.g. "Out[1]: " in
108 ;; IPython) detected properly. Failing that Emacs may hang but, in
109 ;; the case that happens, you can recover with \\[keyboard-quit]. To
110 ;; avoid this issue, a two-step prompt autodetection mechanism is
111 ;; provided: the first step is manual and consists of a collection of
112 ;; regular expressions matching common prompts for Python shells
113 ;; stored in `python-shell-prompt-input-regexps' and
114 ;; `python-shell-prompt-output-regexps', and dir-local friendly vars
115 ;; `python-shell-prompt-regexp', `python-shell-prompt-block-regexp',
116 ;; `python-shell-prompt-output-regexp' which are appended to the
117 ;; former automatically when a shell spawns; the second step is
118 ;; automatic and depends on the `python-shell-prompt-detect' helper
119 ;; function. See its docstring for details on global variables that
120 ;; modify its behavior.
122 ;; Shell completion: hitting tab will try to complete the current
123 ;; word. The two built-in mechanisms depend on Python's readline
124 ;; module: the "native" completion is tried first and is activated
125 ;; when `python-shell-completion-native-enable' is non-nil, the
126 ;; current `python-shell-interpreter' is not a member of the
127 ;; `python-shell-completion-native-disabled-interpreters' variable and
128 ;; `python-shell-completion-native-setup' succeeds; the "fallback" or
129 ;; "legacy" mechanism works by executing Python code in the background
130 ;; and enables auto-completion for shells that do not support
131 ;; receiving escape sequences (with some limitations, i.e. completion
132 ;; in blocks does not work). The code executed for the "fallback"
133 ;; completion can be found in `python-shell-completion-setup-code' and
134 ;; `python-shell-completion-string-code' variables. Their default
135 ;; values enable completion for both CPython and IPython, and probably
136 ;; any readline based shell (it's known to work with PyPy). If your
137 ;; Python installation lacks readline (like CPython for Windows),
138 ;; installing pyreadline (URL `http://ipython.org/pyreadline.html')
139 ;; should suffice. To troubleshoot why you are not getting any
140 ;; completions, you can try the following in your Python shell:
142 ;; >>> import readline, rlcompleter
144 ;; If you see an error, then you need to either install pyreadline or
145 ;; setup custom code that avoids that dependency.
147 ;; Shell virtualenv support: The shell also contains support for
148 ;; virtualenvs and other special environment modifications thanks to
149 ;; `python-shell-process-environment' and `python-shell-exec-path'.
150 ;; These two variables allows you to modify execution paths and
151 ;; environment variables to make easy for you to setup virtualenv rules
152 ;; or behavior modifications when running shells. Here is an example
153 ;; of how to make shell processes to be run using the /path/to/env/
154 ;; virtualenv:
156 ;; (setq python-shell-process-environment
157 ;; (list
158 ;; (format "PATH=%s" (mapconcat
159 ;; 'identity
160 ;; (reverse
161 ;; (cons (getenv "PATH")
162 ;; '("/path/to/env/bin/")))
163 ;; ":"))
164 ;; "VIRTUAL_ENV=/path/to/env/"))
165 ;; (python-shell-exec-path . ("/path/to/env/bin/"))
167 ;; Since the above is cumbersome and can be programmatically
168 ;; calculated, the variable `python-shell-virtualenv-root' is
169 ;; provided. When this variable is set with the path of the
170 ;; virtualenv to use, `process-environment' and `exec-path' get proper
171 ;; values in order to run shells inside the specified virtualenv. So
172 ;; the following will achieve the same as the previous example:
174 ;; (setq python-shell-virtualenv-root "/path/to/env/")
176 ;; Also the `python-shell-extra-pythonpaths' variable have been
177 ;; introduced as simple way of adding paths to the PYTHONPATH without
178 ;; affecting existing values.
180 ;; Shell package support: you can enable a package in the current
181 ;; shell so that relative imports work properly using the
182 ;; `python-shell-package-enable' command.
184 ;; Shell remote support: remote Python shells are started with the
185 ;; correct environment for files opened remotely through tramp, also
186 ;; respecting dir-local variables provided `enable-remote-dir-locals'
187 ;; is non-nil. The logic for this is transparently handled by the
188 ;; `python-shell-with-environment' macro.
190 ;; Shell syntax highlighting: when enabled current input in shell is
191 ;; highlighted. The variable `python-shell-font-lock-enable' controls
192 ;; activation of this feature globally when shells are started.
193 ;; Activation/deactivation can be also controlled on the fly via the
194 ;; `python-shell-font-lock-toggle' command.
196 ;; Pdb tracking: when you execute a block of code that contains some
197 ;; call to pdb (or ipdb) it will prompt the block of code and will
198 ;; follow the execution of pdb marking the current line with an arrow.
200 ;; Symbol completion: you can complete the symbol at point. It uses
201 ;; the shell completion in background so you should run
202 ;; `python-shell-send-buffer' from time to time to get better results.
204 ;; Skeletons: skeletons are provided for simple inserting of things like class,
205 ;; def, for, import, if, try, and while. These skeletons are
206 ;; integrated with abbrev. If you have `abbrev-mode' activated and
207 ;; `python-skeleton-autoinsert' is set to t, then whenever you type
208 ;; the name of any of those defined and hit SPC, they will be
209 ;; automatically expanded. As an alternative you can use the defined
210 ;; skeleton commands: `python-skeleton-<foo>'.
212 ;; FFAP: You can find the filename for a given module when using ffap
213 ;; out of the box. This feature needs an inferior python shell
214 ;; running.
216 ;; Code check: Check the current file for errors with `python-check'
217 ;; using the program defined in `python-check-command'.
219 ;; Eldoc: returns documentation for object at point by using the
220 ;; inferior python subprocess to inspect its documentation. As you
221 ;; might guessed you should run `python-shell-send-buffer' from time
222 ;; to time to get better results too.
224 ;; Imenu: There are two index building functions to be used as
225 ;; `imenu-create-index-function': `python-imenu-create-index' (the
226 ;; default one, builds the alist in form of a tree) and
227 ;; `python-imenu-create-flat-index'. See also
228 ;; `python-imenu-format-item-label-function',
229 ;; `python-imenu-format-parent-item-label-function',
230 ;; `python-imenu-format-parent-item-jump-label-function' variables for
231 ;; changing the way labels are formatted in the tree version.
233 ;; If you used python-mode.el you may miss auto-indentation when
234 ;; inserting newlines. To achieve the same behavior you have two
235 ;; options:
237 ;; 1) Enable the minor-mode `electric-indent-mode' (enabled by
238 ;; default) and use RET. If this mode is disabled use
239 ;; `newline-and-indent', bound to C-j.
241 ;; 2) Add the following hook in your .emacs:
243 ;; (add-hook 'python-mode-hook
244 ;; #'(lambda ()
245 ;; (define-key python-mode-map "\C-m" 'newline-and-indent)))
247 ;; I'd recommend the first one since you'll get the same behavior for
248 ;; all modes out-of-the-box.
250 ;;; Installation:
252 ;; Add this to your .emacs:
254 ;; (add-to-list 'load-path "/folder/containing/file")
255 ;; (require 'python)
257 ;;; TODO:
259 ;;; Code:
261 (require 'ansi-color)
262 (require 'cl-lib)
263 (require 'comint)
264 (require 'json)
265 (require 'tramp-sh)
267 ;; Avoid compiler warnings
268 (defvar view-return-to-alist)
269 (defvar compilation-error-regexp-alist)
270 (defvar outline-heading-end-regexp)
272 (autoload 'comint-mode "comint")
273 (autoload 'help-function-arglist "help-fns")
275 ;;;###autoload
276 (add-to-list 'auto-mode-alist (cons (purecopy "\\.pyw?\\'") 'python-mode))
277 ;;;###autoload
278 (add-to-list 'interpreter-mode-alist (cons (purecopy "python[0-9.]*") 'python-mode))
280 (defgroup python nil
281 "Python Language's flying circus support for Emacs."
282 :group 'languages
283 :version "24.3"
284 :link '(emacs-commentary-link "python"))
287 ;;; 24.x Compat
290 (unless (fboundp 'prog-widen)
291 (defun prog-widen ()
292 (widen)))
294 (unless (fboundp 'prog-first-column)
295 (defun prog-first-column ()
299 ;;; Bindings
301 (defvar python-mode-map
302 (let ((map (make-sparse-keymap)))
303 ;; Movement
304 (define-key map [remap backward-sentence] 'python-nav-backward-block)
305 (define-key map [remap forward-sentence] 'python-nav-forward-block)
306 (define-key map [remap backward-up-list] 'python-nav-backward-up-list)
307 (define-key map [remap mark-defun] 'python-mark-defun)
308 (define-key map "\C-c\C-j" 'imenu)
309 ;; Indent specific
310 (define-key map "\177" 'python-indent-dedent-line-backspace)
311 (define-key map (kbd "<backtab>") 'python-indent-dedent-line)
312 (define-key map "\C-c<" 'python-indent-shift-left)
313 (define-key map "\C-c>" 'python-indent-shift-right)
314 ;; Skeletons
315 (define-key map "\C-c\C-tc" 'python-skeleton-class)
316 (define-key map "\C-c\C-td" 'python-skeleton-def)
317 (define-key map "\C-c\C-tf" 'python-skeleton-for)
318 (define-key map "\C-c\C-ti" 'python-skeleton-if)
319 (define-key map "\C-c\C-tm" 'python-skeleton-import)
320 (define-key map "\C-c\C-tt" 'python-skeleton-try)
321 (define-key map "\C-c\C-tw" 'python-skeleton-while)
322 ;; Shell interaction
323 (define-key map "\C-c\C-p" 'run-python)
324 (define-key map "\C-c\C-s" 'python-shell-send-string)
325 (define-key map "\C-c\C-r" 'python-shell-send-region)
326 (define-key map "\C-\M-x" 'python-shell-send-defun)
327 (define-key map "\C-c\C-c" 'python-shell-send-buffer)
328 (define-key map "\C-c\C-l" 'python-shell-send-file)
329 (define-key map "\C-c\C-z" 'python-shell-switch-to-shell)
330 ;; Some util commands
331 (define-key map "\C-c\C-v" 'python-check)
332 (define-key map "\C-c\C-f" 'python-eldoc-at-point)
333 (define-key map "\C-c\C-d" 'python-describe-at-point)
334 ;; Utilities
335 (substitute-key-definition 'complete-symbol 'completion-at-point
336 map global-map)
337 (easy-menu-define python-menu map "Python Mode menu"
338 `("Python"
339 :help "Python-specific Features"
340 ["Shift region left" python-indent-shift-left :active mark-active
341 :help "Shift region left by a single indentation step"]
342 ["Shift region right" python-indent-shift-right :active mark-active
343 :help "Shift region right by a single indentation step"]
345 ["Start of def/class" beginning-of-defun
346 :help "Go to start of outermost definition around point"]
347 ["End of def/class" end-of-defun
348 :help "Go to end of definition around point"]
349 ["Mark def/class" mark-defun
350 :help "Mark outermost definition around point"]
351 ["Jump to def/class" imenu
352 :help "Jump to a class or function definition"]
353 "--"
354 ("Skeletons")
355 "---"
356 ["Start interpreter" run-python
357 :help "Run inferior Python process in a separate buffer"]
358 ["Switch to shell" python-shell-switch-to-shell
359 :help "Switch to running inferior Python process"]
360 ["Eval string" python-shell-send-string
361 :help "Eval string in inferior Python session"]
362 ["Eval buffer" python-shell-send-buffer
363 :help "Eval buffer in inferior Python session"]
364 ["Eval region" python-shell-send-region
365 :help "Eval region in inferior Python session"]
366 ["Eval defun" python-shell-send-defun
367 :help "Eval defun in inferior Python session"]
368 ["Eval file" python-shell-send-file
369 :help "Eval file in inferior Python session"]
370 ["Debugger" pdb :help "Run pdb under GUD"]
371 "----"
372 ["Check file" python-check
373 :help "Check file for errors"]
374 ["Help on symbol" python-eldoc-at-point
375 :help "Get help on symbol at point"]
376 ["Complete symbol" completion-at-point
377 :help "Complete symbol before point"]))
378 map)
379 "Keymap for `python-mode'.")
382 ;;; Python specialized rx
384 (eval-and-compile
385 (defconst python-rx-constituents
386 `((block-start . ,(rx symbol-start
387 (or "def" "class" "if" "elif" "else" "try"
388 "except" "finally" "for" "while" "with"
389 ;; Python 3.5+ PEP492
390 (and "async" (+ space)
391 (or "def" "for" "with")))
392 symbol-end))
393 (dedenter . ,(rx symbol-start
394 (or "elif" "else" "except" "finally")
395 symbol-end))
396 (block-ender . ,(rx symbol-start
398 "break" "continue" "pass" "raise" "return")
399 symbol-end))
400 (decorator . ,(rx line-start (* space) ?@ (any letter ?_)
401 (* (any word ?_))))
402 (defun . ,(rx symbol-start
403 (or "def" "class"
404 ;; Python 3.5+ PEP492
405 (and "async" (+ space) "def"))
406 symbol-end))
407 (if-name-main . ,(rx line-start "if" (+ space) "__name__"
408 (+ space) "==" (+ space)
409 (any ?' ?\") "__main__" (any ?' ?\")
410 (* space) ?:))
411 (symbol-name . ,(rx (any letter ?_) (* (any word ?_))))
412 (open-paren . ,(rx (or "{" "[" "(")))
413 (close-paren . ,(rx (or "}" "]" ")")))
414 (simple-operator . ,(rx (any ?+ ?- ?/ ?& ?^ ?~ ?| ?* ?< ?> ?= ?%)))
415 ;; FIXME: rx should support (not simple-operator).
416 (not-simple-operator . ,(rx
417 (not
418 (any ?+ ?- ?/ ?& ?^ ?~ ?| ?* ?< ?> ?= ?%))))
419 ;; FIXME: Use regexp-opt.
420 (operator . ,(rx (or "+" "-" "/" "&" "^" "~" "|" "*" "<" ">"
421 "=" "%" "**" "//" "<<" ">>" "<=" "!="
422 "==" ">=" "is" "not")))
423 ;; FIXME: Use regexp-opt.
424 (assignment-operator . ,(rx (or "=" "+=" "-=" "*=" "/=" "//=" "%=" "**="
425 ">>=" "<<=" "&=" "^=" "|=")))
426 (string-delimiter . ,(rx (and
427 ;; Match even number of backslashes.
428 (or (not (any ?\\ ?\' ?\")) point
429 ;; Quotes might be preceded by a escaped quote.
430 (and (or (not (any ?\\)) point) ?\\
431 (* ?\\ ?\\) (any ?\' ?\")))
432 (* ?\\ ?\\)
433 ;; Match single or triple quotes of any kind.
434 (group (or "\"" "\"\"\"" "'" "'''")))))
435 (coding-cookie . ,(rx line-start ?# (* space)
437 ;; # coding=<encoding name>
438 (: "coding" (or ?: ?=) (* space) (group-n 1 (+ (or word ?-))))
439 ;; # -*- coding: <encoding name> -*-
440 (: "-*-" (* space) "coding:" (* space)
441 (group-n 1 (+ (or word ?-))) (* space) "-*-")
442 ;; # vim: set fileencoding=<encoding name> :
443 (: "vim:" (* space) "set" (+ space)
444 "fileencoding" (* space) ?= (* space)
445 (group-n 1 (+ (or word ?-))) (* space) ":")))))
446 "Additional Python specific sexps for `python-rx'")
448 (defmacro python-rx (&rest regexps)
449 "Python mode specialized rx macro.
450 This variant of `rx' supports common Python named REGEXPS."
451 (let ((rx-constituents (append python-rx-constituents rx-constituents)))
452 (cond ((null regexps)
453 (error "No regexp"))
454 ((cdr regexps)
455 (rx-to-string `(and ,@regexps) t))
457 (rx-to-string (car regexps) t))))))
460 ;;; Font-lock and syntax
462 (eval-and-compile
463 (defun python-syntax--context-compiler-macro (form type &optional syntax-ppss)
464 (pcase type
465 (`'comment
466 `(let ((ppss (or ,syntax-ppss (syntax-ppss))))
467 (and (nth 4 ppss) (nth 8 ppss))))
468 (`'string
469 `(let ((ppss (or ,syntax-ppss (syntax-ppss))))
470 (and (nth 3 ppss) (nth 8 ppss))))
471 (`'paren
472 `(nth 1 (or ,syntax-ppss (syntax-ppss))))
473 (_ form))))
475 (defun python-syntax-context (type &optional syntax-ppss)
476 "Return non-nil if point is on TYPE using SYNTAX-PPSS.
477 TYPE can be `comment', `string' or `paren'. It returns the start
478 character address of the specified TYPE."
479 (declare (compiler-macro python-syntax--context-compiler-macro))
480 (let ((ppss (or syntax-ppss (syntax-ppss))))
481 (pcase type
482 (`comment (and (nth 4 ppss) (nth 8 ppss)))
483 (`string (and (nth 3 ppss) (nth 8 ppss)))
484 (`paren (nth 1 ppss))
485 (_ nil))))
487 (defun python-syntax-context-type (&optional syntax-ppss)
488 "Return the context type using SYNTAX-PPSS.
489 The type returned can be `comment', `string' or `paren'."
490 (let ((ppss (or syntax-ppss (syntax-ppss))))
491 (cond
492 ((nth 8 ppss) (if (nth 4 ppss) 'comment 'string))
493 ((nth 1 ppss) 'paren))))
495 (defsubst python-syntax-comment-or-string-p (&optional ppss)
496 "Return non-nil if PPSS is inside comment or string."
497 (nth 8 (or ppss (syntax-ppss))))
499 (defsubst python-syntax-closing-paren-p ()
500 "Return non-nil if char after point is a closing paren."
501 (eql (syntax-class (syntax-after (point)))
502 (syntax-class (string-to-syntax ")"))))
504 (define-obsolete-function-alias
505 'python-info-ppss-context #'python-syntax-context "24.3")
507 (define-obsolete-function-alias
508 'python-info-ppss-context-type #'python-syntax-context-type "24.3")
510 (define-obsolete-function-alias
511 'python-info-ppss-comment-or-string-p
512 #'python-syntax-comment-or-string-p "24.3")
514 (defun python-font-lock-syntactic-face-function (state)
515 "Return syntactic face given STATE."
516 (if (nth 3 state)
517 (if (python-info-docstring-p state)
518 font-lock-doc-face
519 font-lock-string-face)
520 font-lock-comment-face))
522 (defvar python-font-lock-keywords
523 ;; Keywords
524 `(,(rx symbol-start
526 "and" "del" "from" "not" "while" "as" "elif" "global" "or" "with"
527 "assert" "else" "if" "pass" "yield" "break" "except" "import" "class"
528 "in" "raise" "continue" "finally" "is" "return" "def" "for" "lambda"
529 "try"
530 ;; Python 2:
531 "print" "exec"
532 ;; Python 3:
533 ;; False, None, and True are listed as keywords on the Python 3
534 ;; documentation, but since they also qualify as constants they are
535 ;; fontified like that in order to keep font-lock consistent between
536 ;; Python versions.
537 "nonlocal"
538 ;; Python 3.5+ PEP492
539 (and "async" (+ space) (or "def" "for" "with"))
540 "await"
541 ;; Extra:
542 "self")
543 symbol-end)
544 ;; functions
545 (,(rx symbol-start "def" (1+ space) (group (1+ (or word ?_))))
546 (1 font-lock-function-name-face))
547 ;; classes
548 (,(rx symbol-start "class" (1+ space) (group (1+ (or word ?_))))
549 (1 font-lock-type-face))
550 ;; Constants
551 (,(rx symbol-start
553 "Ellipsis" "False" "None" "NotImplemented" "True" "__debug__"
554 ;; copyright, license, credits, quit and exit are added by the site
555 ;; module and they are not intended to be used in programs
556 "copyright" "credits" "exit" "license" "quit")
557 symbol-end) . font-lock-constant-face)
558 ;; Decorators.
559 (,(rx line-start (* (any " \t")) (group "@" (1+ (or word ?_))
560 (0+ "." (1+ (or word ?_)))))
561 (1 font-lock-type-face))
562 ;; Builtin Exceptions
563 (,(rx symbol-start
565 ;; Python 2 and 3:
566 "ArithmeticError" "AssertionError" "AttributeError" "BaseException"
567 "BufferError" "BytesWarning" "DeprecationWarning" "EOFError"
568 "EnvironmentError" "Exception" "FloatingPointError" "FutureWarning"
569 "GeneratorExit" "IOError" "ImportError" "ImportWarning"
570 "IndentationError" "IndexError" "KeyError" "KeyboardInterrupt"
571 "LookupError" "MemoryError" "NameError" "NotImplementedError"
572 "OSError" "OverflowError" "PendingDeprecationWarning"
573 "ReferenceError" "RuntimeError" "RuntimeWarning" "StopIteration"
574 "SyntaxError" "SyntaxWarning" "SystemError" "SystemExit" "TabError"
575 "TypeError" "UnboundLocalError" "UnicodeDecodeError"
576 "UnicodeEncodeError" "UnicodeError" "UnicodeTranslateError"
577 "UnicodeWarning" "UserWarning" "ValueError" "Warning"
578 "ZeroDivisionError"
579 ;; Python 2:
580 "StandardError"
581 ;; Python 3:
582 "BlockingIOError" "BrokenPipeError" "ChildProcessError"
583 "ConnectionAbortedError" "ConnectionError" "ConnectionRefusedError"
584 "ConnectionResetError" "FileExistsError" "FileNotFoundError"
585 "InterruptedError" "IsADirectoryError" "NotADirectoryError"
586 "PermissionError" "ProcessLookupError" "RecursionError"
587 "ResourceWarning" "StopAsyncIteration" "TimeoutError"
588 ;; OS specific
589 "VMSError" "WindowsError"
591 symbol-end) . font-lock-type-face)
592 ;; Builtins
593 (,(rx symbol-start
595 "abs" "all" "any" "bin" "bool" "callable" "chr" "classmethod"
596 "compile" "complex" "delattr" "dict" "dir" "divmod" "enumerate"
597 "eval" "filter" "float" "format" "frozenset" "getattr" "globals"
598 "hasattr" "hash" "help" "hex" "id" "input" "int" "isinstance"
599 "issubclass" "iter" "len" "list" "locals" "map" "max" "memoryview"
600 "min" "next" "object" "oct" "open" "ord" "pow" "print" "property"
601 "range" "repr" "reversed" "round" "set" "setattr" "slice" "sorted"
602 "staticmethod" "str" "sum" "super" "tuple" "type" "vars" "zip"
603 "__import__"
604 ;; Python 2:
605 "basestring" "cmp" "execfile" "file" "long" "raw_input" "reduce"
606 "reload" "unichr" "unicode" "xrange" "apply" "buffer" "coerce"
607 "intern"
608 ;; Python 3:
609 "ascii" "bytearray" "bytes" "exec"
610 ;; Extra:
611 "__all__" "__doc__" "__name__" "__package__")
612 symbol-end) . font-lock-builtin-face)
613 ;; assignments
614 ;; support for a = b = c = 5
615 (,(lambda (limit)
616 (let ((re (python-rx (group (+ (any word ?. ?_)))
617 (? ?\[ (+ (not (any ?\]))) ?\]) (* space)
618 assignment-operator))
619 (res nil))
620 (while (and (setq res (re-search-forward re limit t))
621 (or (python-syntax-context 'paren)
622 (equal (char-after (point)) ?=))))
623 res))
624 (1 font-lock-variable-name-face nil nil))
625 ;; support for a, b, c = (1, 2, 3)
626 (,(lambda (limit)
627 (let ((re (python-rx (group (+ (any word ?. ?_))) (* space)
628 (* ?, (* space) (+ (any word ?. ?_)) (* space))
629 ?, (* space) (+ (any word ?. ?_)) (* space)
630 assignment-operator))
631 (res nil))
632 (while (and (setq res (re-search-forward re limit t))
633 (goto-char (match-end 1))
634 (python-syntax-context 'paren)))
635 res))
636 (1 font-lock-variable-name-face nil nil))))
638 (defconst python-syntax-propertize-function
639 (syntax-propertize-rules
640 ((python-rx string-delimiter)
641 (0 (ignore (python-syntax-stringify))))))
643 (defconst python--prettify-symbols-alist
644 '(("lambda" . ?λ)
645 ("and" . ?∧)
646 ("or" . ?∨)))
648 (defsubst python-syntax-count-quotes (quote-char &optional point limit)
649 "Count number of quotes around point (max is 3).
650 QUOTE-CHAR is the quote char to count. Optional argument POINT is
651 the point where scan starts (defaults to current point), and LIMIT
652 is used to limit the scan."
653 (let ((i 0))
654 (while (and (< i 3)
655 (or (not limit) (< (+ point i) limit))
656 (eq (char-after (+ point i)) quote-char))
657 (setq i (1+ i)))
660 (defun python-syntax-stringify ()
661 "Put `syntax-table' property correctly on single/triple quotes."
662 (let* ((num-quotes (length (match-string-no-properties 1)))
663 (ppss (prog2
664 (backward-char num-quotes)
665 (syntax-ppss)
666 (forward-char num-quotes)))
667 (string-start (and (not (nth 4 ppss)) (nth 8 ppss)))
668 (quote-starting-pos (- (point) num-quotes))
669 (quote-ending-pos (point))
670 (num-closing-quotes
671 (and string-start
672 (python-syntax-count-quotes
673 (char-before) string-start quote-starting-pos))))
674 (cond ((and string-start (= num-closing-quotes 0))
675 ;; This set of quotes doesn't match the string starting
676 ;; kind. Do nothing.
677 nil)
678 ((not string-start)
679 ;; This set of quotes delimit the start of a string.
680 (put-text-property quote-starting-pos (1+ quote-starting-pos)
681 'syntax-table (string-to-syntax "|")))
682 ((= num-quotes num-closing-quotes)
683 ;; This set of quotes delimit the end of a string.
684 (put-text-property (1- quote-ending-pos) quote-ending-pos
685 'syntax-table (string-to-syntax "|")))
686 ((> num-quotes num-closing-quotes)
687 ;; This may only happen whenever a triple quote is closing
688 ;; a single quoted string. Add string delimiter syntax to
689 ;; all three quotes.
690 (put-text-property quote-starting-pos quote-ending-pos
691 'syntax-table (string-to-syntax "|"))))))
693 (defvar python-mode-syntax-table
694 (let ((table (make-syntax-table)))
695 ;; Give punctuation syntax to ASCII that normally has symbol
696 ;; syntax or has word syntax and isn't a letter.
697 (let ((symbol (string-to-syntax "_"))
698 (sst (standard-syntax-table)))
699 (dotimes (i 128)
700 (unless (= i ?_)
701 (if (equal symbol (aref sst i))
702 (modify-syntax-entry i "." table)))))
703 (modify-syntax-entry ?$ "." table)
704 (modify-syntax-entry ?% "." table)
705 ;; exceptions
706 (modify-syntax-entry ?# "<" table)
707 (modify-syntax-entry ?\n ">" table)
708 (modify-syntax-entry ?' "\"" table)
709 (modify-syntax-entry ?` "$" table)
710 table)
711 "Syntax table for Python files.")
713 (defvar python-dotty-syntax-table
714 (let ((table (make-syntax-table python-mode-syntax-table)))
715 (modify-syntax-entry ?. "w" table)
716 (modify-syntax-entry ?_ "w" table)
717 table)
718 "Dotty syntax table for Python files.
719 It makes underscores and dots word constituent chars.")
722 ;;; Indentation
724 (defcustom python-indent-offset 4
725 "Default indentation offset for Python."
726 :group 'python
727 :type 'integer
728 :safe 'integerp)
730 (defcustom python-indent-guess-indent-offset t
731 "Non-nil tells Python mode to guess `python-indent-offset' value."
732 :type 'boolean
733 :group 'python
734 :safe 'booleanp)
736 (defcustom python-indent-guess-indent-offset-verbose t
737 "Non-nil means to emit a warning when indentation guessing fails."
738 :version "25.1"
739 :type 'boolean
740 :group 'python
741 :safe' booleanp)
743 (defcustom python-indent-trigger-commands
744 '(indent-for-tab-command yas-expand yas/expand)
745 "Commands that might trigger a `python-indent-line' call."
746 :type '(repeat symbol)
747 :group 'python)
749 (define-obsolete-variable-alias
750 'python-indent 'python-indent-offset "24.3")
752 (define-obsolete-variable-alias
753 'python-guess-indent 'python-indent-guess-indent-offset "24.3")
755 (defvar python-indent-current-level 0
756 "Deprecated var available for compatibility.")
758 (defvar python-indent-levels '(0)
759 "Deprecated var available for compatibility.")
761 (make-obsolete-variable
762 'python-indent-current-level
763 "The indentation API changed to avoid global state.
764 The function `python-indent-calculate-levels' does not use it
765 anymore. If you were defadvising it and or depended on this
766 variable for indentation customizations, refactor your code to
767 work on `python-indent-calculate-indentation' instead."
768 "24.5")
770 (make-obsolete-variable
771 'python-indent-levels
772 "The indentation API changed to avoid global state.
773 The function `python-indent-calculate-levels' does not use it
774 anymore. If you were defadvising it and or depended on this
775 variable for indentation customizations, refactor your code to
776 work on `python-indent-calculate-indentation' instead."
777 "24.5")
779 (defun python-indent-guess-indent-offset ()
780 "Guess and set `python-indent-offset' for the current buffer."
781 (interactive)
782 (save-excursion
783 (save-restriction
784 (prog-widen)
785 (goto-char (point-min))
786 (let ((block-end))
787 (while (and (not block-end)
788 (re-search-forward
789 (python-rx line-start block-start) nil t))
790 (when (and
791 (not (python-syntax-context-type))
792 (progn
793 (goto-char (line-end-position))
794 (python-util-forward-comment -1)
795 (if (equal (char-before) ?:)
797 (forward-line 1)
798 (when (python-info-block-continuation-line-p)
799 (while (and (python-info-continuation-line-p)
800 (not (eobp)))
801 (forward-line 1))
802 (python-util-forward-comment -1)
803 (when (equal (char-before) ?:)
804 t)))))
805 (setq block-end (point-marker))))
806 (let ((indentation
807 (when block-end
808 (goto-char block-end)
809 (python-util-forward-comment)
810 (current-indentation))))
811 (if (and indentation (not (zerop indentation)))
812 (set (make-local-variable 'python-indent-offset) indentation)
813 (when python-indent-guess-indent-offset-verbose
814 (message "Can't guess python-indent-offset, using defaults: %s"
815 python-indent-offset))))))))
817 (defun python-indent-context ()
818 "Get information about the current indentation context.
819 Context is returned in a cons with the form (STATUS . START).
821 STATUS can be one of the following:
823 keyword
824 -------
826 :after-comment
827 - Point is after a comment line.
828 - START is the position of the \"#\" character.
829 :inside-string
830 - Point is inside string.
831 - START is the position of the first quote that starts it.
832 :no-indent
833 - No possible indentation case matches.
834 - START is always zero.
836 :inside-paren
837 - Fallback case when point is inside paren.
838 - START is the first non space char position *after* the open paren.
839 :inside-paren-at-closing-nested-paren
840 - Point is on a line that contains a nested paren closer.
841 - START is the position of the open paren it closes.
842 :inside-paren-at-closing-paren
843 - Point is on a line that contains a paren closer.
844 - START is the position of the open paren.
845 :inside-paren-newline-start
846 - Point is inside a paren with items starting in their own line.
847 - START is the position of the open paren.
848 :inside-paren-newline-start-from-block
849 - Point is inside a paren with items starting in their own line
850 from a block start.
851 - START is the position of the open paren.
853 :after-backslash
854 - Fallback case when point is after backslash.
855 - START is the char after the position of the backslash.
856 :after-backslash-assignment-continuation
857 - Point is after a backslashed assignment.
858 - START is the char after the position of the backslash.
859 :after-backslash-block-continuation
860 - Point is after a backslashed block continuation.
861 - START is the char after the position of the backslash.
862 :after-backslash-dotted-continuation
863 - Point is after a backslashed dotted continuation. Previous
864 line must contain a dot to align with.
865 - START is the char after the position of the backslash.
866 :after-backslash-first-line
867 - First line following a backslashed continuation.
868 - START is the char after the position of the backslash.
870 :after-block-end
871 - Point is after a line containing a block ender.
872 - START is the position where the ender starts.
873 :after-block-start
874 - Point is after a line starting a block.
875 - START is the position where the block starts.
876 :after-line
877 - Point is after a simple line.
878 - START is the position where the previous line starts.
879 :at-dedenter-block-start
880 - Point is on a line starting a dedenter block.
881 - START is the position where the dedenter block starts."
882 (save-restriction
883 (prog-widen)
884 (let ((ppss (save-excursion
885 (beginning-of-line)
886 (syntax-ppss))))
887 (cond
888 ;; Beginning of buffer.
889 ((= (line-number-at-pos) 1)
890 (cons :no-indent 0))
891 ;; Inside a string.
892 ((let ((start (python-syntax-context 'string ppss)))
893 (when start
894 (cons (if (python-info-docstring-p)
895 :inside-docstring
896 :inside-string) start))))
897 ;; Inside a paren.
898 ((let* ((start (python-syntax-context 'paren ppss))
899 (starts-in-newline
900 (when start
901 (save-excursion
902 (goto-char start)
903 (forward-char)
904 (not
905 (= (line-number-at-pos)
906 (progn
907 (python-util-forward-comment)
908 (line-number-at-pos))))))))
909 (when start
910 (cond
911 ;; Current line only holds the closing paren.
912 ((save-excursion
913 (skip-syntax-forward " ")
914 (when (and (python-syntax-closing-paren-p)
915 (progn
916 (forward-char 1)
917 (not (python-syntax-context 'paren))))
918 (cons :inside-paren-at-closing-paren start))))
919 ;; Current line only holds a closing paren for nested.
920 ((save-excursion
921 (back-to-indentation)
922 (python-syntax-closing-paren-p))
923 (cons :inside-paren-at-closing-nested-paren start))
924 ;; This line starts from a opening block in its own line.
925 ((save-excursion
926 (goto-char start)
927 (when (and
928 starts-in-newline
929 (save-excursion
930 (back-to-indentation)
931 (looking-at (python-rx block-start))))
932 (cons
933 :inside-paren-newline-start-from-block start))))
934 (starts-in-newline
935 (cons :inside-paren-newline-start start))
936 ;; General case.
937 (t (cons :inside-paren
938 (save-excursion
939 (goto-char (1+ start))
940 (skip-syntax-forward "(" 1)
941 (skip-syntax-forward " ")
942 (point))))))))
943 ;; After backslash.
944 ((let ((start (when (not (python-syntax-comment-or-string-p ppss))
945 (python-info-line-ends-backslash-p
946 (1- (line-number-at-pos))))))
947 (when start
948 (cond
949 ;; Continuation of dotted expression.
950 ((save-excursion
951 (back-to-indentation)
952 (when (eq (char-after) ?\.)
953 ;; Move point back until it's not inside a paren.
954 (while (prog2
955 (forward-line -1)
956 (and (not (bobp))
957 (python-syntax-context 'paren))))
958 (goto-char (line-end-position))
959 (while (and (search-backward
960 "." (line-beginning-position) t)
961 (python-syntax-context-type)))
962 ;; Ensure previous statement has dot to align with.
963 (when (and (eq (char-after) ?\.)
964 (not (python-syntax-context-type)))
965 (cons :after-backslash-dotted-continuation (point))))))
966 ;; Continuation of block definition.
967 ((let ((block-continuation-start
968 (python-info-block-continuation-line-p)))
969 (when block-continuation-start
970 (save-excursion
971 (goto-char block-continuation-start)
972 (re-search-forward
973 (python-rx block-start (* space))
974 (line-end-position) t)
975 (cons :after-backslash-block-continuation (point))))))
976 ;; Continuation of assignment.
977 ((let ((assignment-continuation-start
978 (python-info-assignment-continuation-line-p)))
979 (when assignment-continuation-start
980 (save-excursion
981 (goto-char assignment-continuation-start)
982 (cons :after-backslash-assignment-continuation (point))))))
983 ;; First line after backslash continuation start.
984 ((save-excursion
985 (goto-char start)
986 (when (or (= (line-number-at-pos) 1)
987 (not (python-info-beginning-of-backslash
988 (1- (line-number-at-pos)))))
989 (cons :after-backslash-first-line start))))
990 ;; General case.
991 (t (cons :after-backslash start))))))
992 ;; After beginning of block.
993 ((let ((start (save-excursion
994 (back-to-indentation)
995 (python-util-forward-comment -1)
996 (when (equal (char-before) ?:)
997 (python-nav-beginning-of-block)))))
998 (when start
999 (cons :after-block-start start))))
1000 ;; At dedenter statement.
1001 ((let ((start (python-info-dedenter-statement-p)))
1002 (when start
1003 (cons :at-dedenter-block-start start))))
1004 ;; After normal line, comment or ender (default case).
1005 ((save-excursion
1006 (back-to-indentation)
1007 (skip-chars-backward " \t\n")
1008 (if (bobp)
1009 (cons :no-indent 0)
1010 (python-nav-beginning-of-statement)
1011 (cons
1012 (cond ((python-info-current-line-comment-p)
1013 :after-comment)
1014 ((save-excursion
1015 (goto-char (line-end-position))
1016 (python-util-forward-comment -1)
1017 (python-nav-beginning-of-statement)
1018 (looking-at (python-rx block-ender)))
1019 :after-block-end)
1020 (t :after-line))
1021 (point)))))))))
1023 (defun python-indent--calculate-indentation ()
1024 "Internal implementation of `python-indent-calculate-indentation'.
1025 May return an integer for the maximum possible indentation at
1026 current context or a list of integers. The latter case is only
1027 happening for :at-dedenter-block-start context since the
1028 possibilities can be narrowed to specific indentation points."
1029 (save-restriction
1030 (prog-widen)
1031 (save-excursion
1032 (pcase (python-indent-context)
1033 (`(:no-indent . ,_) (prog-first-column)) ; usually 0
1034 (`(,(or :after-line
1035 :after-comment
1036 :inside-string
1037 :after-backslash
1038 :inside-paren-at-closing-paren
1039 :inside-paren-at-closing-nested-paren) . ,start)
1040 ;; Copy previous indentation.
1041 (goto-char start)
1042 (current-indentation))
1043 (`(:inside-docstring . ,start)
1044 (let* ((line-indentation (current-indentation))
1045 (base-indent (progn
1046 (goto-char start)
1047 (current-indentation))))
1048 (max line-indentation base-indent)))
1049 (`(,(or :after-block-start
1050 :after-backslash-first-line
1051 :inside-paren-newline-start) . ,start)
1052 ;; Add one indentation level.
1053 (goto-char start)
1054 (+ (current-indentation) python-indent-offset))
1055 (`(,(or :inside-paren
1056 :after-backslash-block-continuation
1057 :after-backslash-assignment-continuation
1058 :after-backslash-dotted-continuation) . ,start)
1059 ;; Use the column given by the context.
1060 (goto-char start)
1061 (current-column))
1062 (`(:after-block-end . ,start)
1063 ;; Subtract one indentation level.
1064 (goto-char start)
1065 (- (current-indentation) python-indent-offset))
1066 (`(:at-dedenter-block-start . ,_)
1067 ;; List all possible indentation levels from opening blocks.
1068 (let ((opening-block-start-points
1069 (python-info-dedenter-opening-block-positions)))
1070 (if (not opening-block-start-points)
1071 (prog-first-column) ; if not found default to first column
1072 (mapcar (lambda (pos)
1073 (save-excursion
1074 (goto-char pos)
1075 (current-indentation)))
1076 opening-block-start-points))))
1077 (`(,(or :inside-paren-newline-start-from-block) . ,start)
1078 ;; Add two indentation levels to make the suite stand out.
1079 (goto-char start)
1080 (+ (current-indentation) (* python-indent-offset 2)))))))
1082 (defun python-indent--calculate-levels (indentation)
1083 "Calculate levels list given INDENTATION.
1084 Argument INDENTATION can either be an integer or a list of
1085 integers. Levels are returned in ascending order, and in the
1086 case INDENTATION is a list, this order is enforced."
1087 (if (listp indentation)
1088 (sort (copy-sequence indentation) #'<)
1089 (nconc (number-sequence (prog-first-column) (1- indentation)
1090 python-indent-offset)
1091 (list indentation))))
1093 (defun python-indent--previous-level (levels indentation)
1094 "Return previous level from LEVELS relative to INDENTATION."
1095 (let* ((levels (sort (copy-sequence levels) #'>))
1096 (default (car levels)))
1097 (catch 'return
1098 (dolist (level levels)
1099 (when (funcall #'< level indentation)
1100 (throw 'return level)))
1101 default)))
1103 (defun python-indent-calculate-indentation (&optional previous)
1104 "Calculate indentation.
1105 Get indentation of PREVIOUS level when argument is non-nil.
1106 Return the max level of the cycle when indentation reaches the
1107 minimum."
1108 (let* ((indentation (python-indent--calculate-indentation))
1109 (levels (python-indent--calculate-levels indentation)))
1110 (if previous
1111 (python-indent--previous-level levels (current-indentation))
1112 (if levels
1113 (apply #'max levels)
1114 (prog-first-column)))))
1116 (defun python-indent-line (&optional previous)
1117 "Internal implementation of `python-indent-line-function'.
1118 Use the PREVIOUS level when argument is non-nil, otherwise indent
1119 to the maximum available level. When indentation is the minimum
1120 possible and PREVIOUS is non-nil, cycle back to the maximum
1121 level."
1122 (let ((follow-indentation-p
1123 ;; Check if point is within indentation.
1124 (and (<= (line-beginning-position) (point))
1125 (>= (+ (line-beginning-position)
1126 (current-indentation))
1127 (point)))))
1128 (save-excursion
1129 (indent-line-to
1130 (python-indent-calculate-indentation previous))
1131 (python-info-dedenter-opening-block-message))
1132 (when follow-indentation-p
1133 (back-to-indentation))))
1135 (defun python-indent-calculate-levels ()
1136 "Return possible indentation levels."
1137 (python-indent--calculate-levels
1138 (python-indent--calculate-indentation)))
1140 (defun python-indent-line-function ()
1141 "`indent-line-function' for Python mode.
1142 When the variable `last-command' is equal to one of the symbols
1143 inside `python-indent-trigger-commands' it cycles possible
1144 indentation levels from right to left."
1145 (python-indent-line
1146 (and (memq this-command python-indent-trigger-commands)
1147 (eq last-command this-command))))
1149 (defun python-indent-dedent-line ()
1150 "De-indent current line."
1151 (interactive "*")
1152 (when (and (not (bolp))
1153 (not (python-syntax-comment-or-string-p))
1154 (= (current-indentation) (current-column)))
1155 (python-indent-line t)
1158 (defun python-indent-dedent-line-backspace (arg)
1159 "De-indent current line.
1160 Argument ARG is passed to `backward-delete-char-untabify' when
1161 point is not in between the indentation."
1162 (interactive "*p")
1163 (unless (python-indent-dedent-line)
1164 (backward-delete-char-untabify arg)))
1166 (put 'python-indent-dedent-line-backspace 'delete-selection 'supersede)
1168 (defun python-indent-region (start end)
1169 "Indent a Python region automagically.
1171 Called from a program, START and END specify the region to indent."
1172 (let ((deactivate-mark nil))
1173 (save-excursion
1174 (goto-char end)
1175 (setq end (point-marker))
1176 (goto-char start)
1177 (or (bolp) (forward-line 1))
1178 (while (< (point) end)
1179 (or (and (bolp) (eolp))
1180 (when (and
1181 ;; Skip if previous line is empty or a comment.
1182 (save-excursion
1183 (let ((line-is-comment-p
1184 (python-info-current-line-comment-p)))
1185 (forward-line -1)
1186 (not
1187 (or (and (python-info-current-line-comment-p)
1188 ;; Unless this line is a comment too.
1189 (not line-is-comment-p))
1190 (python-info-current-line-empty-p)))))
1191 ;; Don't mess with strings, unless it's the
1192 ;; enclosing set of quotes or a docstring.
1193 (or (not (python-syntax-context 'string))
1195 (syntax-after
1196 (+ (1- (point))
1197 (current-indentation)
1198 (python-syntax-count-quotes (char-after) (point))))
1199 (string-to-syntax "|"))
1200 (python-info-docstring-p))
1201 ;; Skip if current line is a block start, a
1202 ;; dedenter or block ender.
1203 (save-excursion
1204 (back-to-indentation)
1205 (not (looking-at
1206 (python-rx
1207 (or block-start dedenter block-ender))))))
1208 (python-indent-line)))
1209 (forward-line 1))
1210 (move-marker end nil))))
1212 (defun python-indent-shift-left (start end &optional count)
1213 "Shift lines contained in region START END by COUNT columns to the left.
1214 COUNT defaults to `python-indent-offset'. If region isn't
1215 active, the current line is shifted. The shifted region includes
1216 the lines in which START and END lie. An error is signaled if
1217 any lines in the region are indented less than COUNT columns."
1218 (interactive
1219 (if mark-active
1220 (list (region-beginning) (region-end) current-prefix-arg)
1221 (list (line-beginning-position) (line-end-position) current-prefix-arg)))
1222 (if count
1223 (setq count (prefix-numeric-value count))
1224 (setq count python-indent-offset))
1225 (when (> count 0)
1226 (let ((deactivate-mark nil))
1227 (save-excursion
1228 (goto-char start)
1229 (while (< (point) end)
1230 (if (and (< (current-indentation) count)
1231 (not (looking-at "[ \t]*$")))
1232 (user-error "Can't shift all lines enough"))
1233 (forward-line))
1234 (indent-rigidly start end (- count))))))
1236 (defun python-indent-shift-right (start end &optional count)
1237 "Shift lines contained in region START END by COUNT columns to the right.
1238 COUNT defaults to `python-indent-offset'. If region isn't
1239 active, the current line is shifted. The shifted region includes
1240 the lines in which START and END lie."
1241 (interactive
1242 (if mark-active
1243 (list (region-beginning) (region-end) current-prefix-arg)
1244 (list (line-beginning-position) (line-end-position) current-prefix-arg)))
1245 (let ((deactivate-mark nil))
1246 (setq count (if count (prefix-numeric-value count)
1247 python-indent-offset))
1248 (indent-rigidly start end count)))
1250 (defun python-indent-post-self-insert-function ()
1251 "Adjust indentation after insertion of some characters.
1252 This function is intended to be added to `post-self-insert-hook.'
1253 If a line renders a paren alone, after adding a char before it,
1254 the line will be re-indented automatically if needed."
1255 (when (and electric-indent-mode
1256 (eq (char-before) last-command-event))
1257 (cond
1258 ;; Electric indent inside parens
1259 ((and
1260 (not (bolp))
1261 (let ((paren-start (python-syntax-context 'paren)))
1262 ;; Check that point is inside parens.
1263 (when paren-start
1264 (not
1265 ;; Filter the case where input is happening in the same
1266 ;; line where the open paren is.
1267 (= (line-number-at-pos)
1268 (line-number-at-pos paren-start)))))
1269 ;; When content has been added before the closing paren or a
1270 ;; comma has been inserted, it's ok to do the trick.
1272 (memq (char-after) '(?\) ?\] ?\}))
1273 (eq (char-before) ?,)))
1274 (save-excursion
1275 (goto-char (line-beginning-position))
1276 (let ((indentation (python-indent-calculate-indentation)))
1277 (when (and (numberp indentation) (< (current-indentation) indentation))
1278 (indent-line-to indentation)))))
1279 ;; Electric colon
1280 ((and (eq ?: last-command-event)
1281 (memq ?: electric-indent-chars)
1282 (not current-prefix-arg)
1283 ;; Trigger electric colon only at end of line
1284 (eolp)
1285 ;; Avoid re-indenting on extra colon
1286 (not (equal ?: (char-before (1- (point)))))
1287 (not (python-syntax-comment-or-string-p)))
1288 ;; Just re-indent dedenters
1289 (let ((dedenter-pos (python-info-dedenter-statement-p))
1290 (current-pos (point)))
1291 (when dedenter-pos
1292 (save-excursion
1293 (goto-char dedenter-pos)
1294 (python-indent-line)
1295 (unless (= (line-number-at-pos dedenter-pos)
1296 (line-number-at-pos current-pos))
1297 ;; Reindent region if this is a multiline statement
1298 (python-indent-region dedenter-pos current-pos)))))))))
1301 ;;; Mark
1303 (defun python-mark-defun (&optional allow-extend)
1304 "Put mark at end of this defun, point at beginning.
1305 The defun marked is the one that contains point or follows point.
1307 Interactively (or with ALLOW-EXTEND non-nil), if this command is
1308 repeated or (in Transient Mark mode) if the mark is active, it
1309 marks the next defun after the ones already marked."
1310 (interactive "p")
1311 (when (python-info-looking-at-beginning-of-defun)
1312 (end-of-line 1))
1313 (mark-defun allow-extend))
1316 ;;; Navigation
1318 (defvar python-nav-beginning-of-defun-regexp
1319 (python-rx line-start (* space) defun (+ space) (group symbol-name))
1320 "Regexp matching class or function definition.
1321 The name of the defun should be grouped so it can be retrieved
1322 via `match-string'.")
1324 (defun python-nav--beginning-of-defun (&optional arg)
1325 "Internal implementation of `python-nav-beginning-of-defun'.
1326 With positive ARG search backwards, else search forwards."
1327 (when (or (null arg) (= arg 0)) (setq arg 1))
1328 (let* ((re-search-fn (if (> arg 0)
1329 #'re-search-backward
1330 #'re-search-forward))
1331 (line-beg-pos (line-beginning-position))
1332 (line-content-start (+ line-beg-pos (current-indentation)))
1333 (pos (point-marker))
1334 (beg-indentation
1335 (and (> arg 0)
1336 (save-excursion
1337 (while (and
1338 (not (python-info-looking-at-beginning-of-defun))
1339 (python-nav-backward-block)))
1340 (or (and (python-info-looking-at-beginning-of-defun)
1341 (+ (current-indentation) python-indent-offset))
1342 0))))
1343 (found
1344 (progn
1345 (when (and (< arg 0)
1346 (python-info-looking-at-beginning-of-defun))
1347 (end-of-line 1))
1348 (while (and (funcall re-search-fn
1349 python-nav-beginning-of-defun-regexp nil t)
1350 (or (python-syntax-context-type)
1351 ;; Handle nested defuns when moving
1352 ;; backwards by checking indentation.
1353 (and (> arg 0)
1354 (not (= (current-indentation) 0))
1355 (>= (current-indentation) beg-indentation)))))
1356 (and (python-info-looking-at-beginning-of-defun)
1357 (or (not (= (line-number-at-pos pos)
1358 (line-number-at-pos)))
1359 (and (>= (point) line-beg-pos)
1360 (<= (point) line-content-start)
1361 (> pos line-content-start)))))))
1362 (if found
1363 (or (beginning-of-line 1) t)
1364 (and (goto-char pos) nil))))
1366 (defun python-nav-beginning-of-defun (&optional arg)
1367 "Move point to `beginning-of-defun'.
1368 With positive ARG search backwards else search forward.
1369 ARG nil or 0 defaults to 1. When searching backwards,
1370 nested defuns are handled with care depending on current
1371 point position. Return non-nil if point is moved to
1372 `beginning-of-defun'."
1373 (when (or (null arg) (= arg 0)) (setq arg 1))
1374 (let ((found))
1375 (while (and (not (= arg 0))
1376 (let ((keep-searching-p
1377 (python-nav--beginning-of-defun arg)))
1378 (when (and keep-searching-p (null found))
1379 (setq found t))
1380 keep-searching-p))
1381 (setq arg (if (> arg 0) (1- arg) (1+ arg))))
1382 found))
1384 (defun python-nav-end-of-defun ()
1385 "Move point to the end of def or class.
1386 Returns nil if point is not in a def or class."
1387 (interactive)
1388 (let ((beg-defun-indent)
1389 (beg-pos (point)))
1390 (when (or (python-info-looking-at-beginning-of-defun)
1391 (python-nav-beginning-of-defun 1)
1392 (python-nav-beginning-of-defun -1))
1393 (setq beg-defun-indent (current-indentation))
1394 (while (progn
1395 (python-nav-end-of-statement)
1396 (python-util-forward-comment 1)
1397 (and (> (current-indentation) beg-defun-indent)
1398 (not (eobp)))))
1399 (python-util-forward-comment -1)
1400 (forward-line 1)
1401 ;; Ensure point moves forward.
1402 (and (> beg-pos (point)) (goto-char beg-pos)))))
1404 (defun python-nav--syntactically (fn poscompfn &optional contextfn)
1405 "Move point using FN avoiding places with specific context.
1406 FN must take no arguments. POSCOMPFN is a two arguments function
1407 used to compare current and previous point after it is moved
1408 using FN, this is normally a less-than or greater-than
1409 comparison. Optional argument CONTEXTFN defaults to
1410 `python-syntax-context-type' and is used for checking current
1411 point context, it must return a non-nil value if this point must
1412 be skipped."
1413 (let ((contextfn (or contextfn 'python-syntax-context-type))
1414 (start-pos (point-marker))
1415 (prev-pos))
1416 (catch 'found
1417 (while t
1418 (let* ((newpos
1419 (and (funcall fn) (point-marker)))
1420 (context (funcall contextfn)))
1421 (cond ((and (not context) newpos
1422 (or (and (not prev-pos) newpos)
1423 (and prev-pos newpos
1424 (funcall poscompfn newpos prev-pos))))
1425 (throw 'found (point-marker)))
1426 ((and newpos context)
1427 (setq prev-pos (point)))
1428 (t (when (not newpos) (goto-char start-pos))
1429 (throw 'found nil))))))))
1431 (defun python-nav--forward-defun (arg)
1432 "Internal implementation of python-nav-{backward,forward}-defun.
1433 Uses ARG to define which function to call, and how many times
1434 repeat it."
1435 (let ((found))
1436 (while (and (> arg 0)
1437 (setq found
1438 (python-nav--syntactically
1439 (lambda ()
1440 (re-search-forward
1441 python-nav-beginning-of-defun-regexp nil t))
1442 '>)))
1443 (setq arg (1- arg)))
1444 (while (and (< arg 0)
1445 (setq found
1446 (python-nav--syntactically
1447 (lambda ()
1448 (re-search-backward
1449 python-nav-beginning-of-defun-regexp nil t))
1450 '<)))
1451 (setq arg (1+ arg)))
1452 found))
1454 (defun python-nav-backward-defun (&optional arg)
1455 "Navigate to closer defun backward ARG times.
1456 Unlikely `python-nav-beginning-of-defun' this doesn't care about
1457 nested definitions."
1458 (interactive "^p")
1459 (python-nav--forward-defun (- (or arg 1))))
1461 (defun python-nav-forward-defun (&optional arg)
1462 "Navigate to closer defun forward ARG times.
1463 Unlikely `python-nav-beginning-of-defun' this doesn't care about
1464 nested definitions."
1465 (interactive "^p")
1466 (python-nav--forward-defun (or arg 1)))
1468 (defun python-nav-beginning-of-statement ()
1469 "Move to start of current statement."
1470 (interactive "^")
1471 (back-to-indentation)
1472 (let* ((ppss (syntax-ppss))
1473 (context-point
1475 (python-syntax-context 'paren ppss)
1476 (python-syntax-context 'string ppss))))
1477 (cond ((bobp))
1478 (context-point
1479 (goto-char context-point)
1480 (python-nav-beginning-of-statement))
1481 ((save-excursion
1482 (forward-line -1)
1483 (python-info-line-ends-backslash-p))
1484 (forward-line -1)
1485 (python-nav-beginning-of-statement))))
1486 (point-marker))
1488 (defun python-nav-end-of-statement (&optional noend)
1489 "Move to end of current statement.
1490 Optional argument NOEND is internal and makes the logic to not
1491 jump to the end of line when moving forward searching for the end
1492 of the statement."
1493 (interactive "^")
1494 (let (string-start bs-pos)
1495 (while (and (or noend (goto-char (line-end-position)))
1496 (not (eobp))
1497 (cond ((setq string-start (python-syntax-context 'string))
1498 (goto-char string-start)
1499 (if (python-syntax-context 'paren)
1500 ;; Ended up inside a paren, roll again.
1501 (python-nav-end-of-statement t)
1502 ;; This is not inside a paren, move to the
1503 ;; end of this string.
1504 (goto-char (+ (point)
1505 (python-syntax-count-quotes
1506 (char-after (point)) (point))))
1507 (or (re-search-forward (rx (syntax string-delimiter)) nil t)
1508 (goto-char (point-max)))))
1509 ((python-syntax-context 'paren)
1510 ;; The statement won't end before we've escaped
1511 ;; at least one level of parenthesis.
1512 (condition-case err
1513 (goto-char (scan-lists (point) 1 -1))
1514 (scan-error (goto-char (nth 3 err)))))
1515 ((setq bs-pos (python-info-line-ends-backslash-p))
1516 (goto-char bs-pos)
1517 (forward-line 1))))))
1518 (point-marker))
1520 (defun python-nav-backward-statement (&optional arg)
1521 "Move backward to previous statement.
1522 With ARG, repeat. See `python-nav-forward-statement'."
1523 (interactive "^p")
1524 (or arg (setq arg 1))
1525 (python-nav-forward-statement (- arg)))
1527 (defun python-nav-forward-statement (&optional arg)
1528 "Move forward to next statement.
1529 With ARG, repeat. With negative argument, move ARG times
1530 backward to previous statement."
1531 (interactive "^p")
1532 (or arg (setq arg 1))
1533 (while (> arg 0)
1534 (python-nav-end-of-statement)
1535 (python-util-forward-comment)
1536 (python-nav-beginning-of-statement)
1537 (setq arg (1- arg)))
1538 (while (< arg 0)
1539 (python-nav-beginning-of-statement)
1540 (python-util-forward-comment -1)
1541 (python-nav-beginning-of-statement)
1542 (setq arg (1+ arg))))
1544 (defun python-nav-beginning-of-block ()
1545 "Move to start of current block."
1546 (interactive "^")
1547 (let ((starting-pos (point)))
1548 (if (progn
1549 (python-nav-beginning-of-statement)
1550 (looking-at (python-rx block-start)))
1551 (point-marker)
1552 ;; Go to first line beginning a statement
1553 (while (and (not (bobp))
1554 (or (and (python-nav-beginning-of-statement) nil)
1555 (python-info-current-line-comment-p)
1556 (python-info-current-line-empty-p)))
1557 (forward-line -1))
1558 (let ((block-matching-indent
1559 (- (current-indentation) python-indent-offset)))
1560 (while
1561 (and (python-nav-backward-block)
1562 (> (current-indentation) block-matching-indent)))
1563 (if (and (looking-at (python-rx block-start))
1564 (= (current-indentation) block-matching-indent))
1565 (point-marker)
1566 (and (goto-char starting-pos) nil))))))
1568 (defun python-nav-end-of-block ()
1569 "Move to end of current block."
1570 (interactive "^")
1571 (when (python-nav-beginning-of-block)
1572 (let ((block-indentation (current-indentation)))
1573 (python-nav-end-of-statement)
1574 (while (and (forward-line 1)
1575 (not (eobp))
1576 (or (and (> (current-indentation) block-indentation)
1577 (or (python-nav-end-of-statement) t))
1578 (python-info-current-line-comment-p)
1579 (python-info-current-line-empty-p))))
1580 (python-util-forward-comment -1)
1581 (point-marker))))
1583 (defun python-nav-backward-block (&optional arg)
1584 "Move backward to previous block of code.
1585 With ARG, repeat. See `python-nav-forward-block'."
1586 (interactive "^p")
1587 (or arg (setq arg 1))
1588 (python-nav-forward-block (- arg)))
1590 (defun python-nav-forward-block (&optional arg)
1591 "Move forward to next block of code.
1592 With ARG, repeat. With negative argument, move ARG times
1593 backward to previous block."
1594 (interactive "^p")
1595 (or arg (setq arg 1))
1596 (let ((block-start-regexp
1597 (python-rx line-start (* whitespace) block-start))
1598 (starting-pos (point)))
1599 (while (> arg 0)
1600 (python-nav-end-of-statement)
1601 (while (and
1602 (re-search-forward block-start-regexp nil t)
1603 (python-syntax-context-type)))
1604 (setq arg (1- arg)))
1605 (while (< arg 0)
1606 (python-nav-beginning-of-statement)
1607 (while (and
1608 (re-search-backward block-start-regexp nil t)
1609 (python-syntax-context-type)))
1610 (setq arg (1+ arg)))
1611 (python-nav-beginning-of-statement)
1612 (if (not (looking-at (python-rx block-start)))
1613 (and (goto-char starting-pos) nil)
1614 (and (not (= (point) starting-pos)) (point-marker)))))
1616 (defun python-nav--lisp-forward-sexp (&optional arg)
1617 "Standard version `forward-sexp'.
1618 It ignores completely the value of `forward-sexp-function' by
1619 setting it to nil before calling `forward-sexp'. With positive
1620 ARG move forward only one sexp, else move backwards."
1621 (let ((forward-sexp-function)
1622 (arg (if (or (not arg) (> arg 0)) 1 -1)))
1623 (forward-sexp arg)))
1625 (defun python-nav--lisp-forward-sexp-safe (&optional arg)
1626 "Safe version of standard `forward-sexp'.
1627 When at end of sexp (i.e. looking at a opening/closing paren)
1628 skips it instead of throwing an error. With positive ARG move
1629 forward only one sexp, else move backwards."
1630 (let* ((arg (if (or (not arg) (> arg 0)) 1 -1))
1631 (paren-regexp
1632 (if (> arg 0) (python-rx close-paren) (python-rx open-paren)))
1633 (search-fn
1634 (if (> arg 0) #'re-search-forward #'re-search-backward)))
1635 (condition-case nil
1636 (python-nav--lisp-forward-sexp arg)
1637 (error
1638 (while (and (funcall search-fn paren-regexp nil t)
1639 (python-syntax-context 'paren)))))))
1641 (defun python-nav--forward-sexp (&optional dir safe skip-parens-p)
1642 "Move to forward sexp.
1643 With positive optional argument DIR direction move forward, else
1644 backwards. When optional argument SAFE is non-nil do not throw
1645 errors when at end of sexp, skip it instead. With optional
1646 argument SKIP-PARENS-P force sexp motion to ignore parenthesized
1647 expressions when looking at them in either direction."
1648 (setq dir (or dir 1))
1649 (unless (= dir 0)
1650 (let* ((forward-p (if (> dir 0)
1651 (and (setq dir 1) t)
1652 (and (setq dir -1) nil)))
1653 (context-type (python-syntax-context-type)))
1654 (cond
1655 ((memq context-type '(string comment))
1656 ;; Inside of a string, get out of it.
1657 (let ((forward-sexp-function))
1658 (forward-sexp dir)))
1659 ((and (not skip-parens-p)
1660 (or (eq context-type 'paren)
1661 (if forward-p
1662 (eq (syntax-class (syntax-after (point)))
1663 (car (string-to-syntax "(")))
1664 (eq (syntax-class (syntax-after (1- (point))))
1665 (car (string-to-syntax ")"))))))
1666 ;; Inside a paren or looking at it, lisp knows what to do.
1667 (if safe
1668 (python-nav--lisp-forward-sexp-safe dir)
1669 (python-nav--lisp-forward-sexp dir)))
1671 ;; This part handles the lispy feel of
1672 ;; `python-nav-forward-sexp'. Knowing everything about the
1673 ;; current context and the context of the next sexp tries to
1674 ;; follow the lisp sexp motion commands in a symmetric manner.
1675 (let* ((context
1676 (cond
1677 ((python-info-beginning-of-block-p) 'block-start)
1678 ((python-info-end-of-block-p) 'block-end)
1679 ((python-info-beginning-of-statement-p) 'statement-start)
1680 ((python-info-end-of-statement-p) 'statement-end)))
1681 (next-sexp-pos
1682 (save-excursion
1683 (if safe
1684 (python-nav--lisp-forward-sexp-safe dir)
1685 (python-nav--lisp-forward-sexp dir))
1686 (point)))
1687 (next-sexp-context
1688 (save-excursion
1689 (goto-char next-sexp-pos)
1690 (cond
1691 ((python-info-beginning-of-block-p) 'block-start)
1692 ((python-info-end-of-block-p) 'block-end)
1693 ((python-info-beginning-of-statement-p) 'statement-start)
1694 ((python-info-end-of-statement-p) 'statement-end)
1695 ((python-info-statement-starts-block-p) 'starts-block)
1696 ((python-info-statement-ends-block-p) 'ends-block)))))
1697 (if forward-p
1698 (cond ((and (not (eobp))
1699 (python-info-current-line-empty-p))
1700 (python-util-forward-comment dir)
1701 (python-nav--forward-sexp dir safe skip-parens-p))
1702 ((eq context 'block-start)
1703 (python-nav-end-of-block))
1704 ((eq context 'statement-start)
1705 (python-nav-end-of-statement))
1706 ((and (memq context '(statement-end block-end))
1707 (eq next-sexp-context 'ends-block))
1708 (goto-char next-sexp-pos)
1709 (python-nav-end-of-block))
1710 ((and (memq context '(statement-end block-end))
1711 (eq next-sexp-context 'starts-block))
1712 (goto-char next-sexp-pos)
1713 (python-nav-end-of-block))
1714 ((memq context '(statement-end block-end))
1715 (goto-char next-sexp-pos)
1716 (python-nav-end-of-statement))
1717 (t (goto-char next-sexp-pos)))
1718 (cond ((and (not (bobp))
1719 (python-info-current-line-empty-p))
1720 (python-util-forward-comment dir)
1721 (python-nav--forward-sexp dir safe skip-parens-p))
1722 ((eq context 'block-end)
1723 (python-nav-beginning-of-block))
1724 ((eq context 'statement-end)
1725 (python-nav-beginning-of-statement))
1726 ((and (memq context '(statement-start block-start))
1727 (eq next-sexp-context 'starts-block))
1728 (goto-char next-sexp-pos)
1729 (python-nav-beginning-of-block))
1730 ((and (memq context '(statement-start block-start))
1731 (eq next-sexp-context 'ends-block))
1732 (goto-char next-sexp-pos)
1733 (python-nav-beginning-of-block))
1734 ((memq context '(statement-start block-start))
1735 (goto-char next-sexp-pos)
1736 (python-nav-beginning-of-statement))
1737 (t (goto-char next-sexp-pos))))))))))
1739 (defun python-nav-forward-sexp (&optional arg safe skip-parens-p)
1740 "Move forward across expressions.
1741 With ARG, do it that many times. Negative arg -N means move
1742 backward N times. When optional argument SAFE is non-nil do not
1743 throw errors when at end of sexp, skip it instead. With optional
1744 argument SKIP-PARENS-P force sexp motion to ignore parenthesized
1745 expressions when looking at them in either direction (forced to t
1746 in interactive calls)."
1747 (interactive "^p")
1748 (or arg (setq arg 1))
1749 ;; Do not follow parens on interactive calls. This hack to detect
1750 ;; if the function was called interactively copes with the way
1751 ;; `forward-sexp' works by calling `forward-sexp-function', losing
1752 ;; interactive detection by checking `current-prefix-arg'. The
1753 ;; reason to make this distinction is that lisp functions like
1754 ;; `blink-matching-open' get confused causing issues like the one in
1755 ;; Bug#16191. With this approach the user gets a symmetric behavior
1756 ;; when working interactively while called functions expecting
1757 ;; paren-based sexp motion work just fine.
1759 skip-parens-p
1760 (setq skip-parens-p
1761 (memq real-this-command
1762 (list
1763 #'forward-sexp #'backward-sexp
1764 #'python-nav-forward-sexp #'python-nav-backward-sexp
1765 #'python-nav-forward-sexp-safe #'python-nav-backward-sexp))))
1766 (while (> arg 0)
1767 (python-nav--forward-sexp 1 safe skip-parens-p)
1768 (setq arg (1- arg)))
1769 (while (< arg 0)
1770 (python-nav--forward-sexp -1 safe skip-parens-p)
1771 (setq arg (1+ arg))))
1773 (defun python-nav-backward-sexp (&optional arg safe skip-parens-p)
1774 "Move backward across expressions.
1775 With ARG, do it that many times. Negative arg -N means move
1776 forward N times. When optional argument SAFE is non-nil do not
1777 throw errors when at end of sexp, skip it instead. With optional
1778 argument SKIP-PARENS-P force sexp motion to ignore parenthesized
1779 expressions when looking at them in either direction (forced to t
1780 in interactive calls)."
1781 (interactive "^p")
1782 (or arg (setq arg 1))
1783 (python-nav-forward-sexp (- arg) safe skip-parens-p))
1785 (defun python-nav-forward-sexp-safe (&optional arg skip-parens-p)
1786 "Move forward safely across expressions.
1787 With ARG, do it that many times. Negative arg -N means move
1788 backward N times. With optional argument SKIP-PARENS-P force
1789 sexp motion to ignore parenthesized expressions when looking at
1790 them in either direction (forced to t in interactive calls)."
1791 (interactive "^p")
1792 (python-nav-forward-sexp arg t skip-parens-p))
1794 (defun python-nav-backward-sexp-safe (&optional arg skip-parens-p)
1795 "Move backward safely across expressions.
1796 With ARG, do it that many times. Negative arg -N means move
1797 forward N times. With optional argument SKIP-PARENS-P force sexp
1798 motion to ignore parenthesized expressions when looking at them in
1799 either direction (forced to t in interactive calls)."
1800 (interactive "^p")
1801 (python-nav-backward-sexp arg t skip-parens-p))
1803 (defun python-nav--up-list (&optional dir)
1804 "Internal implementation of `python-nav-up-list'.
1805 DIR is always 1 or -1 and comes sanitized from
1806 `python-nav-up-list' calls."
1807 (let ((context (python-syntax-context-type))
1808 (forward-p (> dir 0)))
1809 (cond
1810 ((memq context '(string comment)))
1811 ((eq context 'paren)
1812 (let ((forward-sexp-function))
1813 (up-list dir)))
1814 ((and forward-p (python-info-end-of-block-p))
1815 (let ((parent-end-pos
1816 (save-excursion
1817 (let ((indentation (and
1818 (python-nav-beginning-of-block)
1819 (current-indentation))))
1820 (while (and indentation
1821 (> indentation 0)
1822 (>= (current-indentation) indentation)
1823 (python-nav-backward-block)))
1824 (python-nav-end-of-block)))))
1825 (and (> (or parent-end-pos (point)) (point))
1826 (goto-char parent-end-pos))))
1827 (forward-p (python-nav-end-of-block))
1828 ((and (not forward-p)
1829 (> (current-indentation) 0)
1830 (python-info-beginning-of-block-p))
1831 (let ((prev-block-pos
1832 (save-excursion
1833 (let ((indentation (current-indentation)))
1834 (while (and (python-nav-backward-block)
1835 (>= (current-indentation) indentation))))
1836 (point))))
1837 (and (> (point) prev-block-pos)
1838 (goto-char prev-block-pos))))
1839 ((not forward-p) (python-nav-beginning-of-block)))))
1841 (defun python-nav-up-list (&optional arg)
1842 "Move forward out of one level of parentheses (or blocks).
1843 With ARG, do this that many times.
1844 A negative argument means move backward but still to a less deep spot.
1845 This command assumes point is not in a string or comment."
1846 (interactive "^p")
1847 (or arg (setq arg 1))
1848 (while (> arg 0)
1849 (python-nav--up-list 1)
1850 (setq arg (1- arg)))
1851 (while (< arg 0)
1852 (python-nav--up-list -1)
1853 (setq arg (1+ arg))))
1855 (defun python-nav-backward-up-list (&optional arg)
1856 "Move backward out of one level of parentheses (or blocks).
1857 With ARG, do this that many times.
1858 A negative argument means move forward but still to a less deep spot.
1859 This command assumes point is not in a string or comment."
1860 (interactive "^p")
1861 (or arg (setq arg 1))
1862 (python-nav-up-list (- arg)))
1864 (defun python-nav-if-name-main ()
1865 "Move point at the beginning the __main__ block.
1866 When \"if __name__ == \\='__main__\\=':\" is found returns its
1867 position, else returns nil."
1868 (interactive)
1869 (let ((point (point))
1870 (found (catch 'found
1871 (goto-char (point-min))
1872 (while (re-search-forward
1873 (python-rx line-start
1874 "if" (+ space)
1875 "__name__" (+ space)
1876 "==" (+ space)
1877 (group-n 1 (or ?\" ?\'))
1878 "__main__" (backref 1) (* space) ":")
1879 nil t)
1880 (when (not (python-syntax-context-type))
1881 (beginning-of-line)
1882 (throw 'found t))))))
1883 (if found
1884 (point)
1885 (ignore (goto-char point)))))
1888 ;;; Shell integration
1890 (defcustom python-shell-buffer-name "Python"
1891 "Default buffer name for Python interpreter."
1892 :type 'string
1893 :group 'python
1894 :safe 'stringp)
1896 (defcustom python-shell-interpreter "python"
1897 "Default Python interpreter for shell."
1898 :type 'string
1899 :group 'python)
1901 (defcustom python-shell-internal-buffer-name "Python Internal"
1902 "Default buffer name for the Internal Python interpreter."
1903 :type 'string
1904 :group 'python
1905 :safe 'stringp)
1907 (defcustom python-shell-interpreter-args "-i"
1908 "Default arguments for the Python interpreter."
1909 :type 'string
1910 :group 'python)
1912 (defcustom python-shell-interpreter-interactive-arg "-i"
1913 "Interpreter argument to force it to run interactively."
1914 :type 'string
1915 :version "24.4")
1917 (defcustom python-shell-prompt-detect-enabled t
1918 "Non-nil enables autodetection of interpreter prompts."
1919 :type 'boolean
1920 :safe 'booleanp
1921 :version "24.4")
1923 (defcustom python-shell-prompt-detect-failure-warning t
1924 "Non-nil enables warnings when detection of prompts fail."
1925 :type 'boolean
1926 :safe 'booleanp
1927 :version "24.4")
1929 (defcustom python-shell-prompt-input-regexps
1930 '(">>> " "\\.\\.\\. " ; Python
1931 "In \\[[0-9]+\\]: " ; IPython
1932 " \\.\\.\\.: " ; IPython
1933 ;; Using ipdb outside IPython may fail to cleanup and leave static
1934 ;; IPython prompts activated, this adds some safeguard for that.
1935 "In : " "\\.\\.\\.: ")
1936 "List of regular expressions matching input prompts."
1937 :type '(repeat string)
1938 :version "24.4")
1940 (defcustom python-shell-prompt-output-regexps
1941 '("" ; Python
1942 "Out\\[[0-9]+\\]: " ; IPython
1943 "Out :") ; ipdb safeguard
1944 "List of regular expressions matching output prompts."
1945 :type '(repeat string)
1946 :version "24.4")
1948 (defcustom python-shell-prompt-regexp ">>> "
1949 "Regular expression matching top level input prompt of Python shell.
1950 It should not contain a caret (^) at the beginning."
1951 :type 'string)
1953 (defcustom python-shell-prompt-block-regexp "\\.\\.\\. "
1954 "Regular expression matching block input prompt of Python shell.
1955 It should not contain a caret (^) at the beginning."
1956 :type 'string)
1958 (defcustom python-shell-prompt-output-regexp ""
1959 "Regular expression matching output prompt of Python shell.
1960 It should not contain a caret (^) at the beginning."
1961 :type 'string)
1963 (defcustom python-shell-prompt-pdb-regexp "[(<]*[Ii]?[Pp]db[>)]+ "
1964 "Regular expression matching pdb input prompt of Python shell.
1965 It should not contain a caret (^) at the beginning."
1966 :type 'string)
1968 (define-obsolete-variable-alias
1969 'python-shell-enable-font-lock 'python-shell-font-lock-enable "25.1")
1971 (defcustom python-shell-font-lock-enable t
1972 "Should syntax highlighting be enabled in the Python shell buffer?
1973 Restart the Python shell after changing this variable for it to take effect."
1974 :type 'boolean
1975 :group 'python
1976 :safe 'booleanp)
1978 (defcustom python-shell-unbuffered t
1979 "Should shell output be unbuffered?.
1980 When non-nil, this may prevent delayed and missing output in the
1981 Python shell. See commentary for details."
1982 :type 'boolean
1983 :group 'python
1984 :safe 'booleanp)
1986 (defcustom python-shell-process-environment nil
1987 "List of overridden environment variables for subprocesses to inherit.
1988 Each element should be a string of the form ENVVARNAME=VALUE.
1989 When this variable is non-nil, values are exported into the
1990 process environment before starting it. Any variables already
1991 present in the current environment are superseded by variables
1992 set here."
1993 :type '(repeat string)
1994 :group 'python)
1996 (defcustom python-shell-extra-pythonpaths nil
1997 "List of extra pythonpaths for Python shell.
1998 When this variable is non-nil, values added at the beginning of
1999 the PYTHONPATH before starting processes. Any values present
2000 here that already exists in PYTHONPATH are moved to the beginning
2001 of the list so that they are prioritized when looking for
2002 modules."
2003 :type '(repeat string)
2004 :group 'python)
2006 (defcustom python-shell-exec-path nil
2007 "List of paths for searching executables.
2008 When this variable is non-nil, values added at the beginning of
2009 the PATH before starting processes. Any values present here that
2010 already exists in PATH are moved to the beginning of the list so
2011 that they are prioritized when looking for executables."
2012 :type '(repeat string)
2013 :group 'python)
2015 (defcustom python-shell-remote-exec-path nil
2016 "List of paths to be ensured remotely for searching executables.
2017 When this variable is non-nil, values are exported into remote
2018 hosts PATH before starting processes. Values defined in
2019 `python-shell-exec-path' will take precedence to paths defined
2020 here. Normally you wont use this variable directly unless you
2021 plan to ensure a particular set of paths to all Python shell
2022 executed through tramp connections."
2023 :version "25.1"
2024 :type '(repeat string)
2025 :group 'python)
2027 (defcustom python-shell-virtualenv-root nil
2028 "Path to virtualenv root.
2029 This variable, when set to a string, makes the environment to be
2030 modified such that shells are started within the specified
2031 virtualenv."
2032 :type '(choice (const nil) string)
2033 :group 'python)
2035 (define-obsolete-variable-alias
2036 'python-shell-virtualenv-path 'python-shell-virtualenv-root "25.1")
2038 (defcustom python-shell-setup-codes nil
2039 "List of code run by `python-shell-send-setup-codes'."
2040 :type '(repeat symbol)
2041 :group 'python)
2043 (defcustom python-shell-compilation-regexp-alist
2044 `((,(rx line-start (1+ (any " \t")) "File \""
2045 (group (1+ (not (any "\"<")))) ; avoid `<stdin>' &c
2046 "\", line " (group (1+ digit)))
2047 1 2)
2048 (,(rx " in file " (group (1+ not-newline)) " on line "
2049 (group (1+ digit)))
2050 1 2)
2051 (,(rx line-start "> " (group (1+ (not (any "(\"<"))))
2052 "(" (group (1+ digit)) ")" (1+ (not (any "("))) "()")
2053 1 2))
2054 "`compilation-error-regexp-alist' for inferior Python."
2055 :type '(alist string)
2056 :group 'python)
2058 (defmacro python-shell--add-to-path-with-priority (pathvar paths)
2059 "Modify PATHVAR and ensure PATHS are added only once at beginning."
2060 `(dolist (path (reverse ,paths))
2061 (cl-delete path ,pathvar :test #'string=)
2062 (cl-pushnew path ,pathvar :test #'string=)))
2064 (defun python-shell-calculate-pythonpath ()
2065 "Calculate the PYTHONPATH using `python-shell-extra-pythonpaths'."
2066 (let ((pythonpath
2067 (split-string
2068 (or (getenv "PYTHONPATH") "") path-separator 'omit)))
2069 (python-shell--add-to-path-with-priority
2070 pythonpath python-shell-extra-pythonpaths)
2071 (mapconcat 'identity pythonpath path-separator)))
2073 (defun python-shell-calculate-process-environment ()
2074 "Calculate `process-environment' or `tramp-remote-process-environment'.
2075 Prepends `python-shell-process-environment', sets extra
2076 pythonpaths from `python-shell-extra-pythonpaths' and sets a few
2077 virtualenv related vars. If `default-directory' points to a
2078 remote host, the returned value is intended for
2079 `tramp-remote-process-environment'."
2080 (let* ((remote-p (file-remote-p default-directory))
2081 (process-environment (if remote-p
2082 tramp-remote-process-environment
2083 process-environment))
2084 (virtualenv (when python-shell-virtualenv-root
2085 (directory-file-name python-shell-virtualenv-root))))
2086 (dolist (env python-shell-process-environment)
2087 (pcase-let ((`(,key ,value) (split-string env "=")))
2088 (setenv key value)))
2089 (when python-shell-unbuffered
2090 (setenv "PYTHONUNBUFFERED" "1"))
2091 (when python-shell-extra-pythonpaths
2092 (setenv "PYTHONPATH" (python-shell-calculate-pythonpath)))
2093 (if (not virtualenv)
2094 process-environment
2095 (setenv "PYTHONHOME" nil)
2096 (setenv "VIRTUAL_ENV" virtualenv))
2097 process-environment))
2099 (defun python-shell-calculate-exec-path ()
2100 "Calculate `exec-path'.
2101 Prepends `python-shell-exec-path' and adds the binary directory
2102 for virtualenv if `python-shell-virtualenv-root' is set. If
2103 `default-directory' points to a remote host, the returned value
2104 appends `python-shell-remote-exec-path' instead of `exec-path'."
2105 (let ((new-path (copy-sequence
2106 (if (file-remote-p default-directory)
2107 python-shell-remote-exec-path
2108 exec-path))))
2109 (python-shell--add-to-path-with-priority
2110 new-path python-shell-exec-path)
2111 (if (not python-shell-virtualenv-root)
2112 new-path
2113 (python-shell--add-to-path-with-priority
2114 new-path
2115 (list (expand-file-name "bin" python-shell-virtualenv-root)))
2116 new-path)))
2118 (defun python-shell-tramp-refresh-remote-path (vec paths)
2119 "Update VEC's remote-path giving PATHS priority."
2120 (let ((remote-path (tramp-get-connection-property vec "remote-path" nil)))
2121 (when remote-path
2122 (python-shell--add-to-path-with-priority remote-path paths)
2123 (tramp-set-connection-property vec "remote-path" remote-path)
2124 (tramp-set-remote-path vec))))
2126 (defun python-shell-tramp-refresh-process-environment (vec env)
2127 "Update VEC's process environment with ENV."
2128 ;; Stolen from `tramp-open-connection-setup-interactive-shell'.
2129 (let ((env (append (when (fboundp #'tramp-get-remote-locale)
2130 ;; Emacs<24.4 compat.
2131 (list (tramp-get-remote-locale vec)))
2132 (copy-sequence env)))
2133 (tramp-end-of-heredoc
2134 (if (boundp 'tramp-end-of-heredoc)
2135 tramp-end-of-heredoc
2136 (md5 tramp-end-of-output)))
2137 unset vars item)
2138 (while env
2139 (setq item (split-string (car env) "=" 'omit))
2140 (setcdr item (mapconcat 'identity (cdr item) "="))
2141 (if (and (stringp (cdr item)) (not (string-equal (cdr item) "")))
2142 (push (format "%s %s" (car item) (cdr item)) vars)
2143 (push (car item) unset))
2144 (setq env (cdr env)))
2145 (when vars
2146 (tramp-send-command
2148 (format "while read var val; do export $var=$val; done <<'%s'\n%s\n%s"
2149 tramp-end-of-heredoc
2150 (mapconcat 'identity vars "\n")
2151 tramp-end-of-heredoc)
2153 (when unset
2154 (tramp-send-command
2155 vec (format "unset %s" (mapconcat 'identity unset " ")) t))))
2157 (defmacro python-shell-with-environment (&rest body)
2158 "Modify shell environment during execution of BODY.
2159 Temporarily sets `process-environment' and `exec-path' during
2160 execution of body. If `default-directory' points to a remote
2161 machine then modifies `tramp-remote-process-environment' and
2162 `python-shell-remote-exec-path' instead."
2163 (declare (indent 0) (debug (body)))
2164 (let ((vec (make-symbol "vec")))
2165 `(progn
2166 (let* ((,vec
2167 (when (file-remote-p default-directory)
2168 (ignore-errors
2169 (tramp-dissect-file-name default-directory 'noexpand))))
2170 (process-environment
2171 (if ,vec
2172 process-environment
2173 (python-shell-calculate-process-environment)))
2174 (exec-path
2175 (if ,vec
2176 exec-path
2177 (python-shell-calculate-exec-path)))
2178 (tramp-remote-process-environment
2179 (if ,vec
2180 (python-shell-calculate-process-environment)
2181 tramp-remote-process-environment)))
2182 (when (tramp-get-connection-process ,vec)
2183 ;; For already existing connections, the new exec path must
2184 ;; be re-set, otherwise it won't take effect. One example
2185 ;; of such case is when remote dir-locals are read and
2186 ;; *then* subprocesses are triggered within the same
2187 ;; connection.
2188 (python-shell-tramp-refresh-remote-path
2189 ,vec (python-shell-calculate-exec-path))
2190 ;; The `tramp-remote-process-environment' variable is only
2191 ;; effective when the started process is an interactive
2192 ;; shell, otherwise (like in the case of processes started
2193 ;; with `process-file') the environment is not changed.
2194 ;; This makes environment modifications effective
2195 ;; unconditionally.
2196 (python-shell-tramp-refresh-process-environment
2197 ,vec tramp-remote-process-environment))
2198 ,(macroexp-progn body)))))
2200 (defvar python-shell--prompt-calculated-input-regexp nil
2201 "Calculated input prompt regexp for inferior python shell.
2202 Do not set this variable directly, instead use
2203 `python-shell-prompt-set-calculated-regexps'.")
2205 (defvar python-shell--prompt-calculated-output-regexp nil
2206 "Calculated output prompt regexp for inferior python shell.
2207 Do not set this variable directly, instead use
2208 `python-shell-set-prompt-regexp'.")
2210 (defun python-shell-prompt-detect ()
2211 "Detect prompts for the current `python-shell-interpreter'.
2212 When prompts can be retrieved successfully from the
2213 `python-shell-interpreter' run with
2214 `python-shell-interpreter-interactive-arg', returns a list of
2215 three elements, where the first two are input prompts and the
2216 last one is an output prompt. When no prompts can be detected
2217 and `python-shell-prompt-detect-failure-warning' is non-nil,
2218 shows a warning with instructions to avoid hangs and returns nil.
2219 When `python-shell-prompt-detect-enabled' is nil avoids any
2220 detection and just returns nil."
2221 (when python-shell-prompt-detect-enabled
2222 (python-shell-with-environment
2223 (let* ((code (concat
2224 "import sys\n"
2225 "ps = [getattr(sys, 'ps%s' % i, '') for i in range(1,4)]\n"
2226 ;; JSON is built manually for compatibility
2227 "ps_json = '\\n[\"%s\", \"%s\", \"%s\"]\\n' % tuple(ps)\n"
2228 "print (ps_json)\n"
2229 "sys.exit(0)\n"))
2230 (interpreter python-shell-interpreter)
2231 (interpreter-arg python-shell-interpreter-interactive-arg)
2232 (output
2233 (with-temp-buffer
2234 ;; TODO: improve error handling by using
2235 ;; `condition-case' and displaying the error message to
2236 ;; the user in the no-prompts warning.
2237 (ignore-errors
2238 (let ((code-file (python-shell--save-temp-file code)))
2239 ;; Use `process-file' as it is remote-host friendly.
2240 (process-file
2241 interpreter
2242 code-file
2243 '(t nil)
2245 interpreter-arg)
2246 ;; Try to cleanup
2247 (delete-file code-file)))
2248 (buffer-string)))
2249 (prompts
2250 (catch 'prompts
2251 (dolist (line (split-string output "\n" t))
2252 (let ((res
2253 ;; Check if current line is a valid JSON array
2254 (and (string= (substring line 0 2) "[\"")
2255 (ignore-errors
2256 ;; Return prompts as a list, not vector
2257 (append (json-read-from-string line) nil)))))
2258 ;; The list must contain 3 strings, where the first
2259 ;; is the input prompt, the second is the block
2260 ;; prompt and the last one is the output prompt. The
2261 ;; input prompt is the only one that can't be empty.
2262 (when (and (= (length res) 3)
2263 (cl-every #'stringp res)
2264 (not (string= (car res) "")))
2265 (throw 'prompts res))))
2266 nil)))
2267 (when (and (not prompts)
2268 python-shell-prompt-detect-failure-warning)
2269 (lwarn
2270 '(python python-shell-prompt-regexp)
2271 :warning
2272 (concat
2273 "Python shell prompts cannot be detected.\n"
2274 "If your emacs session hangs when starting python shells\n"
2275 "recover with `keyboard-quit' and then try fixing the\n"
2276 "interactive flag for your interpreter by adjusting the\n"
2277 "`python-shell-interpreter-interactive-arg' or add regexps\n"
2278 "matching shell prompts in the directory-local friendly vars:\n"
2279 " + `python-shell-prompt-regexp'\n"
2280 " + `python-shell-prompt-block-regexp'\n"
2281 " + `python-shell-prompt-output-regexp'\n"
2282 "Or alternatively in:\n"
2283 " + `python-shell-prompt-input-regexps'\n"
2284 " + `python-shell-prompt-output-regexps'")))
2285 prompts))))
2287 (defun python-shell-prompt-validate-regexps ()
2288 "Validate all user provided regexps for prompts.
2289 Signals `user-error' if any of these vars contain invalid
2290 regexps: `python-shell-prompt-regexp',
2291 `python-shell-prompt-block-regexp',
2292 `python-shell-prompt-pdb-regexp',
2293 `python-shell-prompt-output-regexp',
2294 `python-shell-prompt-input-regexps',
2295 `python-shell-prompt-output-regexps'."
2296 (dolist (symbol (list 'python-shell-prompt-input-regexps
2297 'python-shell-prompt-output-regexps
2298 'python-shell-prompt-regexp
2299 'python-shell-prompt-block-regexp
2300 'python-shell-prompt-pdb-regexp
2301 'python-shell-prompt-output-regexp))
2302 (dolist (regexp (let ((regexps (symbol-value symbol)))
2303 (if (listp regexps)
2304 regexps
2305 (list regexps))))
2306 (when (not (python-util-valid-regexp-p regexp))
2307 (user-error "Invalid regexp %s in `%s'"
2308 regexp symbol)))))
2310 (defun python-shell-prompt-set-calculated-regexps ()
2311 "Detect and set input and output prompt regexps.
2312 Build and set the values for `python-shell-input-prompt-regexp'
2313 and `python-shell-output-prompt-regexp' using the values from
2314 `python-shell-prompt-regexp', `python-shell-prompt-block-regexp',
2315 `python-shell-prompt-pdb-regexp',
2316 `python-shell-prompt-output-regexp',
2317 `python-shell-prompt-input-regexps',
2318 `python-shell-prompt-output-regexps' and detected prompts from
2319 `python-shell-prompt-detect'."
2320 (when (not (and python-shell--prompt-calculated-input-regexp
2321 python-shell--prompt-calculated-output-regexp))
2322 (let* ((detected-prompts (python-shell-prompt-detect))
2323 (input-prompts nil)
2324 (output-prompts nil)
2325 (build-regexp
2326 (lambda (prompts)
2327 (concat "^\\("
2328 (mapconcat #'identity
2329 (sort prompts
2330 (lambda (a b)
2331 (let ((length-a (length a))
2332 (length-b (length b)))
2333 (if (= length-a length-b)
2334 (string< a b)
2335 (> (length a) (length b))))))
2336 "\\|")
2337 "\\)"))))
2338 ;; Validate ALL regexps
2339 (python-shell-prompt-validate-regexps)
2340 ;; Collect all user defined input prompts
2341 (dolist (prompt (append python-shell-prompt-input-regexps
2342 (list python-shell-prompt-regexp
2343 python-shell-prompt-block-regexp
2344 python-shell-prompt-pdb-regexp)))
2345 (cl-pushnew prompt input-prompts :test #'string=))
2346 ;; Collect all user defined output prompts
2347 (dolist (prompt (cons python-shell-prompt-output-regexp
2348 python-shell-prompt-output-regexps))
2349 (cl-pushnew prompt output-prompts :test #'string=))
2350 ;; Collect detected prompts if any
2351 (when detected-prompts
2352 (dolist (prompt (butlast detected-prompts))
2353 (setq prompt (regexp-quote prompt))
2354 (cl-pushnew prompt input-prompts :test #'string=))
2355 (cl-pushnew (regexp-quote
2356 (car (last detected-prompts)))
2357 output-prompts :test #'string=))
2358 ;; Set input and output prompt regexps from collected prompts
2359 (setq python-shell--prompt-calculated-input-regexp
2360 (funcall build-regexp input-prompts)
2361 python-shell--prompt-calculated-output-regexp
2362 (funcall build-regexp output-prompts)))))
2364 (defun python-shell-get-process-name (dedicated)
2365 "Calculate the appropriate process name for inferior Python process.
2366 If DEDICATED is t returns a string with the form
2367 `python-shell-buffer-name'[`buffer-name'] else returns the value
2368 of `python-shell-buffer-name'."
2369 (if dedicated
2370 (format "%s[%s]" python-shell-buffer-name (buffer-name))
2371 python-shell-buffer-name))
2373 (defun python-shell-internal-get-process-name ()
2374 "Calculate the appropriate process name for Internal Python process.
2375 The name is calculated from `python-shell-global-buffer-name' and
2376 the `buffer-name'."
2377 (format "%s[%s]" python-shell-internal-buffer-name (buffer-name)))
2379 (defun python-shell-calculate-command ()
2380 "Calculate the string used to execute the inferior Python process."
2381 (format "%s %s"
2382 ;; `python-shell-make-comint' expects to be able to
2383 ;; `split-string-and-unquote' the result of this function.
2384 (combine-and-quote-strings (list python-shell-interpreter))
2385 python-shell-interpreter-args))
2387 (define-obsolete-function-alias
2388 'python-shell-parse-command
2389 #'python-shell-calculate-command "25.1")
2391 (defvar python-shell--package-depth 10)
2393 (defun python-shell-package-enable (directory package)
2394 "Add DIRECTORY parent to $PYTHONPATH and enable PACKAGE."
2395 (interactive
2396 (let* ((dir (expand-file-name
2397 (read-directory-name
2398 "Package root: "
2399 (file-name-directory
2400 (or (buffer-file-name) default-directory)))))
2401 (name (completing-read
2402 "Package: "
2403 (python-util-list-packages
2404 dir python-shell--package-depth))))
2405 (list dir name)))
2406 (python-shell-send-string
2407 (format
2408 (concat
2409 "import os.path;import sys;"
2410 "sys.path.append(os.path.dirname(os.path.dirname('''%s''')));"
2411 "__package__ = '''%s''';"
2412 "import %s")
2413 directory package package)
2414 (python-shell-get-process)))
2416 (defun python-shell-accept-process-output (process &optional timeout regexp)
2417 "Accept PROCESS output with TIMEOUT until REGEXP is found.
2418 Optional argument TIMEOUT is the timeout argument to
2419 `accept-process-output' calls. Optional argument REGEXP
2420 overrides the regexp to match the end of output, defaults to
2421 `comint-prompt-regexp.'. Returns non-nil when output was
2422 properly captured.
2424 This utility is useful in situations where the output may be
2425 received in chunks, since `accept-process-output' gives no
2426 guarantees they will be grabbed in a single call. An example use
2427 case for this would be the CPython shell start-up, where the
2428 banner and the initial prompt are received separately."
2429 (let ((regexp (or regexp comint-prompt-regexp)))
2430 (catch 'found
2431 (while t
2432 (when (not (accept-process-output process timeout))
2433 (throw 'found nil))
2434 (when (looking-back
2435 regexp (car (python-util-comint-last-prompt)))
2436 (throw 'found t))))))
2438 (defun python-shell-comint-end-of-output-p (output)
2439 "Return non-nil if OUTPUT is ends with input prompt."
2440 (string-match
2441 ;; XXX: It seems on macOS an extra carriage return is attached
2442 ;; at the end of output, this handles that too.
2443 (concat
2444 "\r?\n?"
2445 ;; Remove initial caret from calculated regexp
2446 (replace-regexp-in-string
2447 (rx string-start ?^) ""
2448 python-shell--prompt-calculated-input-regexp)
2449 (rx eos))
2450 output))
2452 (define-obsolete-function-alias
2453 'python-comint-output-filter-function
2454 'ansi-color-filter-apply
2455 "25.1")
2457 (defun python-comint-postoutput-scroll-to-bottom (output)
2458 "Faster version of `comint-postoutput-scroll-to-bottom'.
2459 Avoids `recenter' calls until OUTPUT is completely sent."
2460 (when (and (not (string= "" output))
2461 (python-shell-comint-end-of-output-p
2462 (ansi-color-filter-apply output)))
2463 (comint-postoutput-scroll-to-bottom output))
2464 output)
2466 (defvar python-shell--parent-buffer nil)
2468 (defmacro python-shell-with-shell-buffer (&rest body)
2469 "Execute the forms in BODY with the shell buffer temporarily current.
2470 Signals an error if no shell buffer is available for current buffer."
2471 (declare (indent 0) (debug t))
2472 (let ((shell-process (make-symbol "shell-process")))
2473 `(let ((,shell-process (python-shell-get-process-or-error)))
2474 (with-current-buffer (process-buffer ,shell-process)
2475 ,@body))))
2477 (defvar python-shell--font-lock-buffer nil)
2479 (defun python-shell-font-lock-get-or-create-buffer ()
2480 "Get or create a font-lock buffer for current inferior process."
2481 (python-shell-with-shell-buffer
2482 (if python-shell--font-lock-buffer
2483 python-shell--font-lock-buffer
2484 (let ((process-name
2485 (process-name (get-buffer-process (current-buffer)))))
2486 (generate-new-buffer
2487 (format " *%s-font-lock*" process-name))))))
2489 (defun python-shell-font-lock-kill-buffer ()
2490 "Kill the font-lock buffer safely."
2491 (when (and python-shell--font-lock-buffer
2492 (buffer-live-p python-shell--font-lock-buffer))
2493 (kill-buffer python-shell--font-lock-buffer)
2494 (when (derived-mode-p 'inferior-python-mode)
2495 (setq python-shell--font-lock-buffer nil))))
2497 (defmacro python-shell-font-lock-with-font-lock-buffer (&rest body)
2498 "Execute the forms in BODY in the font-lock buffer.
2499 The value returned is the value of the last form in BODY. See
2500 also `with-current-buffer'."
2501 (declare (indent 0) (debug t))
2502 `(python-shell-with-shell-buffer
2503 (save-current-buffer
2504 (when (not (and python-shell--font-lock-buffer
2505 (get-buffer python-shell--font-lock-buffer)))
2506 (setq python-shell--font-lock-buffer
2507 (python-shell-font-lock-get-or-create-buffer)))
2508 (set-buffer python-shell--font-lock-buffer)
2509 (when (not font-lock-mode)
2510 (font-lock-mode 1))
2511 (set (make-local-variable 'delay-mode-hooks) t)
2512 (let ((python-indent-guess-indent-offset nil))
2513 (when (not (derived-mode-p 'python-mode))
2514 (python-mode))
2515 ,@body))))
2517 (defun python-shell-font-lock-cleanup-buffer ()
2518 "Cleanup the font-lock buffer.
2519 Provided as a command because this might be handy if something
2520 goes wrong and syntax highlighting in the shell gets messed up."
2521 (interactive)
2522 (python-shell-with-shell-buffer
2523 (python-shell-font-lock-with-font-lock-buffer
2524 (erase-buffer))))
2526 (defun python-shell-font-lock-comint-output-filter-function (output)
2527 "Clean up the font-lock buffer after any OUTPUT."
2528 (if (and (not (string= "" output))
2529 ;; Is end of output and is not just a prompt.
2530 (not (member
2531 (python-shell-comint-end-of-output-p
2532 (ansi-color-filter-apply output))
2533 '(nil 0))))
2534 ;; If output is other than an input prompt then "real" output has
2535 ;; been received and the font-lock buffer must be cleaned up.
2536 (python-shell-font-lock-cleanup-buffer)
2537 ;; Otherwise just add a newline.
2538 (python-shell-font-lock-with-font-lock-buffer
2539 (goto-char (point-max))
2540 (newline)))
2541 output)
2543 (defun python-shell-font-lock-post-command-hook ()
2544 "Fontifies current line in shell buffer."
2545 (let ((prompt-end (cdr (python-util-comint-last-prompt))))
2546 (when (and prompt-end (> (point) prompt-end)
2547 (process-live-p (get-buffer-process (current-buffer))))
2548 (let* ((input (buffer-substring-no-properties
2549 prompt-end (point-max)))
2550 (deactivate-mark nil)
2551 (start-pos prompt-end)
2552 (buffer-undo-list t)
2553 (font-lock-buffer-pos nil)
2554 (replacement
2555 (python-shell-font-lock-with-font-lock-buffer
2556 (delete-region (line-beginning-position)
2557 (point-max))
2558 (setq font-lock-buffer-pos (point))
2559 (insert input)
2560 ;; Ensure buffer is fontified, keeping it
2561 ;; compatible with Emacs < 24.4.
2562 (if (fboundp 'font-lock-ensure)
2563 (funcall 'font-lock-ensure)
2564 (font-lock-default-fontify-buffer))
2565 (buffer-substring font-lock-buffer-pos
2566 (point-max))))
2567 (replacement-length (length replacement))
2568 (i 0))
2569 ;; Inject text properties to get input fontified.
2570 (while (not (= i replacement-length))
2571 (let* ((plist (text-properties-at i replacement))
2572 (next-change (or (next-property-change i replacement)
2573 replacement-length))
2574 (plist (let ((face (plist-get plist 'face)))
2575 (if (not face)
2576 plist
2577 ;; Replace FACE text properties with
2578 ;; FONT-LOCK-FACE so input is fontified.
2579 (plist-put plist 'face nil)
2580 (plist-put plist 'font-lock-face face)))))
2581 (set-text-properties
2582 (+ start-pos i) (+ start-pos next-change) plist)
2583 (setq i next-change)))))))
2585 (defun python-shell-font-lock-turn-on (&optional msg)
2586 "Turn on shell font-lock.
2587 With argument MSG show activation message."
2588 (interactive "p")
2589 (python-shell-with-shell-buffer
2590 (python-shell-font-lock-kill-buffer)
2591 (set (make-local-variable 'python-shell--font-lock-buffer) nil)
2592 (add-hook 'post-command-hook
2593 #'python-shell-font-lock-post-command-hook nil 'local)
2594 (add-hook 'kill-buffer-hook
2595 #'python-shell-font-lock-kill-buffer nil 'local)
2596 (add-hook 'comint-output-filter-functions
2597 #'python-shell-font-lock-comint-output-filter-function
2598 'append 'local)
2599 (when msg
2600 (message "Shell font-lock is enabled"))))
2602 (defun python-shell-font-lock-turn-off (&optional msg)
2603 "Turn off shell font-lock.
2604 With argument MSG show deactivation message."
2605 (interactive "p")
2606 (python-shell-with-shell-buffer
2607 (python-shell-font-lock-kill-buffer)
2608 (when (python-util-comint-last-prompt)
2609 ;; Cleanup current fontification
2610 (remove-text-properties
2611 (cdr (python-util-comint-last-prompt))
2612 (line-end-position)
2613 '(face nil font-lock-face nil)))
2614 (set (make-local-variable 'python-shell--font-lock-buffer) nil)
2615 (remove-hook 'post-command-hook
2616 #'python-shell-font-lock-post-command-hook 'local)
2617 (remove-hook 'kill-buffer-hook
2618 #'python-shell-font-lock-kill-buffer 'local)
2619 (remove-hook 'comint-output-filter-functions
2620 #'python-shell-font-lock-comint-output-filter-function
2621 'local)
2622 (when msg
2623 (message "Shell font-lock is disabled"))))
2625 (defun python-shell-font-lock-toggle (&optional msg)
2626 "Toggle font-lock for shell.
2627 With argument MSG show activation/deactivation message."
2628 (interactive "p")
2629 (python-shell-with-shell-buffer
2630 (set (make-local-variable 'python-shell-font-lock-enable)
2631 (not python-shell-font-lock-enable))
2632 (if python-shell-font-lock-enable
2633 (python-shell-font-lock-turn-on msg)
2634 (python-shell-font-lock-turn-off msg))
2635 python-shell-font-lock-enable))
2637 (defvar python-shell--first-prompt-received-output-buffer nil)
2638 (defvar python-shell--first-prompt-received nil)
2640 (defcustom python-shell-first-prompt-hook nil
2641 "Hook run upon first (non-pdb) shell prompt detection.
2642 This is the place for shell setup functions that need to wait for
2643 output. Since the first prompt is ensured, this helps the
2644 current process to not hang waiting for output by safeguarding
2645 interactive actions can be performed. This is useful to safely
2646 attach setup code for long-running processes that eventually
2647 provide a shell."
2648 :version "25.1"
2649 :type 'hook
2650 :group 'python)
2652 (defun python-shell-comint-watch-for-first-prompt-output-filter (output)
2653 "Run `python-shell-first-prompt-hook' when first prompt is found in OUTPUT."
2654 (when (not python-shell--first-prompt-received)
2655 (set (make-local-variable 'python-shell--first-prompt-received-output-buffer)
2656 (concat python-shell--first-prompt-received-output-buffer
2657 (ansi-color-filter-apply output)))
2658 (when (python-shell-comint-end-of-output-p
2659 python-shell--first-prompt-received-output-buffer)
2660 (if (string-match-p
2661 (concat python-shell-prompt-pdb-regexp (rx eos))
2662 (or python-shell--first-prompt-received-output-buffer ""))
2663 ;; Skip pdb prompts and reset the buffer.
2664 (setq python-shell--first-prompt-received-output-buffer nil)
2665 (set (make-local-variable 'python-shell--first-prompt-received) t)
2666 (setq python-shell--first-prompt-received-output-buffer nil)
2667 (with-current-buffer (current-buffer)
2668 (let ((inhibit-quit nil))
2669 (run-hooks 'python-shell-first-prompt-hook))))))
2670 output)
2672 ;; Used to hold user interactive overrides to
2673 ;; `python-shell-interpreter' and `python-shell-interpreter-args' that
2674 ;; will be made buffer-local by `inferior-python-mode':
2675 (defvar python-shell--interpreter)
2676 (defvar python-shell--interpreter-args)
2678 (define-derived-mode inferior-python-mode comint-mode "Inferior Python"
2679 "Major mode for Python inferior process.
2680 Runs a Python interpreter as a subprocess of Emacs, with Python
2681 I/O through an Emacs buffer. Variables `python-shell-interpreter'
2682 and `python-shell-interpreter-args' control which Python
2683 interpreter is run. Variables
2684 `python-shell-prompt-regexp',
2685 `python-shell-prompt-output-regexp',
2686 `python-shell-prompt-block-regexp',
2687 `python-shell-font-lock-enable',
2688 `python-shell-completion-setup-code',
2689 `python-shell-completion-string-code',
2690 `python-eldoc-setup-code', `python-eldoc-string-code',
2691 `python-ffap-setup-code' and `python-ffap-string-code' can
2692 customize this mode for different Python interpreters.
2694 This mode resets `comint-output-filter-functions' locally, so you
2695 may want to re-add custom functions to it using the
2696 `inferior-python-mode-hook'.
2698 You can also add additional setup code to be run at
2699 initialization of the interpreter via `python-shell-setup-codes'
2700 variable.
2702 \(Type \\[describe-mode] in the process buffer for a list of commands.)"
2703 (when python-shell--parent-buffer
2704 (python-util-clone-local-variables python-shell--parent-buffer))
2705 (set (make-local-variable 'indent-tabs-mode) nil)
2706 ;; Users can interactively override default values for
2707 ;; `python-shell-interpreter' and `python-shell-interpreter-args'
2708 ;; when calling `run-python'. This ensures values let-bound in
2709 ;; `python-shell-make-comint' are locally set if needed.
2710 (set (make-local-variable 'python-shell-interpreter)
2711 (or python-shell--interpreter python-shell-interpreter))
2712 (set (make-local-variable 'python-shell-interpreter-args)
2713 (or python-shell--interpreter-args python-shell-interpreter-args))
2714 (set (make-local-variable 'python-shell--prompt-calculated-input-regexp) nil)
2715 (set (make-local-variable 'python-shell--prompt-calculated-output-regexp) nil)
2716 (python-shell-prompt-set-calculated-regexps)
2717 (setq comint-prompt-regexp python-shell--prompt-calculated-input-regexp)
2718 (set (make-local-variable 'comint-prompt-read-only) t)
2719 (setq mode-line-process '(":%s"))
2720 (set (make-local-variable 'comint-output-filter-functions)
2721 '(ansi-color-process-output
2722 python-shell-comint-watch-for-first-prompt-output-filter
2723 python-pdbtrack-comint-output-filter-function
2724 python-comint-postoutput-scroll-to-bottom))
2725 (set (make-local-variable 'compilation-error-regexp-alist)
2726 python-shell-compilation-regexp-alist)
2727 (add-hook 'completion-at-point-functions
2728 #'python-shell-completion-at-point nil 'local)
2729 (define-key inferior-python-mode-map "\t"
2730 'python-shell-completion-complete-or-indent)
2731 (make-local-variable 'python-pdbtrack-buffers-to-kill)
2732 (make-local-variable 'python-pdbtrack-tracked-buffer)
2733 (make-local-variable 'python-shell-internal-last-output)
2734 (when python-shell-font-lock-enable
2735 (python-shell-font-lock-turn-on))
2736 (compilation-shell-minor-mode 1))
2738 (defun python-shell-make-comint (cmd proc-name &optional show internal)
2739 "Create a Python shell comint buffer.
2740 CMD is the Python command to be executed and PROC-NAME is the
2741 process name the comint buffer will get. After the comint buffer
2742 is created the `inferior-python-mode' is activated. When
2743 optional argument SHOW is non-nil the buffer is shown. When
2744 optional argument INTERNAL is non-nil this process is run on a
2745 buffer with a name that starts with a space, following the Emacs
2746 convention for temporary/internal buffers, and also makes sure
2747 the user is not queried for confirmation when the process is
2748 killed."
2749 (save-excursion
2750 (python-shell-with-environment
2751 (let* ((proc-buffer-name
2752 (format (if (not internal) "*%s*" " *%s*") proc-name)))
2753 (when (not (comint-check-proc proc-buffer-name))
2754 (let* ((cmdlist (split-string-and-unquote cmd))
2755 (interpreter (car cmdlist))
2756 (args (cdr cmdlist))
2757 (buffer (apply #'make-comint-in-buffer proc-name proc-buffer-name
2758 interpreter nil args))
2759 (python-shell--parent-buffer (current-buffer))
2760 (process (get-buffer-process buffer))
2761 ;; Users can override the interpreter and args
2762 ;; interactively when calling `run-python', let-binding
2763 ;; these allows having the new right values in all
2764 ;; setup code that is done in `inferior-python-mode',
2765 ;; which is important, especially for prompt detection.
2766 (python-shell--interpreter interpreter)
2767 (python-shell--interpreter-args
2768 (mapconcat #'identity args " ")))
2769 (with-current-buffer buffer
2770 (inferior-python-mode))
2771 (when show (display-buffer buffer))
2772 (and internal (set-process-query-on-exit-flag process nil))))
2773 proc-buffer-name))))
2775 ;;;###autoload
2776 (defun run-python (&optional cmd dedicated show)
2777 "Run an inferior Python process.
2779 Argument CMD defaults to `python-shell-calculate-command' return
2780 value. When called interactively with `prefix-arg', it allows
2781 the user to edit such value and choose whether the interpreter
2782 should be DEDICATED for the current buffer. When numeric prefix
2783 arg is other than 0 or 4 do not SHOW.
2785 For a given buffer and same values of DEDICATED, if a process is
2786 already running for it, it will do nothing. This means that if
2787 the current buffer is using a global process, the user is still
2788 able to switch it to use a dedicated one.
2790 Runs the hook `inferior-python-mode-hook' after
2791 `comint-mode-hook' is run. (Type \\[describe-mode] in the
2792 process buffer for a list of commands.)"
2793 (interactive
2794 (if current-prefix-arg
2795 (list
2796 (read-shell-command "Run Python: " (python-shell-calculate-command))
2797 (y-or-n-p "Make dedicated process? ")
2798 (= (prefix-numeric-value current-prefix-arg) 4))
2799 (list (python-shell-calculate-command) nil t)))
2800 (get-buffer-process
2801 (python-shell-make-comint
2802 (or cmd (python-shell-calculate-command))
2803 (python-shell-get-process-name dedicated) show)))
2805 (defun run-python-internal ()
2806 "Run an inferior Internal Python process.
2807 Input and output via buffer named after
2808 `python-shell-internal-buffer-name' and what
2809 `python-shell-internal-get-process-name' returns.
2811 This new kind of shell is intended to be used for generic
2812 communication related to defined configurations; the main
2813 difference with global or dedicated shells is that these ones are
2814 attached to a configuration, not a buffer. This means that can
2815 be used for example to retrieve the sys.path and other stuff,
2816 without messing with user shells. Note that
2817 `python-shell-font-lock-enable' and `inferior-python-mode-hook'
2818 are set to nil for these shells, so setup codes are not sent at
2819 startup."
2820 (let ((python-shell-font-lock-enable nil)
2821 (inferior-python-mode-hook nil))
2822 (get-buffer-process
2823 (python-shell-make-comint
2824 (python-shell-calculate-command)
2825 (python-shell-internal-get-process-name) nil t))))
2827 (defun python-shell-get-buffer ()
2828 "Return inferior Python buffer for current buffer.
2829 If current buffer is in `inferior-python-mode', return it."
2830 (if (derived-mode-p 'inferior-python-mode)
2831 (current-buffer)
2832 (let* ((dedicated-proc-name (python-shell-get-process-name t))
2833 (dedicated-proc-buffer-name (format "*%s*" dedicated-proc-name))
2834 (global-proc-name (python-shell-get-process-name nil))
2835 (global-proc-buffer-name (format "*%s*" global-proc-name))
2836 (dedicated-running (comint-check-proc dedicated-proc-buffer-name))
2837 (global-running (comint-check-proc global-proc-buffer-name)))
2838 ;; Always prefer dedicated
2839 (or (and dedicated-running dedicated-proc-buffer-name)
2840 (and global-running global-proc-buffer-name)))))
2842 (defun python-shell-get-process ()
2843 "Return inferior Python process for current buffer."
2844 (get-buffer-process (python-shell-get-buffer)))
2846 (defun python-shell-get-process-or-error (&optional interactivep)
2847 "Return inferior Python process for current buffer or signal error.
2848 When argument INTERACTIVEP is non-nil, use `user-error' instead
2849 of `error' with a user-friendly message."
2850 (or (python-shell-get-process)
2851 (if interactivep
2852 (user-error
2853 "Start a Python process first with `%s' or `%s'."
2854 (substitute-command-keys "\\[run-python]")
2855 ;; Get the binding.
2856 (key-description
2857 (where-is-internal
2858 #'run-python overriding-local-map t)))
2859 (error
2860 "No inferior Python process running."))))
2862 (defun python-shell-get-or-create-process (&optional cmd dedicated show)
2863 "Get or create an inferior Python process for current buffer and return it.
2864 Arguments CMD, DEDICATED and SHOW are those of `run-python' and
2865 are used to start the shell. If those arguments are not
2866 provided, `run-python' is called interactively and the user will
2867 be asked for their values."
2868 (let ((shell-process (python-shell-get-process)))
2869 (when (not shell-process)
2870 (if (not cmd)
2871 ;; XXX: Refactor code such that calling `run-python'
2872 ;; interactively is not needed anymore.
2873 (call-interactively 'run-python)
2874 (run-python cmd dedicated show)))
2875 (or shell-process (python-shell-get-process))))
2877 (make-obsolete
2878 #'python-shell-get-or-create-process
2879 "Instead call `python-shell-get-process' and create one if returns nil."
2880 "25.1")
2882 (defvar python-shell-internal-buffer nil
2883 "Current internal shell buffer for the current buffer.
2884 This is really not necessary at all for the code to work but it's
2885 there for compatibility with CEDET.")
2887 (defvar python-shell-internal-last-output nil
2888 "Last output captured by the internal shell.
2889 This is really not necessary at all for the code to work but it's
2890 there for compatibility with CEDET.")
2892 (defun python-shell-internal-get-or-create-process ()
2893 "Get or create an inferior Internal Python process."
2894 (let ((proc-name (python-shell-internal-get-process-name)))
2895 (if (process-live-p proc-name)
2896 (get-process proc-name)
2897 (run-python-internal))))
2899 (define-obsolete-function-alias
2900 'python-proc 'python-shell-internal-get-or-create-process "24.3")
2902 (define-obsolete-variable-alias
2903 'python-buffer 'python-shell-internal-buffer "24.3")
2905 (define-obsolete-variable-alias
2906 'python-preoutput-result 'python-shell-internal-last-output "24.3")
2908 (defun python-shell--save-temp-file (string)
2909 (let* ((temporary-file-directory
2910 (if (file-remote-p default-directory)
2911 (concat (file-remote-p default-directory) "/tmp")
2912 temporary-file-directory))
2913 (temp-file-name (make-temp-file "py"))
2914 (coding-system-for-write (python-info-encoding)))
2915 (with-temp-file temp-file-name
2916 (insert string)
2917 (delete-trailing-whitespace))
2918 temp-file-name))
2920 (defun python-shell-send-string (string &optional process msg)
2921 "Send STRING to inferior Python PROCESS.
2922 When optional argument MSG is non-nil, forces display of a
2923 user-friendly message if there's no process running; defaults to
2924 t when called interactively."
2925 (interactive
2926 (list (read-string "Python command: ") nil t))
2927 (let ((process (or process (python-shell-get-process-or-error msg))))
2928 (if (string-match ".\n+." string) ;Multiline.
2929 (let* ((temp-file-name (python-shell--save-temp-file string))
2930 (file-name (or (buffer-file-name) temp-file-name)))
2931 (python-shell-send-file file-name process temp-file-name t))
2932 (comint-send-string process string)
2933 (when (or (not (string-match "\n\\'" string))
2934 (string-match "\n[ \t].*\n?\\'" string))
2935 (comint-send-string process "\n")))))
2937 (defvar python-shell-output-filter-in-progress nil)
2938 (defvar python-shell-output-filter-buffer nil)
2940 (defun python-shell-output-filter (string)
2941 "Filter used in `python-shell-send-string-no-output' to grab output.
2942 STRING is the output received to this point from the process.
2943 This filter saves received output from the process in
2944 `python-shell-output-filter-buffer' and stops receiving it after
2945 detecting a prompt at the end of the buffer."
2946 (setq
2947 string (ansi-color-filter-apply string)
2948 python-shell-output-filter-buffer
2949 (concat python-shell-output-filter-buffer string))
2950 (when (python-shell-comint-end-of-output-p
2951 python-shell-output-filter-buffer)
2952 ;; Output ends when `python-shell-output-filter-buffer' contains
2953 ;; the prompt attached at the end of it.
2954 (setq python-shell-output-filter-in-progress nil
2955 python-shell-output-filter-buffer
2956 (substring python-shell-output-filter-buffer
2957 0 (match-beginning 0)))
2958 (when (string-match
2959 python-shell--prompt-calculated-output-regexp
2960 python-shell-output-filter-buffer)
2961 ;; Some shells, like IPython might append a prompt before the
2962 ;; output, clean that.
2963 (setq python-shell-output-filter-buffer
2964 (substring python-shell-output-filter-buffer (match-end 0)))))
2967 (defun python-shell-send-string-no-output (string &optional process)
2968 "Send STRING to PROCESS and inhibit output.
2969 Return the output."
2970 (let ((process (or process (python-shell-get-process-or-error)))
2971 (comint-preoutput-filter-functions
2972 '(python-shell-output-filter))
2973 (python-shell-output-filter-in-progress t)
2974 (inhibit-quit t))
2976 (with-local-quit
2977 (python-shell-send-string string process)
2978 (while python-shell-output-filter-in-progress
2979 ;; `python-shell-output-filter' takes care of setting
2980 ;; `python-shell-output-filter-in-progress' to NIL after it
2981 ;; detects end of output.
2982 (accept-process-output process))
2983 (prog1
2984 python-shell-output-filter-buffer
2985 (setq python-shell-output-filter-buffer nil)))
2986 (with-current-buffer (process-buffer process)
2987 (comint-interrupt-subjob)))))
2989 (defun python-shell-internal-send-string (string)
2990 "Send STRING to the Internal Python interpreter.
2991 Returns the output. See `python-shell-send-string-no-output'."
2992 ;; XXX Remove `python-shell-internal-last-output' once CEDET is
2993 ;; updated to support this new mode.
2994 (setq python-shell-internal-last-output
2995 (python-shell-send-string-no-output
2996 ;; Makes this function compatible with the old
2997 ;; python-send-receive. (At least for CEDET).
2998 (replace-regexp-in-string "_emacs_out +" "" string)
2999 (python-shell-internal-get-or-create-process))))
3001 (define-obsolete-function-alias
3002 'python-send-receive 'python-shell-internal-send-string "24.3")
3004 (define-obsolete-function-alias
3005 'python-send-string 'python-shell-internal-send-string "24.3")
3007 (defun python-shell-buffer-substring (start end &optional nomain)
3008 "Send buffer substring from START to END formatted for shell.
3009 This is a wrapper over `buffer-substring' that takes care of
3010 different transformations for the code sent to be evaluated in
3011 the python shell:
3012 1. When optional argument NOMAIN is non-nil everything under an
3013 \"if __name__ == \\='__main__\\='\" block will be removed.
3014 2. When a subregion of the buffer is sent, it takes care of
3015 appending extra empty lines so tracebacks are correct.
3016 3. When the region sent is a substring of the current buffer, a
3017 coding cookie is added.
3018 4. Wraps indented regions under an \"if True:\" block so the
3019 interpreter evaluates them correctly."
3020 (let* ((start (save-excursion
3021 ;; Normalize start to the line beginning position.
3022 (goto-char start)
3023 (line-beginning-position)))
3024 (substring (buffer-substring-no-properties start end))
3025 (starts-at-point-min-p (save-restriction
3026 (widen)
3027 (= (point-min) start)))
3028 (encoding (python-info-encoding))
3029 (toplevel-p (zerop (save-excursion
3030 (goto-char start)
3031 (python-util-forward-comment 1)
3032 (current-indentation))))
3033 (fillstr (when (not starts-at-point-min-p)
3034 (concat
3035 (format "# -*- coding: %s -*-\n" encoding)
3036 (make-string
3037 ;; Subtract 2 because of the coding cookie.
3038 (- (line-number-at-pos start) 2) ?\n)))))
3039 (with-temp-buffer
3040 (python-mode)
3041 (when fillstr
3042 (insert fillstr))
3043 (insert substring)
3044 (goto-char (point-min))
3045 (when (not toplevel-p)
3046 (insert "if True:")
3047 (delete-region (point) (line-end-position)))
3048 (when nomain
3049 (let* ((if-name-main-start-end
3050 (and nomain
3051 (save-excursion
3052 (when (python-nav-if-name-main)
3053 (cons (point)
3054 (progn (python-nav-forward-sexp-safe)
3055 ;; Include ending newline
3056 (forward-line 1)
3057 (point)))))))
3058 ;; Oh destructuring bind, how I miss you.
3059 (if-name-main-start (car if-name-main-start-end))
3060 (if-name-main-end (cdr if-name-main-start-end))
3061 (fillstr (make-string
3062 (- (line-number-at-pos if-name-main-end)
3063 (line-number-at-pos if-name-main-start)) ?\n)))
3064 (when if-name-main-start-end
3065 (goto-char if-name-main-start)
3066 (delete-region if-name-main-start if-name-main-end)
3067 (insert fillstr))))
3068 ;; Ensure there's only one coding cookie in the generated string.
3069 (goto-char (point-min))
3070 (when (looking-at-p (python-rx coding-cookie))
3071 (forward-line 1)
3072 (when (looking-at-p (python-rx coding-cookie))
3073 (delete-region
3074 (line-beginning-position) (line-end-position))))
3075 (buffer-substring-no-properties (point-min) (point-max)))))
3077 (defun python-shell-send-region (start end &optional send-main msg)
3078 "Send the region delimited by START and END to inferior Python process.
3079 When optional argument SEND-MAIN is non-nil, allow execution of
3080 code inside blocks delimited by \"if __name__== \\='__main__\\=':\".
3081 When called interactively SEND-MAIN defaults to nil, unless it's
3082 called with prefix argument. When optional argument MSG is
3083 non-nil, forces display of a user-friendly message if there's no
3084 process running; defaults to t when called interactively."
3085 (interactive
3086 (list (region-beginning) (region-end) current-prefix-arg t))
3087 (let* ((string (python-shell-buffer-substring start end (not send-main)))
3088 (process (python-shell-get-process-or-error msg))
3089 (original-string (buffer-substring-no-properties start end))
3090 (_ (string-match "\\`\n*\\(.*\\)" original-string)))
3091 (message "Sent: %s..." (match-string 1 original-string))
3092 (python-shell-send-string string process)))
3094 (defun python-shell-send-buffer (&optional send-main msg)
3095 "Send the entire buffer to inferior Python process.
3096 When optional argument SEND-MAIN is non-nil, allow execution of
3097 code inside blocks delimited by \"if __name__== \\='__main__\\=':\".
3098 When called interactively SEND-MAIN defaults to nil, unless it's
3099 called with prefix argument. When optional argument MSG is
3100 non-nil, forces display of a user-friendly message if there's no
3101 process running; defaults to t when called interactively."
3102 (interactive (list current-prefix-arg t))
3103 (save-restriction
3104 (widen)
3105 (python-shell-send-region (point-min) (point-max) send-main msg)))
3107 (defun python-shell-send-defun (&optional arg msg)
3108 "Send the current defun to inferior Python process.
3109 When argument ARG is non-nil do not include decorators. When
3110 optional argument MSG is non-nil, forces display of a
3111 user-friendly message if there's no process running; defaults to
3112 t when called interactively."
3113 (interactive (list current-prefix-arg t))
3114 (save-excursion
3115 (python-shell-send-region
3116 (progn
3117 (end-of-line 1)
3118 (while (and (or (python-nav-beginning-of-defun)
3119 (beginning-of-line 1))
3120 (> (current-indentation) 0)))
3121 (when (not arg)
3122 (while (and (forward-line -1)
3123 (looking-at (python-rx decorator))))
3124 (forward-line 1))
3125 (point-marker))
3126 (progn
3127 (or (python-nav-end-of-defun)
3128 (end-of-line 1))
3129 (point-marker))
3130 nil ;; noop
3131 msg)))
3133 (defun python-shell-send-file (file-name &optional process temp-file-name
3134 delete msg)
3135 "Send FILE-NAME to inferior Python PROCESS.
3136 If TEMP-FILE-NAME is passed then that file is used for processing
3137 instead, while internally the shell will continue to use
3138 FILE-NAME. If TEMP-FILE-NAME and DELETE are non-nil, then
3139 TEMP-FILE-NAME is deleted after evaluation is performed. When
3140 optional argument MSG is non-nil, forces display of a
3141 user-friendly message if there's no process running; defaults to
3142 t when called interactively."
3143 (interactive
3144 (list
3145 (read-file-name "File to send: ") ; file-name
3146 nil ; process
3147 nil ; temp-file-name
3148 nil ; delete
3149 t)) ; msg
3150 (let* ((process (or process (python-shell-get-process-or-error msg)))
3151 (encoding (with-temp-buffer
3152 (insert-file-contents
3153 (or temp-file-name file-name))
3154 (python-info-encoding)))
3155 (file-name (expand-file-name (file-local-name file-name)))
3156 (temp-file-name (when temp-file-name
3157 (expand-file-name
3158 (file-local-name temp-file-name)))))
3159 (python-shell-send-string
3160 (format
3161 (concat
3162 "import codecs, os;"
3163 "__pyfile = codecs.open('''%s''', encoding='''%s''');"
3164 "__code = __pyfile.read().encode('''%s''');"
3165 "__pyfile.close();"
3166 (when (and delete temp-file-name)
3167 (format "os.remove('''%s''');" temp-file-name))
3168 "exec(compile(__code, '''%s''', 'exec'));")
3169 (or temp-file-name file-name) encoding encoding file-name)
3170 process)))
3172 (defun python-shell-switch-to-shell (&optional msg)
3173 "Switch to inferior Python process buffer.
3174 When optional argument MSG is non-nil, forces display of a
3175 user-friendly message if there's no process running; defaults to
3176 t when called interactively."
3177 (interactive "p")
3178 (pop-to-buffer
3179 (process-buffer (python-shell-get-process-or-error msg)) nil t))
3181 (defun python-shell-send-setup-code ()
3182 "Send all setup code for shell.
3183 This function takes the list of setup code to send from the
3184 `python-shell-setup-codes' list."
3185 (when python-shell-setup-codes
3186 (let ((process (python-shell-get-process))
3187 (code (concat
3188 (mapconcat
3189 (lambda (elt)
3190 (cond ((stringp elt) elt)
3191 ((symbolp elt) (symbol-value elt))
3192 (t "")))
3193 python-shell-setup-codes
3194 "\n\nprint ('python.el: sent setup code')"))))
3195 (python-shell-send-string code process)
3196 (python-shell-accept-process-output process))))
3198 (add-hook 'python-shell-first-prompt-hook
3199 #'python-shell-send-setup-code)
3202 ;;; Shell completion
3204 (defcustom python-shell-completion-setup-code
3206 def __PYTHON_EL_get_completions(text):
3207 completions = []
3208 completer = None
3210 try:
3211 import readline
3213 try:
3214 import __builtin__
3215 except ImportError:
3216 # Python 3
3217 import builtins as __builtin__
3218 builtins = dir(__builtin__)
3220 is_ipython = ('__IPYTHON__' in builtins or
3221 '__IPYTHON__active' in builtins)
3222 splits = text.split()
3223 is_module = splits and splits[0] in ('from', 'import')
3225 if is_ipython and is_module:
3226 from IPython.core.completerlib import module_completion
3227 completions = module_completion(text.strip())
3228 elif is_ipython and '__IP' in builtins:
3229 completions = __IP.complete(text)
3230 elif is_ipython and 'get_ipython' in builtins:
3231 completions = get_ipython().Completer.all_completions(text)
3232 else:
3233 # Try to reuse current completer.
3234 completer = readline.get_completer()
3235 if not completer:
3236 # importing rlcompleter sets the completer, use it as a
3237 # last resort to avoid breaking customizations.
3238 import rlcompleter
3239 completer = readline.get_completer()
3240 if getattr(completer, 'PYTHON_EL_WRAPPED', False):
3241 completer.print_mode = False
3242 i = 0
3243 while True:
3244 completion = completer(text, i)
3245 if not completion:
3246 break
3247 i += 1
3248 completions.append(completion)
3249 except:
3250 pass
3251 finally:
3252 if getattr(completer, 'PYTHON_EL_WRAPPED', False):
3253 completer.print_mode = True
3254 return completions"
3255 "Code used to setup completion in inferior Python processes."
3256 :type 'string
3257 :group 'python)
3259 (defcustom python-shell-completion-string-code
3260 "';'.join(__PYTHON_EL_get_completions('''%s'''))"
3261 "Python code used to get a string of completions separated by semicolons.
3262 The string passed to the function is the current python name or
3263 the full statement in the case of imports."
3264 :type 'string
3265 :group 'python)
3267 (define-obsolete-variable-alias
3268 'python-shell-completion-module-string-code
3269 'python-shell-completion-string-code
3270 "24.4"
3271 "Completion string code must also autocomplete modules.")
3273 (define-obsolete-variable-alias
3274 'python-shell-completion-pdb-string-code
3275 'python-shell-completion-string-code
3276 "25.1"
3277 "Completion string code must work for (i)pdb.")
3279 (defcustom python-shell-completion-native-disabled-interpreters
3280 ;; PyPy's readline cannot handle some escape sequences yet.
3281 (list "pypy")
3282 "List of disabled interpreters.
3283 When a match is found, native completion is disabled."
3284 :version "25.1"
3285 :type '(repeat string))
3287 (defcustom python-shell-completion-native-enable t
3288 "Enable readline based native completion."
3289 :version "25.1"
3290 :type 'boolean)
3292 (defcustom python-shell-completion-native-output-timeout 5.0
3293 "Time in seconds to wait for completion output before giving up."
3294 :version "25.1"
3295 :type 'float)
3297 (defcustom python-shell-completion-native-try-output-timeout 1.0
3298 "Time in seconds to wait for *trying* native completion output."
3299 :version "25.1"
3300 :type 'float)
3302 (defvar python-shell-completion-native-redirect-buffer
3303 " *Python completions redirect*"
3304 "Buffer to be used to redirect output of readline commands.")
3306 (defun python-shell-completion-native-interpreter-disabled-p ()
3307 "Return non-nil if interpreter has native completion disabled."
3308 (when python-shell-completion-native-disabled-interpreters
3309 (string-match
3310 (regexp-opt python-shell-completion-native-disabled-interpreters)
3311 (file-name-nondirectory python-shell-interpreter))))
3313 (defun python-shell-completion-native-try ()
3314 "Return non-nil if can trigger native completion."
3315 (let ((python-shell-completion-native-enable t)
3316 (python-shell-completion-native-output-timeout
3317 python-shell-completion-native-try-output-timeout))
3318 (python-shell-completion-native-get-completions
3319 (get-buffer-process (current-buffer))
3320 nil "_")))
3322 (defun python-shell-completion-native-setup ()
3323 "Try to setup native completion, return non-nil on success."
3324 (let ((process (python-shell-get-process)))
3325 (with-current-buffer (process-buffer process)
3326 (python-shell-send-string "
3327 def __PYTHON_EL_native_completion_setup():
3328 try:
3329 import readline
3331 try:
3332 import __builtin__
3333 except ImportError:
3334 # Python 3
3335 import builtins as __builtin__
3337 builtins = dir(__builtin__)
3338 is_ipython = ('__IPYTHON__' in builtins or
3339 '__IPYTHON__active' in builtins)
3341 class __PYTHON_EL_Completer:
3342 '''Completer wrapper that prints candidates to stdout.
3344 It wraps an existing completer function and changes its behavior so
3345 that the user input is unchanged and real candidates are printed to
3346 stdout.
3348 Returned candidates are '0__dummy_completion__' and
3349 '1__dummy_completion__' in that order ('0__dummy_completion__' is
3350 returned repeatedly until all possible candidates are consumed).
3352 The real candidates are printed to stdout so that they can be
3353 easily retrieved through comint output redirect trickery.
3356 PYTHON_EL_WRAPPED = True
3358 def __init__(self, completer):
3359 self.completer = completer
3360 self.last_completion = None
3361 self.print_mode = True
3363 def __call__(self, text, state):
3364 if state == 0:
3365 # Set the first dummy completion.
3366 self.last_completion = None
3367 completion = '0__dummy_completion__'
3368 else:
3369 completion = self.completer(text, state - 1)
3371 if not completion:
3372 if self.last_completion != '1__dummy_completion__':
3373 # When no more completions are available, returning a
3374 # dummy with non-sharing prefix allow ensuring output
3375 # while preventing changes to current input.
3376 # Coincidentally it's also the end of output.
3377 completion = '1__dummy_completion__'
3378 elif completion.endswith('('):
3379 # Remove parens on callables as it breaks completion on
3380 # arguments (e.g. str(Ari<tab>)).
3381 completion = completion[:-1]
3382 self.last_completion = completion
3384 if completion in (
3385 '0__dummy_completion__', '1__dummy_completion__'):
3386 return completion
3387 elif completion:
3388 # For every non-dummy completion, return a repeated dummy
3389 # one and print the real candidate so it can be retrieved
3390 # by comint output filters.
3391 if self.print_mode:
3392 print (completion)
3393 return '0__dummy_completion__'
3394 else:
3395 return completion
3396 else:
3397 return completion
3399 completer = readline.get_completer()
3401 if not completer:
3402 # Used as last resort to avoid breaking customizations.
3403 import rlcompleter
3404 completer = readline.get_completer()
3406 if completer and not getattr(completer, 'PYTHON_EL_WRAPPED', False):
3407 # Wrap the existing completer function only once.
3408 new_completer = __PYTHON_EL_Completer(completer)
3409 if not is_ipython:
3410 readline.set_completer(new_completer)
3411 else:
3412 # Try both initializations to cope with all IPython versions.
3413 # This works fine for IPython 3.x but not for earlier:
3414 readline.set_completer(new_completer)
3415 # IPython<3 hacks readline such that `readline.set_completer`
3416 # won't work. This workaround injects the new completer
3417 # function into the existing instance directly:
3418 instance = getattr(completer, 'im_self', completer.__self__)
3419 instance.rlcomplete = new_completer
3421 if readline.__doc__ and 'libedit' in readline.__doc__:
3422 readline.parse_and_bind('bind ^I rl_complete')
3423 else:
3424 readline.parse_and_bind('tab: complete')
3425 # Require just one tab to send output.
3426 readline.parse_and_bind('set show-all-if-ambiguous on')
3428 print ('python.el: native completion setup loaded')
3429 except:
3430 print ('python.el: native completion setup failed')
3432 __PYTHON_EL_native_completion_setup()" process)
3433 (when (and
3434 (python-shell-accept-process-output
3435 process python-shell-completion-native-try-output-timeout)
3436 (save-excursion
3437 (re-search-backward
3438 (regexp-quote "python.el: native completion setup loaded") nil t 1)))
3439 (python-shell-completion-native-try)))))
3441 (defun python-shell-completion-native-turn-off (&optional msg)
3442 "Turn off shell native completions.
3443 With argument MSG show deactivation message."
3444 (interactive "p")
3445 (python-shell-with-shell-buffer
3446 (set (make-local-variable 'python-shell-completion-native-enable) nil)
3447 (when msg
3448 (message "Shell native completion is disabled, using fallback"))))
3450 (defun python-shell-completion-native-turn-on (&optional msg)
3451 "Turn on shell native completions.
3452 With argument MSG show deactivation message."
3453 (interactive "p")
3454 (python-shell-with-shell-buffer
3455 (set (make-local-variable 'python-shell-completion-native-enable) t)
3456 (python-shell-completion-native-turn-on-maybe msg)))
3458 (defun python-shell-completion-native-turn-on-maybe (&optional msg)
3459 "Turn on native completions if enabled and available.
3460 With argument MSG show activation/deactivation message."
3461 (interactive "p")
3462 (python-shell-with-shell-buffer
3463 (when python-shell-completion-native-enable
3464 (cond
3465 ((python-shell-completion-native-interpreter-disabled-p)
3466 (python-shell-completion-native-turn-off msg))
3467 ((python-shell-completion-native-setup)
3468 (when msg
3469 (message "Shell native completion is enabled.")))
3470 (t (lwarn
3471 '(python python-shell-completion-native-turn-on-maybe)
3472 :warning
3473 (concat
3474 "Your `python-shell-interpreter' doesn't seem to "
3475 "support readline, yet `python-shell-completion-native' "
3476 (format "was t and %S is not part of the "
3477 (file-name-nondirectory python-shell-interpreter))
3478 "`python-shell-completion-native-disabled-interpreters' "
3479 "list. Native completions have been disabled locally. "))
3480 (python-shell-completion-native-turn-off msg))))))
3482 (defun python-shell-completion-native-turn-on-maybe-with-msg ()
3483 "Like `python-shell-completion-native-turn-on-maybe' but force messages."
3484 (python-shell-completion-native-turn-on-maybe t))
3486 (add-hook 'python-shell-first-prompt-hook
3487 #'python-shell-completion-native-turn-on-maybe-with-msg)
3489 (defun python-shell-completion-native-toggle (&optional msg)
3490 "Toggle shell native completion.
3491 With argument MSG show activation/deactivation message."
3492 (interactive "p")
3493 (python-shell-with-shell-buffer
3494 (if python-shell-completion-native-enable
3495 (python-shell-completion-native-turn-off msg)
3496 (python-shell-completion-native-turn-on msg))
3497 python-shell-completion-native-enable))
3499 (defun python-shell-completion-native-get-completions (process import input)
3500 "Get completions using native readline for PROCESS.
3501 When IMPORT is non-nil takes precedence over INPUT for
3502 completion."
3503 (with-current-buffer (process-buffer process)
3504 (let* ((input (or import input))
3505 (original-filter-fn (process-filter process))
3506 (redirect-buffer (get-buffer-create
3507 python-shell-completion-native-redirect-buffer))
3508 (trigger "\t")
3509 (new-input (concat input trigger))
3510 (input-length
3511 (save-excursion
3512 (+ (- (point-max) (comint-bol)) (length new-input))))
3513 (delete-line-command (make-string input-length ?\b))
3514 (input-to-send (concat new-input delete-line-command)))
3515 ;; Ensure restoring the process filter, even if the user quits
3516 ;; or there's some other error.
3517 (unwind-protect
3518 (with-current-buffer redirect-buffer
3519 ;; Cleanup the redirect buffer
3520 (erase-buffer)
3521 ;; Mimic `comint-redirect-send-command', unfortunately it
3522 ;; can't be used here because it expects a newline in the
3523 ;; command and that's exactly what we are trying to avoid.
3524 (let ((comint-redirect-echo-input nil)
3525 (comint-redirect-completed nil)
3526 (comint-redirect-perform-sanity-check nil)
3527 (comint-redirect-insert-matching-regexp t)
3528 (comint-redirect-finished-regexp
3529 "1__dummy_completion__[[:space:]]*\n")
3530 (comint-redirect-output-buffer redirect-buffer))
3531 ;; Compatibility with Emacs 24.x. Comint changed and
3532 ;; now `comint-redirect-filter' gets 3 args. This
3533 ;; checks which version of `comint-redirect-filter' is
3534 ;; in use based on its args and uses `apply-partially'
3535 ;; to make it up for the 3 args case.
3536 (if (= (length
3537 (help-function-arglist 'comint-redirect-filter)) 3)
3538 (set-process-filter
3539 process (apply-partially
3540 #'comint-redirect-filter original-filter-fn))
3541 (set-process-filter process #'comint-redirect-filter))
3542 (process-send-string process input-to-send)
3543 ;; Grab output until our dummy completion used as
3544 ;; output end marker is found.
3545 (when (python-shell-accept-process-output
3546 process python-shell-completion-native-output-timeout
3547 comint-redirect-finished-regexp)
3548 (re-search-backward "0__dummy_completion__" nil t)
3549 (cl-remove-duplicates
3550 (split-string
3551 (buffer-substring-no-properties
3552 (line-beginning-position) (point-min))
3553 "[ \f\t\n\r\v()]+" t)
3554 :test #'string=))))
3555 (set-process-filter process original-filter-fn)))))
3557 (defun python-shell-completion-get-completions (process import input)
3558 "Do completion at point using PROCESS for IMPORT or INPUT.
3559 When IMPORT is non-nil takes precedence over INPUT for
3560 completion."
3561 (setq input (or import input))
3562 (with-current-buffer (process-buffer process)
3563 (let ((completions
3564 (python-util-strip-string
3565 (python-shell-send-string-no-output
3566 (format
3567 (concat python-shell-completion-setup-code
3568 "\nprint (" python-shell-completion-string-code ")")
3569 input) process))))
3570 (when (> (length completions) 2)
3571 (split-string completions
3572 "^'\\|^\"\\|;\\|'$\\|\"$" t)))))
3574 (defun python-shell-completion-at-point (&optional process)
3575 "Function for `completion-at-point-functions' in `inferior-python-mode'.
3576 Optional argument PROCESS forces completions to be retrieved
3577 using that one instead of current buffer's process."
3578 (setq process (or process (get-buffer-process (current-buffer))))
3579 (let* ((line-start (if (derived-mode-p 'inferior-python-mode)
3580 ;; Working on a shell buffer: use prompt end.
3581 (cdr (python-util-comint-last-prompt))
3582 (line-beginning-position)))
3583 (import-statement
3584 (when (string-match-p
3585 (rx (* space) word-start (or "from" "import") word-end space)
3586 (buffer-substring-no-properties line-start (point)))
3587 (buffer-substring-no-properties line-start (point))))
3588 (start
3589 (save-excursion
3590 (if (not (re-search-backward
3591 (python-rx
3592 (or whitespace open-paren close-paren string-delimiter))
3593 line-start
3594 t 1))
3595 line-start
3596 (forward-char (length (match-string-no-properties 0)))
3597 (point))))
3598 (end (point))
3599 (prompt-boundaries
3600 (with-current-buffer (process-buffer process)
3601 (python-util-comint-last-prompt)))
3602 (prompt
3603 (with-current-buffer (process-buffer process)
3604 (when prompt-boundaries
3605 (buffer-substring-no-properties
3606 (car prompt-boundaries) (cdr prompt-boundaries)))))
3607 (completion-fn
3608 (with-current-buffer (process-buffer process)
3609 (cond ((or (null prompt)
3610 (< (point) (cdr prompt-boundaries)))
3611 #'ignore)
3612 ((or (not python-shell-completion-native-enable)
3613 ;; Even if native completion is enabled, for
3614 ;; pdb interaction always use the fallback
3615 ;; mechanism since the completer is changed.
3616 ;; Also, since pdb interaction is single-line
3617 ;; based, this is enough.
3618 (string-match-p python-shell-prompt-pdb-regexp prompt))
3619 #'python-shell-completion-get-completions)
3620 (t #'python-shell-completion-native-get-completions)))))
3621 (list start end
3622 (completion-table-dynamic
3623 (apply-partially
3624 completion-fn
3625 process import-statement)))))
3627 (define-obsolete-function-alias
3628 'python-shell-completion-complete-at-point
3629 'python-shell-completion-at-point
3630 "25.1")
3632 (defun python-shell-completion-complete-or-indent ()
3633 "Complete or indent depending on the context.
3634 If content before pointer is all whitespace, indent.
3635 If not try to complete."
3636 (interactive)
3637 (if (string-match "^[[:space:]]*$"
3638 (buffer-substring (comint-line-beginning-position)
3639 (point)))
3640 (indent-for-tab-command)
3641 (completion-at-point)))
3644 ;;; PDB Track integration
3646 (defcustom python-pdbtrack-activate t
3647 "Non-nil makes Python shell enable pdbtracking."
3648 :type 'boolean
3649 :group 'python
3650 :safe 'booleanp)
3652 (defcustom python-pdbtrack-stacktrace-info-regexp
3653 "> \\([^\"(<]+\\)(\\([0-9]+\\))\\([?a-zA-Z0-9_<>]+\\)()"
3654 "Regular expression matching stacktrace information.
3655 Used to extract the current line and module being inspected."
3656 :type 'string
3657 :group 'python
3658 :safe 'stringp)
3660 (defvar python-pdbtrack-tracked-buffer nil
3661 "Variable containing the value of the current tracked buffer.
3662 Never set this variable directly, use
3663 `python-pdbtrack-set-tracked-buffer' instead.")
3665 (defvar python-pdbtrack-buffers-to-kill nil
3666 "List of buffers to be deleted after tracking finishes.")
3668 (defun python-pdbtrack-set-tracked-buffer (file-name)
3669 "Set the buffer for FILE-NAME as the tracked buffer.
3670 Internally it uses the `python-pdbtrack-tracked-buffer' variable.
3671 Returns the tracked buffer."
3672 (let* ((file-name-prospect (concat (file-remote-p default-directory)
3673 file-name))
3674 (file-buffer (get-file-buffer file-name-prospect)))
3675 (if file-buffer
3676 (setq python-pdbtrack-tracked-buffer file-buffer)
3677 (cond
3678 ((file-exists-p file-name-prospect)
3679 (setq file-buffer (find-file-noselect file-name-prospect)))
3680 ((and (not (equal file-name file-name-prospect))
3681 (file-exists-p file-name))
3682 ;; Fallback to a locally available copy of the file.
3683 (setq file-buffer (find-file-noselect file-name-prospect))))
3684 (when (not (member file-buffer python-pdbtrack-buffers-to-kill))
3685 (add-to-list 'python-pdbtrack-buffers-to-kill file-buffer)))
3686 file-buffer))
3688 (defun python-pdbtrack-comint-output-filter-function (output)
3689 "Move overlay arrow to current pdb line in tracked buffer.
3690 Argument OUTPUT is a string with the output from the comint process."
3691 (when (and python-pdbtrack-activate (not (string= output "")))
3692 (let* ((full-output (ansi-color-filter-apply
3693 (buffer-substring comint-last-input-end (point-max))))
3694 (line-number)
3695 (file-name
3696 (with-temp-buffer
3697 (insert full-output)
3698 ;; When the debugger encounters a pdb.set_trace()
3699 ;; command, it prints a single stack frame. Sometimes
3700 ;; it prints a bit of extra information about the
3701 ;; arguments of the present function. When ipdb
3702 ;; encounters an exception, it prints the _entire_ stack
3703 ;; trace. To handle all of these cases, we want to find
3704 ;; the _last_ stack frame printed in the most recent
3705 ;; batch of output, then jump to the corresponding
3706 ;; file/line number.
3707 (goto-char (point-max))
3708 (when (re-search-backward python-pdbtrack-stacktrace-info-regexp nil t)
3709 (setq line-number (string-to-number
3710 (match-string-no-properties 2)))
3711 (match-string-no-properties 1)))))
3712 (if (and file-name line-number)
3713 (let* ((tracked-buffer
3714 (python-pdbtrack-set-tracked-buffer file-name))
3715 (shell-buffer (current-buffer))
3716 (tracked-buffer-window (get-buffer-window tracked-buffer))
3717 (tracked-buffer-line-pos))
3718 (with-current-buffer tracked-buffer
3719 (set (make-local-variable 'overlay-arrow-string) "=>")
3720 (set (make-local-variable 'overlay-arrow-position) (make-marker))
3721 (setq tracked-buffer-line-pos (progn
3722 (goto-char (point-min))
3723 (forward-line (1- line-number))
3724 (point-marker)))
3725 (when tracked-buffer-window
3726 (set-window-point
3727 tracked-buffer-window tracked-buffer-line-pos))
3728 (set-marker overlay-arrow-position tracked-buffer-line-pos))
3729 (pop-to-buffer tracked-buffer)
3730 (switch-to-buffer-other-window shell-buffer))
3731 (when python-pdbtrack-tracked-buffer
3732 (with-current-buffer python-pdbtrack-tracked-buffer
3733 (set-marker overlay-arrow-position nil))
3734 (mapc #'(lambda (buffer)
3735 (ignore-errors (kill-buffer buffer)))
3736 python-pdbtrack-buffers-to-kill)
3737 (setq python-pdbtrack-tracked-buffer nil
3738 python-pdbtrack-buffers-to-kill nil)))))
3739 output)
3742 ;;; Symbol completion
3744 (defun python-completion-at-point ()
3745 "Function for `completion-at-point-functions' in `python-mode'.
3746 For this to work as best as possible you should call
3747 `python-shell-send-buffer' from time to time so context in
3748 inferior Python process is updated properly."
3749 (let ((process (python-shell-get-process)))
3750 (when process
3751 (python-shell-completion-at-point process))))
3753 (define-obsolete-function-alias
3754 'python-completion-complete-at-point
3755 'python-completion-at-point
3756 "25.1")
3759 ;;; Fill paragraph
3761 (defcustom python-fill-comment-function 'python-fill-comment
3762 "Function to fill comments.
3763 This is the function used by `python-fill-paragraph' to
3764 fill comments."
3765 :type 'symbol
3766 :group 'python)
3768 (defcustom python-fill-string-function 'python-fill-string
3769 "Function to fill strings.
3770 This is the function used by `python-fill-paragraph' to
3771 fill strings."
3772 :type 'symbol
3773 :group 'python)
3775 (defcustom python-fill-decorator-function 'python-fill-decorator
3776 "Function to fill decorators.
3777 This is the function used by `python-fill-paragraph' to
3778 fill decorators."
3779 :type 'symbol
3780 :group 'python)
3782 (defcustom python-fill-paren-function 'python-fill-paren
3783 "Function to fill parens.
3784 This is the function used by `python-fill-paragraph' to
3785 fill parens."
3786 :type 'symbol
3787 :group 'python)
3789 (defcustom python-fill-docstring-style 'pep-257
3790 "Style used to fill docstrings.
3791 This affects `python-fill-string' behavior with regards to
3792 triple quotes positioning.
3794 Possible values are `django', `onetwo', `pep-257', `pep-257-nn',
3795 `symmetric', and nil. A value of nil won't care about quotes
3796 position and will treat docstrings a normal string, any other
3797 value may result in one of the following docstring styles:
3799 `django':
3801 \"\"\"
3802 Process foo, return bar.
3803 \"\"\"
3805 \"\"\"
3806 Process foo, return bar.
3808 If processing fails throw ProcessingError.
3809 \"\"\"
3811 `onetwo':
3813 \"\"\"Process foo, return bar.\"\"\"
3815 \"\"\"
3816 Process foo, return bar.
3818 If processing fails throw ProcessingError.
3820 \"\"\"
3822 `pep-257':
3824 \"\"\"Process foo, return bar.\"\"\"
3826 \"\"\"Process foo, return bar.
3828 If processing fails throw ProcessingError.
3830 \"\"\"
3832 `pep-257-nn':
3834 \"\"\"Process foo, return bar.\"\"\"
3836 \"\"\"Process foo, return bar.
3838 If processing fails throw ProcessingError.
3839 \"\"\"
3841 `symmetric':
3843 \"\"\"Process foo, return bar.\"\"\"
3845 \"\"\"
3846 Process foo, return bar.
3848 If processing fails throw ProcessingError.
3849 \"\"\""
3850 :type '(choice
3851 (const :tag "Don't format docstrings" nil)
3852 (const :tag "Django's coding standards style." django)
3853 (const :tag "One newline and start and Two at end style." onetwo)
3854 (const :tag "PEP-257 with 2 newlines at end of string." pep-257)
3855 (const :tag "PEP-257 with 1 newline at end of string." pep-257-nn)
3856 (const :tag "Symmetric style." symmetric))
3857 :group 'python
3858 :safe (lambda (val)
3859 (memq val '(django onetwo pep-257 pep-257-nn symmetric nil))))
3861 (defun python-fill-paragraph (&optional justify)
3862 "`fill-paragraph-function' handling multi-line strings and possibly comments.
3863 If any of the current line is in or at the end of a multi-line string,
3864 fill the string or the paragraph of it that point is in, preserving
3865 the string's indentation.
3866 Optional argument JUSTIFY defines if the paragraph should be justified."
3867 (interactive "P")
3868 (save-excursion
3869 (cond
3870 ;; Comments
3871 ((python-syntax-context 'comment)
3872 (funcall python-fill-comment-function justify))
3873 ;; Strings/Docstrings
3874 ((save-excursion (or (python-syntax-context 'string)
3875 (equal (string-to-syntax "|")
3876 (syntax-after (point)))))
3877 (funcall python-fill-string-function justify))
3878 ;; Decorators
3879 ((equal (char-after (save-excursion
3880 (python-nav-beginning-of-statement))) ?@)
3881 (funcall python-fill-decorator-function justify))
3882 ;; Parens
3883 ((or (python-syntax-context 'paren)
3884 (looking-at (python-rx open-paren))
3885 (save-excursion
3886 (skip-syntax-forward "^(" (line-end-position))
3887 (looking-at (python-rx open-paren))))
3888 (funcall python-fill-paren-function justify))
3889 (t t))))
3891 (defun python-fill-comment (&optional justify)
3892 "Comment fill function for `python-fill-paragraph'.
3893 JUSTIFY should be used (if applicable) as in `fill-paragraph'."
3894 (fill-comment-paragraph justify))
3896 (defun python-fill-string (&optional justify)
3897 "String fill function for `python-fill-paragraph'.
3898 JUSTIFY should be used (if applicable) as in `fill-paragraph'."
3899 (let* ((str-start-pos
3900 (set-marker
3901 (make-marker)
3902 (or (python-syntax-context 'string)
3903 (and (equal (string-to-syntax "|")
3904 (syntax-after (point)))
3905 (point)))))
3906 (num-quotes (python-syntax-count-quotes
3907 (char-after str-start-pos) str-start-pos))
3908 (str-end-pos
3909 (save-excursion
3910 (goto-char (+ str-start-pos num-quotes))
3911 (or (re-search-forward (rx (syntax string-delimiter)) nil t)
3912 (goto-char (point-max)))
3913 (point-marker)))
3914 (multi-line-p
3915 ;; Docstring styles may vary for oneliners and multi-liners.
3916 (> (count-matches "\n" str-start-pos str-end-pos) 0))
3917 (delimiters-style
3918 (pcase python-fill-docstring-style
3919 ;; delimiters-style is a cons cell with the form
3920 ;; (START-NEWLINES . END-NEWLINES). When any of the sexps
3921 ;; is NIL means to not add any newlines for start or end
3922 ;; of docstring. See `python-fill-docstring-style' for a
3923 ;; graphic idea of each style.
3924 (`django (cons 1 1))
3925 (`onetwo (and multi-line-p (cons 1 2)))
3926 (`pep-257 (and multi-line-p (cons nil 2)))
3927 (`pep-257-nn (and multi-line-p (cons nil 1)))
3928 (`symmetric (and multi-line-p (cons 1 1)))))
3929 (fill-paragraph-function))
3930 (save-restriction
3931 (narrow-to-region str-start-pos str-end-pos)
3932 (fill-paragraph justify))
3933 (save-excursion
3934 (when (and (python-info-docstring-p) python-fill-docstring-style)
3935 ;; Add the number of newlines indicated by the selected style
3936 ;; at the start of the docstring.
3937 (goto-char (+ str-start-pos num-quotes))
3938 (delete-region (point) (progn
3939 (skip-syntax-forward "> ")
3940 (point)))
3941 (and (car delimiters-style)
3942 (or (newline (car delimiters-style)) t)
3943 ;; Indent only if a newline is added.
3944 (indent-according-to-mode))
3945 ;; Add the number of newlines indicated by the selected style
3946 ;; at the end of the docstring.
3947 (goto-char (if (not (= str-end-pos (point-max)))
3948 (- str-end-pos num-quotes)
3949 str-end-pos))
3950 (delete-region (point) (progn
3951 (skip-syntax-backward "> ")
3952 (point)))
3953 (and (cdr delimiters-style)
3954 ;; Add newlines only if string ends.
3955 (not (= str-end-pos (point-max)))
3956 (or (newline (cdr delimiters-style)) t)
3957 ;; Again indent only if a newline is added.
3958 (indent-according-to-mode))))) t)
3960 (defun python-fill-decorator (&optional _justify)
3961 "Decorator fill function for `python-fill-paragraph'.
3962 JUSTIFY should be used (if applicable) as in `fill-paragraph'."
3965 (defun python-fill-paren (&optional justify)
3966 "Paren fill function for `python-fill-paragraph'.
3967 JUSTIFY should be used (if applicable) as in `fill-paragraph'."
3968 (save-restriction
3969 (narrow-to-region (progn
3970 (while (python-syntax-context 'paren)
3971 (goto-char (1- (point))))
3972 (line-beginning-position))
3973 (progn
3974 (when (not (python-syntax-context 'paren))
3975 (end-of-line)
3976 (when (not (python-syntax-context 'paren))
3977 (skip-syntax-backward "^)")))
3978 (while (and (python-syntax-context 'paren)
3979 (not (eobp)))
3980 (goto-char (1+ (point))))
3981 (point)))
3982 (let ((paragraph-start "\f\\|[ \t]*$")
3983 (paragraph-separate ",")
3984 (fill-paragraph-function))
3985 (goto-char (point-min))
3986 (fill-paragraph justify))
3987 (while (not (eobp))
3988 (forward-line 1)
3989 (python-indent-line)
3990 (goto-char (line-end-position))))
3994 ;;; Skeletons
3996 (defcustom python-skeleton-autoinsert nil
3997 "Non-nil means template skeletons will be automagically inserted.
3998 This happens when pressing \"if<SPACE>\", for example, to prompt for
3999 the if condition."
4000 :type 'boolean
4001 :group 'python
4002 :safe 'booleanp)
4004 (define-obsolete-variable-alias
4005 'python-use-skeletons 'python-skeleton-autoinsert "24.3")
4007 (defvar python-skeleton-available '()
4008 "Internal list of available skeletons.")
4010 (define-abbrev-table 'python-mode-skeleton-abbrev-table ()
4011 "Abbrev table for Python mode skeletons."
4012 :case-fixed t
4013 ;; Allow / inside abbrevs.
4014 :regexp "\\(?:^\\|[^/]\\)\\<\\([[:word:]/]+\\)\\W*"
4015 ;; Only expand in code.
4016 :enable-function (lambda ()
4017 (and
4018 (not (python-syntax-comment-or-string-p))
4019 python-skeleton-autoinsert)))
4021 (defmacro python-skeleton-define (name doc &rest skel)
4022 "Define a `python-mode' skeleton using NAME DOC and SKEL.
4023 The skeleton will be bound to python-skeleton-NAME and will
4024 be added to `python-mode-skeleton-abbrev-table'."
4025 (declare (indent 2))
4026 (let* ((name (symbol-name name))
4027 (function-name (intern (concat "python-skeleton-" name))))
4028 `(progn
4029 (define-abbrev python-mode-skeleton-abbrev-table
4030 ,name "" ',function-name :system t)
4031 (setq python-skeleton-available
4032 (cons ',function-name python-skeleton-available))
4033 (define-skeleton ,function-name
4034 ,(or doc
4035 (format "Insert %s statement." name))
4036 ,@skel))))
4038 (define-abbrev-table 'python-mode-abbrev-table ()
4039 "Abbrev table for Python mode."
4040 :parents (list python-mode-skeleton-abbrev-table))
4042 (defmacro python-define-auxiliary-skeleton (name &optional doc &rest skel)
4043 "Define a `python-mode' auxiliary skeleton using NAME DOC and SKEL.
4044 The skeleton will be bound to python-skeleton-NAME."
4045 (declare (indent 2))
4046 (let* ((name (symbol-name name))
4047 (function-name (intern (concat "python-skeleton--" name)))
4048 (msg (funcall (if (fboundp 'format-message) #'format-message #'format)
4049 "Add `%s' clause? " name)))
4050 (when (not skel)
4051 (setq skel
4052 `(< ,(format "%s:" name) \n \n
4053 > _ \n)))
4054 `(define-skeleton ,function-name
4055 ,(or doc
4056 (format "Auxiliary skeleton for %s statement." name))
4058 (unless (y-or-n-p ,msg)
4059 (signal 'quit t))
4060 ,@skel)))
4062 (python-define-auxiliary-skeleton else)
4064 (python-define-auxiliary-skeleton except)
4066 (python-define-auxiliary-skeleton finally)
4068 (python-skeleton-define if nil
4069 "Condition: "
4070 "if " str ":" \n
4071 _ \n
4072 ("other condition, %s: "
4074 "elif " str ":" \n
4075 > _ \n nil)
4076 '(python-skeleton--else) | ^)
4078 (python-skeleton-define while nil
4079 "Condition: "
4080 "while " str ":" \n
4081 > _ \n
4082 '(python-skeleton--else) | ^)
4084 (python-skeleton-define for nil
4085 "Iteration spec: "
4086 "for " str ":" \n
4087 > _ \n
4088 '(python-skeleton--else) | ^)
4090 (python-skeleton-define import nil
4091 "Import from module: "
4092 "from " str & " " | -5
4093 "import "
4094 ("Identifier: " str ", ") -2 \n _)
4096 (python-skeleton-define try nil
4098 "try:" \n
4099 > _ \n
4100 ("Exception, %s: "
4102 "except " str ":" \n
4103 > _ \n nil)
4104 resume:
4105 '(python-skeleton--except)
4106 '(python-skeleton--else)
4107 '(python-skeleton--finally) | ^)
4109 (python-skeleton-define def nil
4110 "Function name: "
4111 "def " str "(" ("Parameter, %s: "
4112 (unless (equal ?\( (char-before)) ", ")
4113 str) "):" \n
4114 "\"\"\"" - "\"\"\"" \n
4115 > _ \n)
4117 (python-skeleton-define class nil
4118 "Class name: "
4119 "class " str "(" ("Inheritance, %s: "
4120 (unless (equal ?\( (char-before)) ", ")
4121 str)
4122 & ")" | -1
4123 ":" \n
4124 "\"\"\"" - "\"\"\"" \n
4125 > _ \n)
4127 (defun python-skeleton-add-menu-items ()
4128 "Add menu items to Python->Skeletons menu."
4129 (let ((skeletons (sort python-skeleton-available 'string<)))
4130 (dolist (skeleton skeletons)
4131 (easy-menu-add-item
4132 nil '("Python" "Skeletons")
4133 `[,(format
4134 "Insert %s" (nth 2 (split-string (symbol-name skeleton) "-")))
4135 ,skeleton t]))))
4137 ;;; FFAP
4139 (defcustom python-ffap-setup-code
4141 def __FFAP_get_module_path(objstr):
4142 try:
4143 import inspect
4144 import os.path
4145 # NameError exceptions are delayed until this point.
4146 obj = eval(objstr)
4147 module = inspect.getmodule(obj)
4148 filename = module.__file__
4149 ext = os.path.splitext(filename)[1]
4150 if ext in ('.pyc', '.pyo'):
4151 # Point to the source file.
4152 filename = filename[:-1]
4153 if os.path.exists(filename):
4154 return filename
4155 return ''
4156 except:
4157 return ''"
4158 "Python code to get a module path."
4159 :type 'string
4160 :group 'python)
4162 (defcustom python-ffap-string-code
4163 "__FFAP_get_module_path('''%s''')"
4164 "Python code used to get a string with the path of a module."
4165 :type 'string
4166 :group 'python)
4168 (defun python-ffap-module-path (module)
4169 "Function for `ffap-alist' to return path for MODULE."
4170 (let ((process (or
4171 (and (derived-mode-p 'inferior-python-mode)
4172 (get-buffer-process (current-buffer)))
4173 (python-shell-get-process))))
4174 (if (not process)
4176 (let ((module-file
4177 (python-shell-send-string-no-output
4178 (concat
4179 python-ffap-setup-code
4180 "\nprint (" (format python-ffap-string-code module) ")")
4181 process)))
4182 (unless (zerop (length module-file))
4183 (python-util-strip-string module-file))))))
4185 (defvar ffap-alist)
4187 (eval-after-load "ffap"
4188 '(progn
4189 (push '(python-mode . python-ffap-module-path) ffap-alist)
4190 (push '(inferior-python-mode . python-ffap-module-path) ffap-alist)))
4193 ;;; Code check
4195 (defcustom python-check-command
4196 (or (executable-find "pyflakes")
4197 (executable-find "epylint")
4198 "install pyflakes, pylint or something else")
4199 "Command used to check a Python file."
4200 :type 'string
4201 :group 'python)
4203 (defcustom python-check-buffer-name
4204 "*Python check: %s*"
4205 "Buffer name used for check commands."
4206 :type 'string
4207 :group 'python)
4209 (defvar python-check-custom-command nil
4210 "Internal use.")
4211 ;; XXX: Avoid `defvar-local' for compat with Emacs<24.3
4212 (make-variable-buffer-local 'python-check-custom-command)
4214 (defun python-check (command)
4215 "Check a Python file (default current buffer's file).
4216 Runs COMMAND, a shell command, as if by `compile'.
4217 See `python-check-command' for the default."
4218 (interactive
4219 (list (read-string "Check command: "
4220 (or python-check-custom-command
4221 (concat python-check-command " "
4222 (shell-quote-argument
4224 (let ((name (buffer-file-name)))
4225 (and name
4226 (file-name-nondirectory name)))
4227 "")))))))
4228 (setq python-check-custom-command command)
4229 (save-some-buffers (not compilation-ask-about-save) nil)
4230 (python-shell-with-environment
4231 (compilation-start command nil
4232 (lambda (_modename)
4233 (format python-check-buffer-name command)))))
4236 ;;; Eldoc
4238 (defcustom python-eldoc-setup-code
4239 "def __PYDOC_get_help(obj):
4240 try:
4241 import inspect
4242 try:
4243 str_type = basestring
4244 except NameError:
4245 str_type = str
4246 if isinstance(obj, str_type):
4247 obj = eval(obj, globals())
4248 doc = inspect.getdoc(obj)
4249 if not doc and callable(obj):
4250 target = None
4251 if inspect.isclass(obj) and hasattr(obj, '__init__'):
4252 target = obj.__init__
4253 objtype = 'class'
4254 else:
4255 target = obj
4256 objtype = 'def'
4257 if target:
4258 args = inspect.formatargspec(
4259 *inspect.getargspec(target)
4261 name = obj.__name__
4262 doc = '{objtype} {name}{args}'.format(
4263 objtype=objtype, name=name, args=args
4265 else:
4266 doc = doc.splitlines()[0]
4267 except:
4268 doc = ''
4269 return doc"
4270 "Python code to setup documentation retrieval."
4271 :type 'string
4272 :group 'python)
4274 (defcustom python-eldoc-string-code
4275 "__PYDOC_get_help('''%s''')"
4276 "Python code used to get a string with the documentation of an object."
4277 :type 'string
4278 :group 'python)
4280 (defun python-eldoc--get-symbol-at-point ()
4281 "Get the current symbol for eldoc.
4282 Returns the current symbol handling point within arguments."
4283 (save-excursion
4284 (let ((start (python-syntax-context 'paren)))
4285 (when start
4286 (goto-char start))
4287 (when (or start
4288 (eobp)
4289 (memq (char-syntax (char-after)) '(?\ ?-)))
4290 ;; Try to adjust to closest symbol if not in one.
4291 (python-util-forward-comment -1)))
4292 (python-info-current-symbol t)))
4294 (defun python-eldoc--get-doc-at-point (&optional force-input force-process)
4295 "Internal implementation to get documentation at point.
4296 If not FORCE-INPUT is passed then what `python-eldoc--get-symbol-at-point'
4297 returns will be used. If not FORCE-PROCESS is passed what
4298 `python-shell-get-process' returns is used."
4299 (let ((process (or force-process (python-shell-get-process))))
4300 (when process
4301 (let* ((input (or force-input
4302 (python-eldoc--get-symbol-at-point)))
4303 (docstring
4304 (when input
4305 ;; Prevent resizing the echo area when iPython is
4306 ;; enabled. Bug#18794.
4307 (python-util-strip-string
4308 (python-shell-send-string-no-output
4309 (concat
4310 python-eldoc-setup-code
4311 "\nprint(" (format python-eldoc-string-code input) ")")
4312 process)))))
4313 (unless (zerop (length docstring))
4314 docstring)))))
4316 (defvar-local python-eldoc-get-doc t
4317 "Non-nil means eldoc should fetch the documentation
4318 automatically. Set to nil by `python-eldoc-function' if
4319 `python-eldoc-function-timeout-permanent' is non-nil and
4320 `python-eldoc-function' times out.")
4322 (defcustom python-eldoc-function-timeout 1
4323 "Timeout for `python-eldoc-function' in seconds."
4324 :group 'python
4325 :type 'integer
4326 :version "25.1")
4328 (defcustom python-eldoc-function-timeout-permanent t
4329 "Non-nil means that when `python-eldoc-function' times out
4330 `python-eldoc-get-doc' will be set to nil"
4331 :group 'python
4332 :type 'boolean
4333 :version "25.1")
4335 (defun python-eldoc-function ()
4336 "`eldoc-documentation-function' for Python.
4337 For this to work as best as possible you should call
4338 `python-shell-send-buffer' from time to time so context in
4339 inferior Python process is updated properly.
4341 If `python-eldoc-function-timeout' seconds elapse before this
4342 function returns then if
4343 `python-eldoc-function-timeout-permanent' is non-nil
4344 `python-eldoc-get-doc' will be set to nil and eldoc will no
4345 longer return the documentation at the point automatically.
4347 Set `python-eldoc-get-doc' to t to reenable eldoc documentation
4348 fetching"
4349 (when python-eldoc-get-doc
4350 (with-timeout (python-eldoc-function-timeout
4351 (if python-eldoc-function-timeout-permanent
4352 (progn
4353 (message "Eldoc echo-area display muted in this buffer, see `python-eldoc-function'")
4354 (setq python-eldoc-get-doc nil))
4355 (message "`python-eldoc-function' timed out, see `python-eldoc-function-timeout'")))
4356 (python-eldoc--get-doc-at-point))))
4358 (defun python-eldoc-at-point (symbol)
4359 "Get help on SYMBOL using `help'.
4360 Interactively, prompt for symbol."
4361 (interactive
4362 (let ((symbol (python-eldoc--get-symbol-at-point))
4363 (enable-recursive-minibuffers t))
4364 (list (read-string (if symbol
4365 (format "Describe symbol (default %s): " symbol)
4366 "Describe symbol: ")
4367 nil nil symbol))))
4368 (message (python-eldoc--get-doc-at-point symbol)))
4370 (defun python-describe-at-point (symbol process)
4371 (interactive (list (python-info-current-symbol)
4372 (python-shell-get-process)))
4373 (comint-send-string process (concat "help('" symbol "')\n")))
4376 ;;; Hideshow
4378 (defun python-hideshow-forward-sexp-function (arg)
4379 "Python specific `forward-sexp' function for `hs-minor-mode'.
4380 Argument ARG is ignored."
4381 arg ; Shut up, byte compiler.
4382 (python-nav-end-of-defun)
4383 (unless (python-info-current-line-empty-p)
4384 (backward-char)))
4387 ;;; Imenu
4389 (defvar python-imenu-format-item-label-function
4390 'python-imenu-format-item-label
4391 "Imenu function used to format an item label.
4392 It must be a function with two arguments: TYPE and NAME.")
4394 (defvar python-imenu-format-parent-item-label-function
4395 'python-imenu-format-parent-item-label
4396 "Imenu function used to format a parent item label.
4397 It must be a function with two arguments: TYPE and NAME.")
4399 (defvar python-imenu-format-parent-item-jump-label-function
4400 'python-imenu-format-parent-item-jump-label
4401 "Imenu function used to format a parent jump item label.
4402 It must be a function with two arguments: TYPE and NAME.")
4404 (defun python-imenu-format-item-label (type name)
4405 "Return Imenu label for single node using TYPE and NAME."
4406 (format "%s (%s)" name type))
4408 (defun python-imenu-format-parent-item-label (type name)
4409 "Return Imenu label for parent node using TYPE and NAME."
4410 (format "%s..." (python-imenu-format-item-label type name)))
4412 (defun python-imenu-format-parent-item-jump-label (type _name)
4413 "Return Imenu label for parent node jump using TYPE and NAME."
4414 (if (string= type "class")
4415 "*class definition*"
4416 "*function definition*"))
4418 (defun python-imenu--get-defun-type-name ()
4419 "Return defun type and name at current position."
4420 (when (looking-at python-nav-beginning-of-defun-regexp)
4421 (let ((split (split-string (match-string-no-properties 0))))
4422 (if (= (length split) 2)
4423 split
4424 (list (concat (car split) " " (cadr split))
4425 (car (last split)))))))
4427 (defun python-imenu--put-parent (type name pos tree)
4428 "Add the parent with TYPE, NAME and POS to TREE."
4429 (let ((label
4430 (funcall python-imenu-format-item-label-function type name))
4431 (jump-label
4432 (funcall python-imenu-format-parent-item-jump-label-function type name)))
4433 (if (not tree)
4434 (cons label pos)
4435 (cons label (cons (cons jump-label pos) tree)))))
4437 (defun python-imenu--build-tree (&optional min-indent prev-indent tree)
4438 "Recursively build the tree of nested definitions of a node.
4439 Arguments MIN-INDENT, PREV-INDENT and TREE are internal and should
4440 not be passed explicitly unless you know what you are doing."
4441 (setq min-indent (or min-indent 0)
4442 prev-indent (or prev-indent python-indent-offset))
4443 (let* ((pos (python-nav-backward-defun))
4444 (defun-type-name (and pos (python-imenu--get-defun-type-name)))
4445 (type (car defun-type-name))
4446 (name (cadr defun-type-name))
4447 (label (when name
4448 (funcall python-imenu-format-item-label-function type name)))
4449 (indent (current-indentation))
4450 (children-indent-limit (+ python-indent-offset min-indent)))
4451 (cond ((not pos)
4452 ;; Nothing found, probably near to bobp.
4453 nil)
4454 ((<= indent min-indent)
4455 ;; The current indentation points that this is a parent
4456 ;; node, add it to the tree and stop recursing.
4457 (python-imenu--put-parent type name pos tree))
4459 (python-imenu--build-tree
4460 min-indent
4461 indent
4462 (if (<= indent children-indent-limit)
4463 ;; This lies within the children indent offset range,
4464 ;; so it's a normal child of its parent (i.e., not
4465 ;; a child of a child).
4466 (cons (cons label pos) tree)
4467 ;; Oh no, a child of a child?! Fear not, we
4468 ;; know how to roll. We recursively parse these by
4469 ;; swapping prev-indent and min-indent plus adding this
4470 ;; newly found item to a fresh subtree. This works, I
4471 ;; promise.
4472 (cons
4473 (python-imenu--build-tree
4474 prev-indent indent (list (cons label pos)))
4475 tree)))))))
4477 (defun python-imenu-create-index ()
4478 "Return tree Imenu alist for the current Python buffer.
4479 Change `python-imenu-format-item-label-function',
4480 `python-imenu-format-parent-item-label-function',
4481 `python-imenu-format-parent-item-jump-label-function' to
4482 customize how labels are formatted."
4483 (goto-char (point-max))
4484 (let ((index)
4485 (tree))
4486 (while (setq tree (python-imenu--build-tree))
4487 (setq index (cons tree index)))
4488 index))
4490 (defun python-imenu-create-flat-index (&optional alist prefix)
4491 "Return flat outline of the current Python buffer for Imenu.
4492 Optional argument ALIST is the tree to be flattened; when nil
4493 `python-imenu-build-index' is used with
4494 `python-imenu-format-parent-item-jump-label-function'
4495 `python-imenu-format-parent-item-label-function'
4496 `python-imenu-format-item-label-function' set to
4497 (lambda (type name) name)
4498 Optional argument PREFIX is used in recursive calls and should
4499 not be passed explicitly.
4501 Converts this:
4503 ((\"Foo\" . 103)
4504 (\"Bar\" . 138)
4505 (\"decorator\"
4506 (\"decorator\" . 173)
4507 (\"wrap\"
4508 (\"wrap\" . 353)
4509 (\"wrapped_f\" . 393))))
4511 To this:
4513 ((\"Foo\" . 103)
4514 (\"Bar\" . 138)
4515 (\"decorator\" . 173)
4516 (\"decorator.wrap\" . 353)
4517 (\"decorator.wrapped_f\" . 393))"
4518 ;; Inspired by imenu--flatten-index-alist removed in revno 21853.
4519 (apply
4520 'nconc
4521 (mapcar
4522 (lambda (item)
4523 (let ((name (if prefix
4524 (concat prefix "." (car item))
4525 (car item)))
4526 (pos (cdr item)))
4527 (cond ((or (numberp pos) (markerp pos))
4528 (list (cons name pos)))
4529 ((listp pos)
4530 (cons
4531 (cons name (cdar pos))
4532 (python-imenu-create-flat-index (cddr item) name))))))
4533 (or alist
4534 (let* ((fn (lambda (_type name) name))
4535 (python-imenu-format-item-label-function fn)
4536 (python-imenu-format-parent-item-label-function fn)
4537 (python-imenu-format-parent-item-jump-label-function fn))
4538 (python-imenu-create-index))))))
4541 ;;; Misc helpers
4543 (defun python-info-current-defun (&optional include-type)
4544 "Return name of surrounding function with Python compatible dotty syntax.
4545 Optional argument INCLUDE-TYPE indicates to include the type of the defun.
4546 This function can be used as the value of `add-log-current-defun-function'
4547 since it returns nil if point is not inside a defun."
4548 (save-restriction
4549 (prog-widen)
4550 (save-excursion
4551 (end-of-line 1)
4552 (let ((names)
4553 (starting-indentation (current-indentation))
4554 (starting-pos (point))
4555 (first-run t)
4556 (last-indent)
4557 (type))
4558 (catch 'exit
4559 (while (python-nav-beginning-of-defun 1)
4560 (when (save-match-data
4561 (and
4562 (or (not last-indent)
4563 (< (current-indentation) last-indent))
4565 (and first-run
4566 (save-excursion
4567 ;; If this is the first run, we may add
4568 ;; the current defun at point.
4569 (setq first-run nil)
4570 (goto-char starting-pos)
4571 (python-nav-beginning-of-statement)
4572 (beginning-of-line 1)
4573 (looking-at-p
4574 python-nav-beginning-of-defun-regexp)))
4575 (< starting-pos
4576 (save-excursion
4577 (let ((min-indent
4578 (+ (current-indentation)
4579 python-indent-offset)))
4580 (if (< starting-indentation min-indent)
4581 ;; If the starting indentation is not
4582 ;; within the min defun indent make the
4583 ;; check fail.
4584 starting-pos
4585 ;; Else go to the end of defun and add
4586 ;; up the current indentation to the
4587 ;; ending position.
4588 (python-nav-end-of-defun)
4589 (+ (point)
4590 (if (>= (current-indentation) min-indent)
4591 (1+ (current-indentation))
4592 0)))))))))
4593 (save-match-data (setq last-indent (current-indentation)))
4594 (if (or (not include-type) type)
4595 (setq names (cons (match-string-no-properties 1) names))
4596 (let ((match (split-string (match-string-no-properties 0))))
4597 (setq type (car match))
4598 (setq names (cons (cadr match) names)))))
4599 ;; Stop searching ASAP.
4600 (and (= (current-indentation) 0) (throw 'exit t))))
4601 (and names
4602 (concat (and type (format "%s " type))
4603 (mapconcat 'identity names ".")))))))
4605 (defun python-info-current-symbol (&optional replace-self)
4606 "Return current symbol using dotty syntax.
4607 With optional argument REPLACE-SELF convert \"self\" to current
4608 parent defun name."
4609 (let ((name
4610 (and (not (python-syntax-comment-or-string-p))
4611 (with-syntax-table python-dotty-syntax-table
4612 (let ((sym (symbol-at-point)))
4613 (and sym
4614 (substring-no-properties (symbol-name sym))))))))
4615 (when name
4616 (if (not replace-self)
4617 name
4618 (let ((current-defun (python-info-current-defun)))
4619 (if (not current-defun)
4620 name
4621 (replace-regexp-in-string
4622 (python-rx line-start word-start "self" word-end ?.)
4623 (concat
4624 (mapconcat 'identity
4625 (butlast (split-string current-defun "\\."))
4626 ".") ".")
4627 name)))))))
4629 (defun python-info-statement-starts-block-p ()
4630 "Return non-nil if current statement opens a block."
4631 (save-excursion
4632 (python-nav-beginning-of-statement)
4633 (looking-at (python-rx block-start))))
4635 (defun python-info-statement-ends-block-p ()
4636 "Return non-nil if point is at end of block."
4637 (let ((end-of-block-pos (save-excursion
4638 (python-nav-end-of-block)))
4639 (end-of-statement-pos (save-excursion
4640 (python-nav-end-of-statement))))
4641 (and end-of-block-pos end-of-statement-pos
4642 (= end-of-block-pos end-of-statement-pos))))
4644 (defun python-info-beginning-of-statement-p ()
4645 "Return non-nil if point is at beginning of statement."
4646 (= (point) (save-excursion
4647 (python-nav-beginning-of-statement)
4648 (point))))
4650 (defun python-info-end-of-statement-p ()
4651 "Return non-nil if point is at end of statement."
4652 (= (point) (save-excursion
4653 (python-nav-end-of-statement)
4654 (point))))
4656 (defun python-info-beginning-of-block-p ()
4657 "Return non-nil if point is at beginning of block."
4658 (and (python-info-beginning-of-statement-p)
4659 (python-info-statement-starts-block-p)))
4661 (defun python-info-end-of-block-p ()
4662 "Return non-nil if point is at end of block."
4663 (and (python-info-end-of-statement-p)
4664 (python-info-statement-ends-block-p)))
4666 (define-obsolete-function-alias
4667 'python-info-closing-block
4668 'python-info-dedenter-opening-block-position "24.4")
4670 (defun python-info-dedenter-opening-block-position ()
4671 "Return the point of the closest block the current line closes.
4672 Returns nil if point is not on a dedenter statement or no opening
4673 block can be detected. The latter case meaning current file is
4674 likely an invalid python file."
4675 (let ((positions (python-info-dedenter-opening-block-positions))
4676 (indentation (current-indentation))
4677 (position))
4678 (while (and (not position)
4679 positions)
4680 (save-excursion
4681 (goto-char (car positions))
4682 (if (<= (current-indentation) indentation)
4683 (setq position (car positions))
4684 (setq positions (cdr positions)))))
4685 position))
4687 (defun python-info-dedenter-opening-block-positions ()
4688 "Return points of blocks the current line may close sorted by closer.
4689 Returns nil if point is not on a dedenter statement or no opening
4690 block can be detected. The latter case meaning current file is
4691 likely an invalid python file."
4692 (save-excursion
4693 (let ((dedenter-pos (python-info-dedenter-statement-p)))
4694 (when dedenter-pos
4695 (goto-char dedenter-pos)
4696 (let* ((cur-line (line-beginning-position))
4697 (pairs '(("elif" "elif" "if")
4698 ("else" "if" "elif" "except" "for" "while")
4699 ("except" "except" "try")
4700 ("finally" "else" "except" "try")))
4701 (dedenter (match-string-no-properties 0))
4702 (possible-opening-blocks (cdr (assoc-string dedenter pairs)))
4703 (collected-indentations)
4704 (opening-blocks))
4705 (catch 'exit
4706 (while (python-nav--syntactically
4707 (lambda ()
4708 (re-search-backward (python-rx block-start) nil t))
4709 #'<)
4710 (let ((indentation (current-indentation)))
4711 (when (and (not (memq indentation collected-indentations))
4712 (or (not collected-indentations)
4713 (< indentation (apply #'min collected-indentations)))
4714 ;; There must be no line with indentation
4715 ;; smaller than `indentation' (except for
4716 ;; blank lines) between the found opening
4717 ;; block and the current line, otherwise it
4718 ;; is not an opening block.
4719 (save-excursion
4720 (forward-line)
4721 (let ((no-back-indent t))
4722 (save-match-data
4723 (while (and (< (point) cur-line)
4724 (setq no-back-indent
4725 (or (> (current-indentation) indentation)
4726 (python-info-current-line-empty-p))))
4727 (forward-line)))
4728 no-back-indent)))
4729 (setq collected-indentations
4730 (cons indentation collected-indentations))
4731 (when (member (match-string-no-properties 0)
4732 possible-opening-blocks)
4733 (setq opening-blocks (cons (point) opening-blocks))))
4734 (when (zerop indentation)
4735 (throw 'exit nil)))))
4736 ;; sort by closer
4737 (nreverse opening-blocks))))))
4739 (define-obsolete-function-alias
4740 'python-info-closing-block-message
4741 'python-info-dedenter-opening-block-message "24.4")
4743 (defun python-info-dedenter-opening-block-message ()
4744 "Message the first line of the block the current statement closes."
4745 (let ((point (python-info-dedenter-opening-block-position)))
4746 (when point
4747 (save-restriction
4748 (prog-widen)
4749 (message "Closes %s" (save-excursion
4750 (goto-char point)
4751 (buffer-substring
4752 (point) (line-end-position))))))))
4754 (defun python-info-dedenter-statement-p ()
4755 "Return point if current statement is a dedenter.
4756 Sets `match-data' to the keyword that starts the dedenter
4757 statement."
4758 (save-excursion
4759 (python-nav-beginning-of-statement)
4760 (when (and (not (python-syntax-context-type))
4761 (looking-at (python-rx dedenter)))
4762 (point))))
4764 (defun python-info-line-ends-backslash-p (&optional line-number)
4765 "Return non-nil if current line ends with backslash.
4766 With optional argument LINE-NUMBER, check that line instead."
4767 (save-excursion
4768 (save-restriction
4769 (prog-widen)
4770 (when line-number
4771 (python-util-goto-line line-number))
4772 (while (and (not (eobp))
4773 (goto-char (line-end-position))
4774 (python-syntax-context 'paren)
4775 (not (equal (char-before (point)) ?\\)))
4776 (forward-line 1))
4777 (when (equal (char-before) ?\\)
4778 (point-marker)))))
4780 (defun python-info-beginning-of-backslash (&optional line-number)
4781 "Return the point where the backslashed line start.
4782 Optional argument LINE-NUMBER forces the line number to check against."
4783 (save-excursion
4784 (save-restriction
4785 (prog-widen)
4786 (when line-number
4787 (python-util-goto-line line-number))
4788 (when (python-info-line-ends-backslash-p)
4789 (while (save-excursion
4790 (goto-char (line-beginning-position))
4791 (python-syntax-context 'paren))
4792 (forward-line -1))
4793 (back-to-indentation)
4794 (point-marker)))))
4796 (defun python-info-continuation-line-p ()
4797 "Check if current line is continuation of another.
4798 When current line is continuation of another return the point
4799 where the continued line ends."
4800 (save-excursion
4801 (save-restriction
4802 (prog-widen)
4803 (let* ((context-type (progn
4804 (back-to-indentation)
4805 (python-syntax-context-type)))
4806 (line-start (line-number-at-pos))
4807 (context-start (when context-type
4808 (python-syntax-context context-type))))
4809 (cond ((equal context-type 'paren)
4810 ;; Lines inside a paren are always a continuation line
4811 ;; (except the first one).
4812 (python-util-forward-comment -1)
4813 (point-marker))
4814 ((member context-type '(string comment))
4815 ;; move forward an roll again
4816 (goto-char context-start)
4817 (python-util-forward-comment)
4818 (python-info-continuation-line-p))
4820 ;; Not within a paren, string or comment, the only way
4821 ;; we are dealing with a continuation line is that
4822 ;; previous line contains a backslash, and this can
4823 ;; only be the previous line from current
4824 (back-to-indentation)
4825 (python-util-forward-comment -1)
4826 (when (and (equal (1- line-start) (line-number-at-pos))
4827 (python-info-line-ends-backslash-p))
4828 (point-marker))))))))
4830 (defun python-info-block-continuation-line-p ()
4831 "Return non-nil if current line is a continuation of a block."
4832 (save-excursion
4833 (when (python-info-continuation-line-p)
4834 (forward-line -1)
4835 (back-to-indentation)
4836 (when (looking-at (python-rx block-start))
4837 (point-marker)))))
4839 (defun python-info-assignment-statement-p (&optional current-line-only)
4840 "Check if current line is an assignment.
4841 With argument CURRENT-LINE-ONLY is non-nil, don't follow any
4842 continuations, just check the if current line is an assignment."
4843 (save-excursion
4844 (let ((found nil))
4845 (if current-line-only
4846 (back-to-indentation)
4847 (python-nav-beginning-of-statement))
4848 (while (and
4849 (re-search-forward (python-rx not-simple-operator
4850 assignment-operator
4851 (group not-simple-operator))
4852 (line-end-position) t)
4853 (not found))
4854 (save-excursion
4855 ;; The assignment operator should not be inside a string.
4856 (backward-char (length (match-string-no-properties 1)))
4857 (setq found (not (python-syntax-context-type)))))
4858 (when found
4859 (skip-syntax-forward " ")
4860 (point-marker)))))
4862 ;; TODO: rename to clarify this is only for the first continuation
4863 ;; line or remove it and move its body to `python-indent-context'.
4864 (defun python-info-assignment-continuation-line-p ()
4865 "Check if current line is the first continuation of an assignment.
4866 When current line is continuation of another with an assignment
4867 return the point of the first non-blank character after the
4868 operator."
4869 (save-excursion
4870 (when (python-info-continuation-line-p)
4871 (forward-line -1)
4872 (python-info-assignment-statement-p t))))
4874 (defun python-info-looking-at-beginning-of-defun (&optional syntax-ppss)
4875 "Check if point is at `beginning-of-defun' using SYNTAX-PPSS."
4876 (and (not (python-syntax-context-type (or syntax-ppss (syntax-ppss))))
4877 (save-excursion
4878 (beginning-of-line 1)
4879 (looking-at python-nav-beginning-of-defun-regexp))))
4881 (defun python-info-current-line-comment-p ()
4882 "Return non-nil if current line is a comment line."
4883 (char-equal
4884 (or (char-after (+ (line-beginning-position) (current-indentation))) ?_)
4885 ?#))
4887 (defun python-info-current-line-empty-p ()
4888 "Return non-nil if current line is empty, ignoring whitespace."
4889 (save-excursion
4890 (beginning-of-line 1)
4891 (looking-at
4892 (python-rx line-start (* whitespace)
4893 (group (* not-newline))
4894 (* whitespace) line-end))
4895 (string-equal "" (match-string-no-properties 1))))
4897 (defun python-info-docstring-p (&optional syntax-ppss)
4898 "Return non-nil if point is in a docstring.
4899 When optional argument SYNTAX-PPSS is given, use that instead of
4900 point's current `syntax-ppss'."
4901 ;;; https://www.python.org/dev/peps/pep-0257/#what-is-a-docstring
4902 (save-excursion
4903 (when (and syntax-ppss (python-syntax-context 'string syntax-ppss))
4904 (goto-char (nth 8 syntax-ppss)))
4905 (python-nav-beginning-of-statement)
4906 (let ((counter 1)
4907 (indentation (current-indentation))
4908 (backward-sexp-point)
4909 (re (concat "[uU]?[rR]?"
4910 (python-rx string-delimiter))))
4911 (when (and
4912 (not (python-info-assignment-statement-p))
4913 (looking-at-p re)
4914 ;; Allow up to two consecutive docstrings only.
4917 (let (last-backward-sexp-point)
4918 (while (save-excursion
4919 (python-nav-backward-sexp)
4920 (setq backward-sexp-point (point))
4921 (and (= indentation (current-indentation))
4922 ;; Make sure we're always moving point.
4923 ;; If we get stuck in the same position
4924 ;; on consecutive loop iterations,
4925 ;; bail out.
4926 (prog1 (not (eql last-backward-sexp-point
4927 backward-sexp-point))
4928 (setq last-backward-sexp-point
4929 backward-sexp-point))
4930 (looking-at-p
4931 (concat "[uU]?[rR]?"
4932 (python-rx string-delimiter)))))
4933 ;; Previous sexp was a string, restore point.
4934 (goto-char backward-sexp-point)
4935 (cl-incf counter))
4936 counter)))
4937 (python-util-forward-comment -1)
4938 (python-nav-beginning-of-statement)
4939 (cond ((bobp))
4940 ((python-info-assignment-statement-p) t)
4941 ((python-info-looking-at-beginning-of-defun))
4942 (t nil))))))
4944 (defun python-info-encoding-from-cookie ()
4945 "Detect current buffer's encoding from its coding cookie.
4946 Returns the encoding as a symbol."
4947 (let ((first-two-lines
4948 (save-excursion
4949 (save-restriction
4950 (widen)
4951 (goto-char (point-min))
4952 (forward-line 2)
4953 (buffer-substring-no-properties
4954 (point)
4955 (point-min))))))
4956 (when (string-match (python-rx coding-cookie) first-two-lines)
4957 (intern (match-string-no-properties 1 first-two-lines)))))
4959 (defun python-info-encoding ()
4960 "Return encoding for file.
4961 Try `python-info-encoding-from-cookie', if none is found then
4962 default to utf-8."
4963 ;; If no encoding is defined, then it's safe to use UTF-8: Python 2
4964 ;; uses ASCII as default while Python 3 uses UTF-8. This means that
4965 ;; in the worst case scenario python.el will make things work for
4966 ;; Python 2 files with unicode data and no encoding defined.
4967 (or (python-info-encoding-from-cookie)
4968 'utf-8))
4971 ;;; Utility functions
4973 (defun python-util-goto-line (line-number)
4974 "Move point to LINE-NUMBER."
4975 (goto-char (point-min))
4976 (forward-line (1- line-number)))
4978 ;; Stolen from org-mode
4979 (defun python-util-clone-local-variables (from-buffer &optional regexp)
4980 "Clone local variables from FROM-BUFFER.
4981 Optional argument REGEXP selects variables to clone and defaults
4982 to \"^python-\"."
4983 (mapc
4984 (lambda (pair)
4985 (and (symbolp (car pair))
4986 (string-match (or regexp "^python-")
4987 (symbol-name (car pair)))
4988 (set (make-local-variable (car pair))
4989 (cdr pair))))
4990 (buffer-local-variables from-buffer)))
4992 (defvar comint-last-prompt-overlay) ; Shut up, byte compiler.
4994 (defun python-util-comint-last-prompt ()
4995 "Return comint last prompt overlay start and end.
4996 This is for compatibility with Emacs < 24.4."
4997 (cond ((bound-and-true-p comint-last-prompt-overlay)
4998 (cons (overlay-start comint-last-prompt-overlay)
4999 (overlay-end comint-last-prompt-overlay)))
5000 ((bound-and-true-p comint-last-prompt)
5001 comint-last-prompt)
5002 (t nil)))
5004 (defun python-util-forward-comment (&optional direction)
5005 "Python mode specific version of `forward-comment'.
5006 Optional argument DIRECTION defines the direction to move to."
5007 (let ((comment-start (python-syntax-context 'comment))
5008 (factor (if (< (or direction 0) 0)
5009 -99999
5010 99999)))
5011 (when comment-start
5012 (goto-char comment-start))
5013 (forward-comment factor)))
5015 (defun python-util-list-directories (directory &optional predicate max-depth)
5016 "List DIRECTORY subdirs, filtered by PREDICATE and limited by MAX-DEPTH.
5017 Argument PREDICATE defaults to `identity' and must be a function
5018 that takes one argument (a full path) and returns non-nil for
5019 allowed files. When optional argument MAX-DEPTH is non-nil, stop
5020 searching when depth is reached, else don't limit."
5021 (let* ((dir (expand-file-name directory))
5022 (dir-length (length dir))
5023 (predicate (or predicate #'identity))
5024 (to-scan (list dir))
5025 (tally nil))
5026 (while to-scan
5027 (let ((current-dir (car to-scan)))
5028 (when (funcall predicate current-dir)
5029 (setq tally (cons current-dir tally)))
5030 (setq to-scan (append (cdr to-scan)
5031 (python-util-list-files
5032 current-dir #'file-directory-p)
5033 nil))
5034 (when (and max-depth
5035 (<= max-depth
5036 (length (split-string
5037 (substring current-dir dir-length)
5038 "/\\|\\\\" t))))
5039 (setq to-scan nil))))
5040 (nreverse tally)))
5042 (defun python-util-list-files (dir &optional predicate)
5043 "List files in DIR, filtering with PREDICATE.
5044 Argument PREDICATE defaults to `identity' and must be a function
5045 that takes one argument (a full path) and returns non-nil for
5046 allowed files."
5047 (let ((dir-name (file-name-as-directory dir)))
5048 (apply #'nconc
5049 (mapcar (lambda (file-name)
5050 (let ((full-file-name (expand-file-name file-name dir-name)))
5051 (when (and
5052 (not (member file-name '("." "..")))
5053 (funcall (or predicate #'identity) full-file-name))
5054 (list full-file-name))))
5055 (directory-files dir-name)))))
5057 (defun python-util-list-packages (dir &optional max-depth)
5058 "List packages in DIR, limited by MAX-DEPTH.
5059 When optional argument MAX-DEPTH is non-nil, stop searching when
5060 depth is reached, else don't limit."
5061 (let* ((dir (expand-file-name dir))
5062 (parent-dir (file-name-directory
5063 (directory-file-name
5064 (file-name-directory
5065 (file-name-as-directory dir)))))
5066 (subpath-length (length parent-dir)))
5067 (mapcar
5068 (lambda (file-name)
5069 (replace-regexp-in-string
5070 (rx (or ?\\ ?/)) "." (substring file-name subpath-length)))
5071 (python-util-list-directories
5072 (directory-file-name dir)
5073 (lambda (dir)
5074 (file-exists-p (expand-file-name "__init__.py" dir)))
5075 max-depth))))
5077 (defun python-util-popn (lst n)
5078 "Return LST first N elements.
5079 N should be an integer, when negative its opposite is used.
5080 When N is bigger than the length of LST, the list is
5081 returned as is."
5082 (let* ((n (min (abs n)))
5083 (len (length lst))
5084 (acc))
5085 (if (> n len)
5087 (while (< 0 n)
5088 (setq acc (cons (car lst) acc)
5089 lst (cdr lst)
5090 n (1- n)))
5091 (reverse acc))))
5093 (defun python-util-strip-string (string)
5094 "Strip STRING whitespace and newlines from end and beginning."
5095 (replace-regexp-in-string
5096 (rx (or (: string-start (* (any whitespace ?\r ?\n)))
5097 (: (* (any whitespace ?\r ?\n)) string-end)))
5099 string))
5101 (defun python-util-valid-regexp-p (regexp)
5102 "Return non-nil if REGEXP is valid."
5103 (ignore-errors (string-match regexp "") t))
5106 (defun python-electric-pair-string-delimiter ()
5107 (when (and electric-pair-mode
5108 (memq last-command-event '(?\" ?\'))
5109 (let ((count 0))
5110 (while (eq (char-before (- (point) count)) last-command-event)
5111 (cl-incf count))
5112 (= count 3))
5113 (eq (char-after) last-command-event))
5114 (save-excursion (insert (make-string 2 last-command-event)))))
5116 (defvar electric-indent-inhibit)
5118 ;;;###autoload
5119 (define-derived-mode python-mode prog-mode "Python"
5120 "Major mode for editing Python files.
5122 \\{python-mode-map}"
5123 (set (make-local-variable 'tab-width) 8)
5124 (set (make-local-variable 'indent-tabs-mode) nil)
5126 (set (make-local-variable 'comment-start) "# ")
5127 (set (make-local-variable 'comment-start-skip) "#+\\s-*")
5129 (set (make-local-variable 'parse-sexp-lookup-properties) t)
5130 (set (make-local-variable 'parse-sexp-ignore-comments) t)
5132 (set (make-local-variable 'forward-sexp-function)
5133 'python-nav-forward-sexp)
5135 (set (make-local-variable 'font-lock-defaults)
5136 '(python-font-lock-keywords
5137 nil nil nil nil
5138 (font-lock-syntactic-face-function
5139 . python-font-lock-syntactic-face-function)))
5141 (set (make-local-variable 'syntax-propertize-function)
5142 python-syntax-propertize-function)
5144 (set (make-local-variable 'indent-line-function)
5145 #'python-indent-line-function)
5146 (set (make-local-variable 'indent-region-function) #'python-indent-region)
5147 ;; Because indentation is not redundant, we cannot safely reindent code.
5148 (set (make-local-variable 'electric-indent-inhibit) t)
5149 (set (make-local-variable 'electric-indent-chars)
5150 (cons ?: electric-indent-chars))
5152 ;; Add """ ... """ pairing to electric-pair-mode.
5153 (add-hook 'post-self-insert-hook
5154 #'python-electric-pair-string-delimiter 'append t)
5156 (set (make-local-variable 'paragraph-start) "\\s-*$")
5157 (set (make-local-variable 'fill-paragraph-function)
5158 #'python-fill-paragraph)
5160 (set (make-local-variable 'beginning-of-defun-function)
5161 #'python-nav-beginning-of-defun)
5162 (set (make-local-variable 'end-of-defun-function)
5163 #'python-nav-end-of-defun)
5165 (add-hook 'completion-at-point-functions
5166 #'python-completion-at-point nil 'local)
5168 (add-hook 'post-self-insert-hook
5169 #'python-indent-post-self-insert-function 'append 'local)
5171 (set (make-local-variable 'imenu-create-index-function)
5172 #'python-imenu-create-index)
5174 (set (make-local-variable 'add-log-current-defun-function)
5175 #'python-info-current-defun)
5177 (add-hook 'which-func-functions #'python-info-current-defun nil t)
5179 (set (make-local-variable 'skeleton-further-elements)
5180 '((abbrev-mode nil)
5181 (< '(backward-delete-char-untabify (min python-indent-offset
5182 (current-column))))
5183 (^ '(- (1+ (current-indentation))))))
5185 (if (null eldoc-documentation-function)
5186 ;; Emacs<25
5187 (set (make-local-variable 'eldoc-documentation-function)
5188 #'python-eldoc-function)
5189 (add-function :before-until (local 'eldoc-documentation-function)
5190 #'python-eldoc-function))
5192 (add-to-list
5193 'hs-special-modes-alist
5194 `(python-mode
5195 "\\s-*\\_<\\(?:def\\|class\\)\\_>"
5196 ;; Use the empty string as end regexp so it doesn't default to
5197 ;; "\\s)". This way parens at end of defun are properly hidden.
5200 python-hideshow-forward-sexp-function
5201 nil))
5203 (set (make-local-variable 'outline-regexp)
5204 (python-rx (* space) block-start))
5205 (set (make-local-variable 'outline-heading-end-regexp) ":[^\n]*\n")
5206 (set (make-local-variable 'outline-level)
5207 #'(lambda ()
5208 "`outline-level' function for Python mode."
5209 (1+ (/ (current-indentation) python-indent-offset))))
5211 (set (make-local-variable 'prettify-symbols-alist)
5212 python--prettify-symbols-alist)
5214 (python-skeleton-add-menu-items)
5216 (make-local-variable 'python-shell-internal-buffer)
5218 (when python-indent-guess-indent-offset
5219 (python-indent-guess-indent-offset)))
5222 (provide 'python)
5224 ;; Local Variables:
5225 ;; indent-tabs-mode: nil
5226 ;; End:
5228 ;;; python.el ends here