]> code.delx.au - gnu-emacs/blob - lisp/progmodes/python.el
Fix a bunch of custom types (thank you cus-test.el)
[gnu-emacs] / lisp / progmodes / python.el
1 ;;; python.el --- Python's flying circus support for Emacs
2
3 ;; Copyright (C) 2003-2013 Free Software Foundation, Inc.
4
5 ;; Author: Fabián E. Gallina <fabian@anue.biz>
6 ;; URL: https://github.com/fgallina/python.el
7 ;; Version: 0.24.2
8 ;; Maintainer: FSF
9 ;; Created: Jul 2010
10 ;; Keywords: languages
11
12 ;; This file is part of GNU Emacs.
13
14 ;; GNU Emacs is free software: you can redistribute it and/or modify
15 ;; it under the terms of the GNU General Public License as published
16 ;; by the Free Software Foundation, either version 3 of the License,
17 ;; or (at your option) any later version.
18
19 ;; GNU Emacs is distributed in the hope that it will be useful, but
20 ;; WITHOUT ANY WARRANTY; without even the implied warranty of
21 ;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22 ;; General Public License for more details.
23
24 ;; You should have received a copy of the GNU General Public License
25 ;; along with GNU Emacs. If not, see <http://www.gnu.org/licenses/>.
26
27 ;;; Commentary:
28
29 ;; Major mode for editing Python files with some fontification and
30 ;; indentation bits extracted from original Dave Love's python.el
31 ;; found in GNU/Emacs.
32
33 ;; Implements Syntax highlighting, Indentation, Movement, Shell
34 ;; interaction, Shell completion, Shell virtualenv support, Pdb
35 ;; tracking, Symbol completion, Skeletons, FFAP, Code Check, Eldoc,
36 ;; Imenu.
37
38 ;; Syntax highlighting: Fontification of code is provided and supports
39 ;; python's triple quoted strings properly.
40
41 ;; Indentation: Automatic indentation with indentation cycling is
42 ;; provided, it allows you to navigate different available levels of
43 ;; indentation by hitting <tab> several times. Also when inserting a
44 ;; colon the `python-indent-electric-colon' command is invoked and
45 ;; causes the current line to be dedented automatically if needed.
46
47 ;; Movement: `beginning-of-defun' and `end-of-defun' functions are
48 ;; properly implemented. There are also specialized
49 ;; `forward-sentence' and `backward-sentence' replacements called
50 ;; `python-nav-forward-block', `python-nav-backward-block'
51 ;; respectively which navigate between beginning of blocks of code.
52 ;; Extra functions `python-nav-forward-statement',
53 ;; `python-nav-backward-statement',
54 ;; `python-nav-beginning-of-statement', `python-nav-end-of-statement',
55 ;; `python-nav-beginning-of-block' and `python-nav-end-of-block' are
56 ;; included but no bound to any key. At last but not least the
57 ;; specialized `python-nav-forward-sexp' allows easy navigation
58 ;; between code blocks. If you prefer `cc-mode'-like `forward-sexp'
59 ;; movement, setting `forward-sexp-function' to nil is enough, You can
60 ;; do that using the `python-mode-hook':
61
62 ;; (add-hook 'python-mode-hook
63 ;; (lambda () (setq forward-sexp-function nil)))
64
65 ;; Shell interaction: is provided and allows you to execute easily any
66 ;; block of code of your current buffer in an inferior Python process.
67
68 ;; Shell completion: hitting tab will try to complete the current
69 ;; word. Shell completion is implemented in a manner that if you
70 ;; change the `python-shell-interpreter' to any other (for example
71 ;; IPython) it should be easy to integrate another way to calculate
72 ;; completions. You just need to specify your custom
73 ;; `python-shell-completion-setup-code' and
74 ;; `python-shell-completion-string-code'.
75
76 ;; Here is a complete example of the settings you would use for
77 ;; iPython 0.11:
78
79 ;; (setq
80 ;; python-shell-interpreter "ipython"
81 ;; python-shell-interpreter-args ""
82 ;; python-shell-prompt-regexp "In \\[[0-9]+\\]: "
83 ;; python-shell-prompt-output-regexp "Out\\[[0-9]+\\]: "
84 ;; python-shell-completion-setup-code
85 ;; "from IPython.core.completerlib import module_completion"
86 ;; python-shell-completion-module-string-code
87 ;; "';'.join(module_completion('''%s'''))\n"
88 ;; python-shell-completion-string-code
89 ;; "';'.join(get_ipython().Completer.all_completions('''%s'''))\n")
90
91 ;; For iPython 0.10 everything would be the same except for
92 ;; `python-shell-completion-string-code' and
93 ;; `python-shell-completion-module-string-code':
94
95 ;; (setq python-shell-completion-string-code
96 ;; "';'.join(__IP.complete('''%s'''))\n"
97 ;; python-shell-completion-module-string-code "")
98
99 ;; Unfortunately running iPython on Windows needs some more tweaking.
100 ;; The way you must set `python-shell-interpreter' and
101 ;; `python-shell-interpreter-args' is as follows:
102
103 ;; (setq
104 ;; python-shell-interpreter "C:\\Python27\\python.exe"
105 ;; python-shell-interpreter-args
106 ;; "-i C:\\Python27\\Scripts\\ipython-script.py")
107
108 ;; That will spawn the iPython process correctly (Of course you need
109 ;; to modify the paths according to your system).
110
111 ;; Please note that the default completion system depends on the
112 ;; readline module, so if you are using some Operating System that
113 ;; bundles Python without it (like Windows) just install the
114 ;; pyreadline from http://ipython.scipy.org/moin/PyReadline/Intro and
115 ;; you should be good to go.
116
117 ;; Shell virtualenv support: The shell also contains support for
118 ;; virtualenvs and other special environment modifications thanks to
119 ;; `python-shell-process-environment' and `python-shell-exec-path'.
120 ;; These two variables allows you to modify execution paths and
121 ;; environment variables to make easy for you to setup virtualenv rules
122 ;; or behavior modifications when running shells. Here is an example
123 ;; of how to make shell processes to be run using the /path/to/env/
124 ;; virtualenv:
125
126 ;; (setq python-shell-process-environment
127 ;; (list
128 ;; (format "PATH=%s" (mapconcat
129 ;; 'identity
130 ;; (reverse
131 ;; (cons (getenv "PATH")
132 ;; '("/path/to/env/bin/")))
133 ;; ":"))
134 ;; "VIRTUAL_ENV=/path/to/env/"))
135 ;; (python-shell-exec-path . ("/path/to/env/bin/"))
136
137 ;; Since the above is cumbersome and can be programmatically
138 ;; calculated, the variable `python-shell-virtualenv-path' is
139 ;; provided. When this variable is set with the path of the
140 ;; virtualenv to use, `process-environment' and `exec-path' get proper
141 ;; values in order to run shells inside the specified virtualenv. So
142 ;; the following will achieve the same as the previous example:
143
144 ;; (setq python-shell-virtualenv-path "/path/to/env/")
145
146 ;; Also the `python-shell-extra-pythonpaths' variable have been
147 ;; introduced as simple way of adding paths to the PYTHONPATH without
148 ;; affecting existing values.
149
150 ;; Pdb tracking: when you execute a block of code that contains some
151 ;; call to pdb (or ipdb) it will prompt the block of code and will
152 ;; follow the execution of pdb marking the current line with an arrow.
153
154 ;; Symbol completion: you can complete the symbol at point. It uses
155 ;; the shell completion in background so you should run
156 ;; `python-shell-send-buffer' from time to time to get better results.
157
158 ;; Skeletons: 6 skeletons are provided for simple inserting of class,
159 ;; def, for, if, try and while. These skeletons are integrated with
160 ;; dabbrev. If you have `dabbrev-mode' activated and
161 ;; `python-skeleton-autoinsert' is set to t, then whenever you type
162 ;; the name of any of those defined and hit SPC, they will be
163 ;; automatically expanded. As an alternative you can use the defined
164 ;; skeleton commands: `python-skeleton-class', `python-skeleton-def'
165 ;; `python-skeleton-for', `python-skeleton-if', `python-skeleton-try'
166 ;; and `python-skeleton-while'.
167
168 ;; FFAP: You can find the filename for a given module when using ffap
169 ;; out of the box. This feature needs an inferior python shell
170 ;; running.
171
172 ;; Code check: Check the current file for errors with `python-check'
173 ;; using the program defined in `python-check-command'.
174
175 ;; Eldoc: returns documentation for object at point by using the
176 ;; inferior python subprocess to inspect its documentation. As you
177 ;; might guessed you should run `python-shell-send-buffer' from time
178 ;; to time to get better results too.
179
180 ;; Imenu: There are two index building functions to be used as
181 ;; `imenu-create-index-function': `python-imenu-create-index' (the
182 ;; default one, builds the alist in form of a tree) and
183 ;; `python-imenu-create-flat-index'. See also
184 ;; `python-imenu-format-item-label-function',
185 ;; `python-imenu-format-parent-item-label-function',
186 ;; `python-imenu-format-parent-item-jump-label-function' variables for
187 ;; changing the way labels are formatted in the tree version.
188
189 ;; If you used python-mode.el you probably will miss auto-indentation
190 ;; when inserting newlines. To achieve the same behavior you have
191 ;; two options:
192
193 ;; 1) Use GNU/Emacs' standard binding for `newline-and-indent': C-j.
194
195 ;; 2) Add the following hook in your .emacs:
196
197 ;; (add-hook 'python-mode-hook
198 ;; #'(lambda ()
199 ;; (define-key python-mode-map "\C-m" 'newline-and-indent)))
200
201 ;; I'd recommend the first one since you'll get the same behavior for
202 ;; all modes out-of-the-box.
203
204 ;;; Installation:
205
206 ;; Add this to your .emacs:
207
208 ;; (add-to-list 'load-path "/folder/containing/file")
209 ;; (require 'python)
210
211 ;;; TODO:
212
213 ;;; Code:
214
215 (require 'ansi-color)
216 (require 'comint)
217
218 ;; Avoid compiler warnings
219 (defvar view-return-to-alist)
220 (defvar compilation-error-regexp-alist)
221 (defvar outline-heading-end-regexp)
222
223 (autoload 'comint-mode "comint")
224
225 ;;;###autoload
226 (add-to-list 'auto-mode-alist (cons (purecopy "\\.py\\'") 'python-mode))
227 ;;;###autoload
228 (add-to-list 'interpreter-mode-alist (cons (purecopy "python") 'python-mode))
229
230 (defgroup python nil
231 "Python Language's flying circus support for Emacs."
232 :group 'languages
233 :version "24.3"
234 :link '(emacs-commentary-link "python"))
235
236 \f
237 ;;; Bindings
238
239 (defvar python-mode-map
240 (let ((map (make-sparse-keymap)))
241 ;; Movement
242 (define-key map [remap backward-sentence] 'python-nav-backward-block)
243 (define-key map [remap forward-sentence] 'python-nav-forward-block)
244 (define-key map [remap backward-up-list] 'python-nav-backward-up-list)
245 (define-key map "\C-c\C-j" 'imenu)
246 ;; Indent specific
247 (define-key map "\177" 'python-indent-dedent-line-backspace)
248 (define-key map (kbd "<backtab>") 'python-indent-dedent-line)
249 (define-key map "\C-c<" 'python-indent-shift-left)
250 (define-key map "\C-c>" 'python-indent-shift-right)
251 (define-key map ":" 'python-indent-electric-colon)
252 ;; Skeletons
253 (define-key map "\C-c\C-tc" 'python-skeleton-class)
254 (define-key map "\C-c\C-td" 'python-skeleton-def)
255 (define-key map "\C-c\C-tf" 'python-skeleton-for)
256 (define-key map "\C-c\C-ti" 'python-skeleton-if)
257 (define-key map "\C-c\C-tt" 'python-skeleton-try)
258 (define-key map "\C-c\C-tw" 'python-skeleton-while)
259 ;; Shell interaction
260 (define-key map "\C-c\C-p" 'run-python)
261 (define-key map "\C-c\C-s" 'python-shell-send-string)
262 (define-key map "\C-c\C-r" 'python-shell-send-region)
263 (define-key map "\C-\M-x" 'python-shell-send-defun)
264 (define-key map "\C-c\C-c" 'python-shell-send-buffer)
265 (define-key map "\C-c\C-l" 'python-shell-send-file)
266 (define-key map "\C-c\C-z" 'python-shell-switch-to-shell)
267 ;; Some util commands
268 (define-key map "\C-c\C-v" 'python-check)
269 (define-key map "\C-c\C-f" 'python-eldoc-at-point)
270 ;; Utilities
271 (substitute-key-definition 'complete-symbol 'completion-at-point
272 map global-map)
273 (easy-menu-define python-menu map "Python Mode menu"
274 `("Python"
275 :help "Python-specific Features"
276 ["Shift region left" python-indent-shift-left :active mark-active
277 :help "Shift region left by a single indentation step"]
278 ["Shift region right" python-indent-shift-right :active mark-active
279 :help "Shift region right by a single indentation step"]
280 "-"
281 ["Start of def/class" beginning-of-defun
282 :help "Go to start of outermost definition around point"]
283 ["End of def/class" end-of-defun
284 :help "Go to end of definition around point"]
285 ["Mark def/class" mark-defun
286 :help "Mark outermost definition around point"]
287 ["Jump to def/class" imenu
288 :help "Jump to a class or function definition"]
289 "--"
290 ("Skeletons")
291 "---"
292 ["Start interpreter" run-python
293 :help "Run inferior Python process in a separate buffer"]
294 ["Switch to shell" python-shell-switch-to-shell
295 :help "Switch to running inferior Python process"]
296 ["Eval string" python-shell-send-string
297 :help "Eval string in inferior Python session"]
298 ["Eval buffer" python-shell-send-buffer
299 :help "Eval buffer in inferior Python session"]
300 ["Eval region" python-shell-send-region
301 :help "Eval region in inferior Python session"]
302 ["Eval defun" python-shell-send-defun
303 :help "Eval defun in inferior Python session"]
304 ["Eval file" python-shell-send-file
305 :help "Eval file in inferior Python session"]
306 ["Debugger" pdb :help "Run pdb under GUD"]
307 "----"
308 ["Check file" python-check
309 :help "Check file for errors"]
310 ["Help on symbol" python-eldoc-at-point
311 :help "Get help on symbol at point"]
312 ["Complete symbol" completion-at-point
313 :help "Complete symbol before point"]))
314 map)
315 "Keymap for `python-mode'.")
316
317 \f
318 ;;; Python specialized rx
319
320 (eval-when-compile
321 (defconst python-rx-constituents
322 `((block-start . ,(rx symbol-start
323 (or "def" "class" "if" "elif" "else" "try"
324 "except" "finally" "for" "while" "with")
325 symbol-end))
326 (decorator . ,(rx line-start (* space) ?@ (any letter ?_)
327 (* (any word ?_))))
328 (defun . ,(rx symbol-start (or "def" "class") symbol-end))
329 (if-name-main . ,(rx line-start "if" (+ space) "__name__"
330 (+ space) "==" (+ space)
331 (any ?' ?\") "__main__" (any ?' ?\")
332 (* space) ?:))
333 (symbol-name . ,(rx (any letter ?_) (* (any word ?_))))
334 (open-paren . ,(rx (or "{" "[" "(")))
335 (close-paren . ,(rx (or "}" "]" ")")))
336 (simple-operator . ,(rx (any ?+ ?- ?/ ?& ?^ ?~ ?| ?* ?< ?> ?= ?%)))
337 ;; FIXME: rx should support (not simple-operator).
338 (not-simple-operator . ,(rx
339 (not
340 (any ?+ ?- ?/ ?& ?^ ?~ ?| ?* ?< ?> ?= ?%))))
341 ;; FIXME: Use regexp-opt.
342 (operator . ,(rx (or "+" "-" "/" "&" "^" "~" "|" "*" "<" ">"
343 "=" "%" "**" "//" "<<" ">>" "<=" "!="
344 "==" ">=" "is" "not")))
345 ;; FIXME: Use regexp-opt.
346 (assignment-operator . ,(rx (or "=" "+=" "-=" "*=" "/=" "//=" "%=" "**="
347 ">>=" "<<=" "&=" "^=" "|=")))
348 (string-delimiter . ,(rx (and
349 ;; Match even number of backslashes.
350 (or (not (any ?\\ ?\' ?\")) point
351 ;; Quotes might be preceded by a escaped quote.
352 (and (or (not (any ?\\)) point) ?\\
353 (* ?\\ ?\\) (any ?\' ?\")))
354 (* ?\\ ?\\)
355 ;; Match single or triple quotes of any kind.
356 (group (or "\"" "\"\"\"" "'" "'''"))))))
357 "Additional Python specific sexps for `python-rx'")
358
359 (defmacro python-rx (&rest regexps)
360 "Python mode specialized rx macro.
361 This variant of `rx' supports common python named REGEXPS."
362 (let ((rx-constituents (append python-rx-constituents rx-constituents)))
363 (cond ((null regexps)
364 (error "No regexp"))
365 ((cdr regexps)
366 (rx-to-string `(and ,@regexps) t))
367 (t
368 (rx-to-string (car regexps) t))))))
369
370 \f
371 ;;; Font-lock and syntax
372
373 (eval-when-compile
374 (defun python-syntax--context-compiler-macro (form type &optional syntax-ppss)
375 (pcase type
376 (`'comment
377 `(let ((ppss (or ,syntax-ppss (syntax-ppss))))
378 (and (nth 4 ppss) (nth 8 ppss))))
379 (`'string
380 `(let ((ppss (or ,syntax-ppss (syntax-ppss))))
381 (and (nth 3 ppss) (nth 8 ppss))))
382 (`'paren
383 `(nth 1 (or ,syntax-ppss (syntax-ppss))))
384 (_ form))))
385
386 (defun python-syntax-context (type &optional syntax-ppss)
387 "Return non-nil if point is on TYPE using SYNTAX-PPSS.
388 TYPE can be `comment', `string' or `paren'. It returns the start
389 character address of the specified TYPE."
390 (declare (compiler-macro python-syntax--context-compiler-macro))
391 (let ((ppss (or syntax-ppss (syntax-ppss))))
392 (pcase type
393 (`comment (and (nth 4 ppss) (nth 8 ppss)))
394 (`string (and (nth 3 ppss) (nth 8 ppss)))
395 (`paren (nth 1 ppss))
396 (_ nil))))
397
398 (defun python-syntax-context-type (&optional syntax-ppss)
399 "Return the context type using SYNTAX-PPSS.
400 The type returned can be `comment', `string' or `paren'."
401 (let ((ppss (or syntax-ppss (syntax-ppss))))
402 (cond
403 ((nth 8 ppss) (if (nth 4 ppss) 'comment 'string))
404 ((nth 1 ppss) 'paren))))
405
406 (defsubst python-syntax-comment-or-string-p ()
407 "Return non-nil if point is inside 'comment or 'string."
408 (nth 8 (syntax-ppss)))
409
410 (define-obsolete-function-alias
411 'python-info-ppss-context #'python-syntax-context "24.3")
412
413 (define-obsolete-function-alias
414 'python-info-ppss-context-type #'python-syntax-context-type "24.3")
415
416 (define-obsolete-function-alias
417 'python-info-ppss-comment-or-string-p
418 #'python-syntax-comment-or-string-p "24.3")
419
420 (defvar python-font-lock-keywords
421 ;; Keywords
422 `(,(rx symbol-start
423 (or
424 "and" "del" "from" "not" "while" "as" "elif" "global" "or" "with"
425 "assert" "else" "if" "pass" "yield" "break" "except" "import" "class"
426 "in" "raise" "continue" "finally" "is" "return" "def" "for" "lambda"
427 "try"
428 ;; Python 2:
429 "print" "exec"
430 ;; Python 3:
431 ;; False, None, and True are listed as keywords on the Python 3
432 ;; documentation, but since they also qualify as constants they are
433 ;; fontified like that in order to keep font-lock consistent between
434 ;; Python versions.
435 "nonlocal"
436 ;; Extra:
437 "self")
438 symbol-end)
439 ;; functions
440 (,(rx symbol-start "def" (1+ space) (group (1+ (or word ?_))))
441 (1 font-lock-function-name-face))
442 ;; classes
443 (,(rx symbol-start "class" (1+ space) (group (1+ (or word ?_))))
444 (1 font-lock-type-face))
445 ;; Constants
446 (,(rx symbol-start
447 (or
448 "Ellipsis" "False" "None" "NotImplemented" "True" "__debug__"
449 ;; copyright, license, credits, quit and exit are added by the site
450 ;; module and they are not intended to be used in programs
451 "copyright" "credits" "exit" "license" "quit")
452 symbol-end) . font-lock-constant-face)
453 ;; Decorators.
454 (,(rx line-start (* (any " \t")) (group "@" (1+ (or word ?_))
455 (0+ "." (1+ (or word ?_)))))
456 (1 font-lock-type-face))
457 ;; Builtin Exceptions
458 (,(rx symbol-start
459 (or
460 "ArithmeticError" "AssertionError" "AttributeError" "BaseException"
461 "DeprecationWarning" "EOFError" "EnvironmentError" "Exception"
462 "FloatingPointError" "FutureWarning" "GeneratorExit" "IOError"
463 "ImportError" "ImportWarning" "IndexError" "KeyError"
464 "KeyboardInterrupt" "LookupError" "MemoryError" "NameError"
465 "NotImplementedError" "OSError" "OverflowError"
466 "PendingDeprecationWarning" "ReferenceError" "RuntimeError"
467 "RuntimeWarning" "StopIteration" "SyntaxError" "SyntaxWarning"
468 "SystemError" "SystemExit" "TypeError" "UnboundLocalError"
469 "UnicodeDecodeError" "UnicodeEncodeError" "UnicodeError"
470 "UnicodeTranslateError" "UnicodeWarning" "UserWarning" "VMSError"
471 "ValueError" "Warning" "WindowsError" "ZeroDivisionError"
472 ;; Python 2:
473 "StandardError"
474 ;; Python 3:
475 "BufferError" "BytesWarning" "IndentationError" "ResourceWarning"
476 "TabError")
477 symbol-end) . font-lock-type-face)
478 ;; Builtins
479 (,(rx symbol-start
480 (or
481 "abs" "all" "any" "bin" "bool" "callable" "chr" "classmethod"
482 "compile" "complex" "delattr" "dict" "dir" "divmod" "enumerate"
483 "eval" "filter" "float" "format" "frozenset" "getattr" "globals"
484 "hasattr" "hash" "help" "hex" "id" "input" "int" "isinstance"
485 "issubclass" "iter" "len" "list" "locals" "map" "max" "memoryview"
486 "min" "next" "object" "oct" "open" "ord" "pow" "print" "property"
487 "range" "repr" "reversed" "round" "set" "setattr" "slice" "sorted"
488 "staticmethod" "str" "sum" "super" "tuple" "type" "vars" "zip"
489 "__import__"
490 ;; Python 2:
491 "basestring" "cmp" "execfile" "file" "long" "raw_input" "reduce"
492 "reload" "unichr" "unicode" "xrange" "apply" "buffer" "coerce"
493 "intern"
494 ;; Python 3:
495 "ascii" "bytearray" "bytes" "exec"
496 ;; Extra:
497 "__all__" "__doc__" "__name__" "__package__")
498 symbol-end) . font-lock-builtin-face)
499 ;; assignments
500 ;; support for a = b = c = 5
501 (,(lambda (limit)
502 (let ((re (python-rx (group (+ (any word ?. ?_)))
503 (? ?\[ (+ (not (any ?\]))) ?\]) (* space)
504 assignment-operator)))
505 (when (re-search-forward re limit t)
506 (while (and (python-syntax-context 'paren)
507 (re-search-forward re limit t)))
508 (if (not (or (python-syntax-context 'paren)
509 (equal (char-after (point-marker)) ?=)))
510 t
511 (set-match-data nil)))))
512 (1 font-lock-variable-name-face nil nil))
513 ;; support for a, b, c = (1, 2, 3)
514 (,(lambda (limit)
515 (let ((re (python-rx (group (+ (any word ?. ?_))) (* space)
516 (* ?, (* space) (+ (any word ?. ?_)) (* space))
517 ?, (* space) (+ (any word ?. ?_)) (* space)
518 assignment-operator)))
519 (when (and (re-search-forward re limit t)
520 (goto-char (nth 3 (match-data))))
521 (while (and (python-syntax-context 'paren)
522 (re-search-forward re limit t))
523 (goto-char (nth 3 (match-data))))
524 (if (not (python-syntax-context 'paren))
525 t
526 (set-match-data nil)))))
527 (1 font-lock-variable-name-face nil nil))))
528
529 (defconst python-syntax-propertize-function
530 (syntax-propertize-rules
531 ((python-rx string-delimiter)
532 (0 (ignore (python-syntax-stringify))))))
533
534 (defsubst python-syntax-count-quotes (quote-char &optional point limit)
535 "Count number of quotes around point (max is 3).
536 QUOTE-CHAR is the quote char to count. Optional argument POINT is
537 the point where scan starts (defaults to current point) and LIMIT
538 is used to limit the scan."
539 (let ((i 0))
540 (while (and (< i 3)
541 (or (not limit) (< (+ point i) limit))
542 (eq (char-after (+ point i)) quote-char))
543 (setq i (1+ i)))
544 i))
545
546 (defun python-syntax-stringify ()
547 "Put `syntax-table' property correctly on single/triple quotes."
548 (let* ((num-quotes (length (match-string-no-properties 1)))
549 (ppss (prog2
550 (backward-char num-quotes)
551 (syntax-ppss)
552 (forward-char num-quotes)))
553 (string-start (and (not (nth 4 ppss)) (nth 8 ppss)))
554 (quote-starting-pos (- (point) num-quotes))
555 (quote-ending-pos (point))
556 (num-closing-quotes
557 (and string-start
558 (python-syntax-count-quotes
559 (char-before) string-start quote-starting-pos))))
560 (cond ((and string-start (= num-closing-quotes 0))
561 ;; This set of quotes doesn't match the string starting
562 ;; kind. Do nothing.
563 nil)
564 ((not string-start)
565 ;; This set of quotes delimit the start of a string.
566 (put-text-property quote-starting-pos (1+ quote-starting-pos)
567 'syntax-table (string-to-syntax "|")))
568 ((= num-quotes num-closing-quotes)
569 ;; This set of quotes delimit the end of a string.
570 (put-text-property (1- quote-ending-pos) quote-ending-pos
571 'syntax-table (string-to-syntax "|")))
572 ((> num-quotes num-closing-quotes)
573 ;; This may only happen whenever a triple quote is closing
574 ;; a single quoted string. Add string delimiter syntax to
575 ;; all three quotes.
576 (put-text-property quote-starting-pos quote-ending-pos
577 'syntax-table (string-to-syntax "|"))))))
578
579 (defvar python-mode-syntax-table
580 (let ((table (make-syntax-table)))
581 ;; Give punctuation syntax to ASCII that normally has symbol
582 ;; syntax or has word syntax and isn't a letter.
583 (let ((symbol (string-to-syntax "_"))
584 (sst (standard-syntax-table)))
585 (dotimes (i 128)
586 (unless (= i ?_)
587 (if (equal symbol (aref sst i))
588 (modify-syntax-entry i "." table)))))
589 (modify-syntax-entry ?$ "." table)
590 (modify-syntax-entry ?% "." table)
591 ;; exceptions
592 (modify-syntax-entry ?# "<" table)
593 (modify-syntax-entry ?\n ">" table)
594 (modify-syntax-entry ?' "\"" table)
595 (modify-syntax-entry ?` "$" table)
596 table)
597 "Syntax table for Python files.")
598
599 (defvar python-dotty-syntax-table
600 (let ((table (make-syntax-table python-mode-syntax-table)))
601 (modify-syntax-entry ?. "w" table)
602 (modify-syntax-entry ?_ "w" table)
603 table)
604 "Dotty syntax table for Python files.
605 It makes underscores and dots word constituent chars.")
606
607 \f
608 ;;; Indentation
609
610 (defcustom python-indent-offset 4
611 "Default indentation offset for Python."
612 :group 'python
613 :type 'integer
614 :safe 'integerp)
615
616 (defcustom python-indent-guess-indent-offset t
617 "Non-nil tells Python mode to guess `python-indent-offset' value."
618 :type 'boolean
619 :group 'python
620 :safe 'booleanp)
621
622 (defcustom python-indent-trigger-commands
623 '(indent-for-tab-command yas-expand yas/expand)
624 "Commands that might trigger a `python-indent-line' call."
625 :type '(repeat symbol)
626 :group 'python)
627
628 (define-obsolete-variable-alias
629 'python-indent 'python-indent-offset "24.3")
630
631 (define-obsolete-variable-alias
632 'python-guess-indent 'python-indent-guess-indent-offset "24.3")
633
634 (defvar python-indent-current-level 0
635 "Current indentation level `python-indent-line-function' is using.")
636
637 (defvar python-indent-levels '(0)
638 "Levels of indentation available for `python-indent-line-function'.")
639
640 (defvar python-indent-dedenters '("else" "elif" "except" "finally")
641 "List of words that should be dedented.
642 These make `python-indent-calculate-indentation' subtract the value of
643 `python-indent-offset'.")
644
645 (defvar python-indent-block-enders '("return" "pass")
646 "List of words that mark the end of a block.
647 These make `python-indent-calculate-indentation' subtract the
648 value of `python-indent-offset' when `python-indent-context' is
649 AFTER-LINE.")
650
651 (defun python-indent-guess-indent-offset ()
652 "Guess and set `python-indent-offset' for the current buffer."
653 (interactive)
654 (save-excursion
655 (save-restriction
656 (widen)
657 (goto-char (point-min))
658 (let ((block-end))
659 (while (and (not block-end)
660 (re-search-forward
661 (python-rx line-start block-start) nil t))
662 (when (and
663 (not (python-syntax-context-type))
664 (progn
665 (goto-char (line-end-position))
666 (python-util-forward-comment -1)
667 (if (equal (char-before) ?:)
668 t
669 (forward-line 1)
670 (when (python-info-block-continuation-line-p)
671 (while (and (python-info-continuation-line-p)
672 (not (eobp)))
673 (forward-line 1))
674 (python-util-forward-comment -1)
675 (when (equal (char-before) ?:)
676 t)))))
677 (setq block-end (point-marker))))
678 (let ((indentation
679 (when block-end
680 (goto-char block-end)
681 (python-util-forward-comment)
682 (current-indentation))))
683 (if indentation
684 (set (make-local-variable 'python-indent-offset) indentation)
685 (message "Can't guess python-indent-offset, using defaults: %s"
686 python-indent-offset)))))))
687
688 (defun python-indent-context ()
689 "Get information on indentation context.
690 Context information is returned with a cons with the form:
691 \(STATUS . START)
692
693 Where status can be any of the following symbols:
694 * inside-paren: If point in between (), {} or []
695 * inside-string: If point is inside a string
696 * after-backslash: Previous line ends in a backslash
697 * after-beginning-of-block: Point is after beginning of block
698 * after-line: Point is after normal line
699 * no-indent: Point is at beginning of buffer or other special case
700 START is the buffer position where the sexp starts."
701 (save-restriction
702 (widen)
703 (let ((ppss (save-excursion (beginning-of-line) (syntax-ppss)))
704 (start))
705 (cons
706 (cond
707 ;; Beginning of buffer
708 ((save-excursion
709 (goto-char (line-beginning-position))
710 (bobp))
711 'no-indent)
712 ;; Inside string
713 ((setq start (python-syntax-context 'string ppss))
714 'inside-string)
715 ;; Inside a paren
716 ((setq start (python-syntax-context 'paren ppss))
717 'inside-paren)
718 ;; After backslash
719 ((setq start (when (not (or (python-syntax-context 'string ppss)
720 (python-syntax-context 'comment ppss)))
721 (let ((line-beg-pos (line-number-at-pos)))
722 (python-info-line-ends-backslash-p
723 (1- line-beg-pos)))))
724 'after-backslash)
725 ;; After beginning of block
726 ((setq start (save-excursion
727 (when (progn
728 (back-to-indentation)
729 (python-util-forward-comment -1)
730 (equal (char-before) ?:))
731 ;; Move to the first block start that's not in within
732 ;; a string, comment or paren and that's not a
733 ;; continuation line.
734 (while (and (re-search-backward
735 (python-rx block-start) nil t)
736 (or
737 (python-syntax-context-type)
738 (python-info-continuation-line-p))))
739 (when (looking-at (python-rx block-start))
740 (point-marker)))))
741 'after-beginning-of-block)
742 ;; After normal line
743 ((setq start (save-excursion
744 (back-to-indentation)
745 (skip-chars-backward (rx (or whitespace ?\n)))
746 (python-nav-beginning-of-statement)
747 (point-marker)))
748 'after-line)
749 ;; Do not indent
750 (t 'no-indent))
751 start))))
752
753 (defun python-indent-calculate-indentation ()
754 "Calculate correct indentation offset for the current line."
755 (let* ((indentation-context (python-indent-context))
756 (context-status (car indentation-context))
757 (context-start (cdr indentation-context)))
758 (save-restriction
759 (widen)
760 (save-excursion
761 (pcase context-status
762 (`no-indent 0)
763 ;; When point is after beginning of block just add one level
764 ;; of indentation relative to the context-start
765 (`after-beginning-of-block
766 (goto-char context-start)
767 (+ (current-indentation) python-indent-offset))
768 ;; When after a simple line just use previous line
769 ;; indentation, in the case current line starts with a
770 ;; `python-indent-dedenters' de-indent one level.
771 (`after-line
772 (-
773 (save-excursion
774 (goto-char context-start)
775 (current-indentation))
776 (if (or (save-excursion
777 (back-to-indentation)
778 (looking-at (regexp-opt python-indent-dedenters)))
779 (save-excursion
780 (python-util-forward-comment -1)
781 (python-nav-beginning-of-statement)
782 (member (current-word) python-indent-block-enders)))
783 python-indent-offset
784 0)))
785 ;; When inside of a string, do nothing. just use the current
786 ;; indentation. XXX: perhaps it would be a good idea to
787 ;; invoke standard text indentation here
788 (`inside-string
789 (goto-char context-start)
790 (current-indentation))
791 ;; After backslash we have several possibilities.
792 (`after-backslash
793 (cond
794 ;; Check if current line is a dot continuation. For this
795 ;; the current line must start with a dot and previous
796 ;; line must contain a dot too.
797 ((save-excursion
798 (back-to-indentation)
799 (when (looking-at "\\.")
800 ;; If after moving one line back point is inside a paren it
801 ;; needs to move back until it's not anymore
802 (while (prog2
803 (forward-line -1)
804 (and (not (bobp))
805 (python-syntax-context 'paren))))
806 (goto-char (line-end-position))
807 (while (and (re-search-backward
808 "\\." (line-beginning-position) t)
809 (python-syntax-context-type)))
810 (if (and (looking-at "\\.")
811 (not (python-syntax-context-type)))
812 ;; The indentation is the same column of the
813 ;; first matching dot that's not inside a
814 ;; comment, a string or a paren
815 (current-column)
816 ;; No dot found on previous line, just add another
817 ;; indentation level.
818 (+ (current-indentation) python-indent-offset)))))
819 ;; Check if prev line is a block continuation
820 ((let ((block-continuation-start
821 (python-info-block-continuation-line-p)))
822 (when block-continuation-start
823 ;; If block-continuation-start is set jump to that
824 ;; marker and use first column after the block start
825 ;; as indentation value.
826 (goto-char block-continuation-start)
827 (re-search-forward
828 (python-rx block-start (* space))
829 (line-end-position) t)
830 (current-column))))
831 ;; Check if current line is an assignment continuation
832 ((let ((assignment-continuation-start
833 (python-info-assignment-continuation-line-p)))
834 (when assignment-continuation-start
835 ;; If assignment-continuation is set jump to that
836 ;; marker and use first column after the assignment
837 ;; operator as indentation value.
838 (goto-char assignment-continuation-start)
839 (current-column))))
840 (t
841 (forward-line -1)
842 (goto-char (python-info-beginning-of-backslash))
843 (if (save-excursion
844 (and
845 (forward-line -1)
846 (goto-char
847 (or (python-info-beginning-of-backslash) (point)))
848 (python-info-line-ends-backslash-p)))
849 ;; The two previous lines ended in a backslash so we must
850 ;; respect previous line indentation.
851 (current-indentation)
852 ;; What happens here is that we are dealing with the second
853 ;; line of a backslash continuation, in that case we just going
854 ;; to add one indentation level.
855 (+ (current-indentation) python-indent-offset)))))
856 ;; When inside a paren there's a need to handle nesting
857 ;; correctly
858 (`inside-paren
859 (cond
860 ;; If current line closes the outermost open paren use the
861 ;; current indentation of the context-start line.
862 ((save-excursion
863 (skip-syntax-forward "\s" (line-end-position))
864 (when (and (looking-at (regexp-opt '(")" "]" "}")))
865 (progn
866 (forward-char 1)
867 (not (python-syntax-context 'paren))))
868 (goto-char context-start)
869 (current-indentation))))
870 ;; If open paren is contained on a line by itself add another
871 ;; indentation level, else look for the first word after the
872 ;; opening paren and use it's column position as indentation
873 ;; level.
874 ((let* ((content-starts-in-newline)
875 (indent
876 (save-excursion
877 (if (setq content-starts-in-newline
878 (progn
879 (goto-char context-start)
880 (forward-char)
881 (save-restriction
882 (narrow-to-region
883 (line-beginning-position)
884 (line-end-position))
885 (python-util-forward-comment))
886 (looking-at "$")))
887 (+ (current-indentation) python-indent-offset)
888 (current-column)))))
889 ;; Adjustments
890 (cond
891 ;; If current line closes a nested open paren de-indent one
892 ;; level.
893 ((progn
894 (back-to-indentation)
895 (looking-at (regexp-opt '(")" "]" "}"))))
896 (- indent python-indent-offset))
897 ;; If the line of the opening paren that wraps the current
898 ;; line starts a block add another level of indentation to
899 ;; follow new pep8 recommendation. See: http://ur1.ca/5rojx
900 ((save-excursion
901 (when (and content-starts-in-newline
902 (progn
903 (goto-char context-start)
904 (back-to-indentation)
905 (looking-at (python-rx block-start))))
906 (+ indent python-indent-offset))))
907 (t indent)))))))))))
908
909 (defun python-indent-calculate-levels ()
910 "Calculate `python-indent-levels' and reset `python-indent-current-level'."
911 (let* ((indentation (python-indent-calculate-indentation))
912 (remainder (% indentation python-indent-offset))
913 (steps (/ (- indentation remainder) python-indent-offset)))
914 (setq python-indent-levels (list 0))
915 (dotimes (step steps)
916 (push (* python-indent-offset (1+ step)) python-indent-levels))
917 (when (not (eq 0 remainder))
918 (push (+ (* python-indent-offset steps) remainder) python-indent-levels))
919 (setq python-indent-levels (nreverse python-indent-levels))
920 (setq python-indent-current-level (1- (length python-indent-levels)))))
921
922 (defun python-indent-toggle-levels ()
923 "Toggle `python-indent-current-level' over `python-indent-levels'."
924 (setq python-indent-current-level (1- python-indent-current-level))
925 (when (< python-indent-current-level 0)
926 (setq python-indent-current-level (1- (length python-indent-levels)))))
927
928 (defun python-indent-line (&optional force-toggle)
929 "Internal implementation of `python-indent-line-function'.
930 Uses the offset calculated in
931 `python-indent-calculate-indentation' and available levels
932 indicated by the variable `python-indent-levels' to set the
933 current indentation.
934
935 When the variable `last-command' is equal to one of the symbols
936 inside `python-indent-trigger-commands' or FORCE-TOGGLE is
937 non-nil it cycles levels indicated in the variable
938 `python-indent-levels' by setting the current level in the
939 variable `python-indent-current-level'.
940
941 When the variable `last-command' is not equal to one of the
942 symbols inside `python-indent-trigger-commands' and FORCE-TOGGLE
943 is nil it calculates possible indentation levels and saves it in
944 the variable `python-indent-levels'. Afterwards it sets the
945 variable `python-indent-current-level' correctly so offset is
946 equal to (`nth' `python-indent-current-level'
947 `python-indent-levels')"
948 (or
949 (and (or (and (memq this-command python-indent-trigger-commands)
950 (eq last-command this-command))
951 force-toggle)
952 (not (equal python-indent-levels '(0)))
953 (or (python-indent-toggle-levels) t))
954 (python-indent-calculate-levels))
955 (let* ((starting-pos (point-marker))
956 (indent-ending-position
957 (+ (line-beginning-position) (current-indentation)))
958 (follow-indentation-p
959 (or (bolp)
960 (and (<= (line-beginning-position) starting-pos)
961 (>= indent-ending-position starting-pos))))
962 (next-indent (nth python-indent-current-level python-indent-levels)))
963 (unless (= next-indent (current-indentation))
964 (beginning-of-line)
965 (delete-horizontal-space)
966 (indent-to next-indent)
967 (goto-char starting-pos))
968 (and follow-indentation-p (back-to-indentation)))
969 (python-info-closing-block-message))
970
971 (defun python-indent-line-function ()
972 "`indent-line-function' for Python mode.
973 See `python-indent-line' for details."
974 (python-indent-line))
975
976 (defun python-indent-dedent-line ()
977 "De-indent current line."
978 (interactive "*")
979 (when (and (not (python-syntax-comment-or-string-p))
980 (<= (point-marker) (save-excursion
981 (back-to-indentation)
982 (point-marker)))
983 (> (current-column) 0))
984 (python-indent-line t)
985 t))
986
987 (defun python-indent-dedent-line-backspace (arg)
988 "De-indent current line.
989 Argument ARG is passed to `backward-delete-char-untabify' when
990 point is not in between the indentation."
991 (interactive "*p")
992 (when (not (python-indent-dedent-line))
993 (backward-delete-char-untabify arg)))
994 (put 'python-indent-dedent-line-backspace 'delete-selection 'supersede)
995
996 (defun python-indent-region (start end)
997 "Indent a python region automagically.
998
999 Called from a program, START and END specify the region to indent."
1000 (let ((deactivate-mark nil))
1001 (save-excursion
1002 (goto-char end)
1003 (setq end (point-marker))
1004 (goto-char start)
1005 (or (bolp) (forward-line 1))
1006 (while (< (point) end)
1007 (or (and (bolp) (eolp))
1008 (let (word)
1009 (forward-line -1)
1010 (back-to-indentation)
1011 (setq word (current-word))
1012 (forward-line 1)
1013 (when (and word
1014 ;; Don't mess with strings, unless it's the
1015 ;; enclosing set of quotes.
1016 (or (not (python-syntax-context 'string))
1017 (eq
1018 (syntax-after
1019 (+ (1- (point))
1020 (current-indentation)
1021 (python-syntax-count-quotes (char-after) (point))))
1022 (string-to-syntax "|"))))
1023 (beginning-of-line)
1024 (delete-horizontal-space)
1025 (indent-to (python-indent-calculate-indentation)))))
1026 (forward-line 1))
1027 (move-marker end nil))))
1028
1029 (defun python-indent-shift-left (start end &optional count)
1030 "Shift lines contained in region START END by COUNT columns to the left.
1031 COUNT defaults to `python-indent-offset'. If region isn't
1032 active, the current line is shifted. The shifted region includes
1033 the lines in which START and END lie. An error is signaled if
1034 any lines in the region are indented less than COUNT columns."
1035 (interactive
1036 (if mark-active
1037 (list (region-beginning) (region-end) current-prefix-arg)
1038 (list (line-beginning-position) (line-end-position) current-prefix-arg)))
1039 (if count
1040 (setq count (prefix-numeric-value count))
1041 (setq count python-indent-offset))
1042 (when (> count 0)
1043 (let ((deactivate-mark nil))
1044 (save-excursion
1045 (goto-char start)
1046 (while (< (point) end)
1047 (if (and (< (current-indentation) count)
1048 (not (looking-at "[ \t]*$")))
1049 (error "Can't shift all lines enough"))
1050 (forward-line))
1051 (indent-rigidly start end (- count))))))
1052
1053 (add-to-list 'debug-ignored-errors "^Can't shift all lines enough")
1054
1055 (defun python-indent-shift-right (start end &optional count)
1056 "Shift lines contained in region START END by COUNT columns to the left.
1057 COUNT defaults to `python-indent-offset'. If region isn't
1058 active, the current line is shifted. The shifted region includes
1059 the lines in which START and END lie."
1060 (interactive
1061 (if mark-active
1062 (list (region-beginning) (region-end) current-prefix-arg)
1063 (list (line-beginning-position) (line-end-position) current-prefix-arg)))
1064 (let ((deactivate-mark nil))
1065 (if count
1066 (setq count (prefix-numeric-value count))
1067 (setq count python-indent-offset))
1068 (indent-rigidly start end count)))
1069
1070 (defun python-indent-electric-colon (arg)
1071 "Insert a colon and maybe de-indent the current line.
1072 With numeric ARG, just insert that many colons. With
1073 \\[universal-argument], just insert a single colon."
1074 (interactive "*P")
1075 (self-insert-command (if (not (integerp arg)) 1 arg))
1076 (when (and (not arg)
1077 (eolp)
1078 (not (equal ?: (char-after (- (point-marker) 2))))
1079 (not (python-syntax-comment-or-string-p)))
1080 (let ((indentation (current-indentation))
1081 (calculated-indentation (python-indent-calculate-indentation)))
1082 (python-info-closing-block-message)
1083 (when (> indentation calculated-indentation)
1084 (save-excursion
1085 (indent-line-to calculated-indentation)
1086 (when (not (python-info-closing-block-message))
1087 (indent-line-to indentation)))))))
1088 (put 'python-indent-electric-colon 'delete-selection t)
1089
1090 (defun python-indent-post-self-insert-function ()
1091 "Adjust closing paren line indentation after a char is added.
1092 This function is intended to be added to the
1093 `post-self-insert-hook.' If a line renders a paren alone, after
1094 adding a char before it, the line will be re-indented
1095 automatically if needed."
1096 (when (and (eq (char-before) last-command-event)
1097 (not (bolp))
1098 (memq (char-after) '(?\) ?\] ?\})))
1099 (save-excursion
1100 (goto-char (line-beginning-position))
1101 ;; If after going to the beginning of line the point
1102 ;; is still inside a paren it's ok to do the trick
1103 (when (python-syntax-context 'paren)
1104 (let ((indentation (python-indent-calculate-indentation)))
1105 (when (< (current-indentation) indentation)
1106 (indent-line-to indentation)))))))
1107
1108 \f
1109 ;;; Navigation
1110
1111 (defvar python-nav-beginning-of-defun-regexp
1112 (python-rx line-start (* space) defun (+ space) (group symbol-name))
1113 "Regexp matching class or function definition.
1114 The name of the defun should be grouped so it can be retrieved
1115 via `match-string'.")
1116
1117 (defun python-nav--beginning-of-defun (&optional arg)
1118 "Internal implementation of `python-nav-beginning-of-defun'.
1119 With positive ARG search backwards, else search forwards."
1120 (when (or (null arg) (= arg 0)) (setq arg 1))
1121 (let* ((re-search-fn (if (> arg 0)
1122 #'re-search-backward
1123 #'re-search-forward))
1124 (line-beg-pos (line-beginning-position))
1125 (line-content-start (+ line-beg-pos (current-indentation)))
1126 (pos (point-marker))
1127 (beg-indentation
1128 (and (> arg 0)
1129 (save-excursion
1130 (while (and
1131 (not (python-info-looking-at-beginning-of-defun))
1132 (python-nav-backward-block)))
1133 (or (and (python-info-looking-at-beginning-of-defun)
1134 (+ (current-indentation) python-indent-offset))
1135 0))))
1136 (found
1137 (progn
1138 (when (and (< arg 0)
1139 (python-info-looking-at-beginning-of-defun))
1140 (end-of-line 1))
1141 (while (and (funcall re-search-fn
1142 python-nav-beginning-of-defun-regexp nil t)
1143 (or (python-syntax-context-type)
1144 ;; Handle nested defuns when moving
1145 ;; backwards by checking indentation.
1146 (and (> arg 0)
1147 (not (= (current-indentation) 0))
1148 (>= (current-indentation) beg-indentation)))))
1149 (and (python-info-looking-at-beginning-of-defun)
1150 (or (not (= (line-number-at-pos pos)
1151 (line-number-at-pos)))
1152 (and (>= (point) line-beg-pos)
1153 (<= (point) line-content-start)
1154 (> pos line-content-start)))))))
1155 (if found
1156 (or (beginning-of-line 1) t)
1157 (and (goto-char pos) nil))))
1158
1159 (defun python-nav-beginning-of-defun (&optional arg)
1160 "Move point to `beginning-of-defun'.
1161 With positive ARG search backwards else search forward. When ARG
1162 is nil or 0 defaults to 1. When searching backwards nested
1163 defuns are handled with care depending on current point
1164 position. Return non-nil if point is moved to
1165 `beginning-of-defun'."
1166 (when (or (null arg) (= arg 0)) (setq arg 1))
1167 (let ((found))
1168 (cond ((and (eq this-command 'mark-defun)
1169 (python-info-looking-at-beginning-of-defun)))
1170 (t
1171 (dotimes (i (if (> arg 0) arg (- arg)))
1172 (when (and (python-nav--beginning-of-defun arg)
1173 (not found))
1174 (setq found t)))))
1175 found))
1176
1177 (defun python-nav-end-of-defun ()
1178 "Move point to the end of def or class.
1179 Returns nil if point is not in a def or class."
1180 (interactive)
1181 (let ((beg-defun-indent)
1182 (beg-pos (point)))
1183 (when (or (python-info-looking-at-beginning-of-defun)
1184 (python-nav-beginning-of-defun 1)
1185 (python-nav-beginning-of-defun -1))
1186 (setq beg-defun-indent (current-indentation))
1187 (while (progn
1188 (python-nav-end-of-statement)
1189 (python-util-forward-comment 1)
1190 (and (> (current-indentation) beg-defun-indent)
1191 (not (eobp)))))
1192 (python-util-forward-comment -1)
1193 (forward-line 1)
1194 ;; Ensure point moves forward.
1195 (and (> beg-pos (point)) (goto-char beg-pos)))))
1196
1197 (defun python-nav--syntactically (fn poscompfn &optional contextfn)
1198 "Move point using FN avoiding places with specific context.
1199 FN must take no arguments. POSCOMPFN is a two arguments function
1200 used to compare current and previous point after it is moved
1201 using FN, this is normally a less-than or greater-than
1202 comparison. Optional argument CONTEXTFN defaults to
1203 `python-syntax-context-type' and is used for checking current
1204 point context, it must return a non-nil value if this point must
1205 be skipped."
1206 (let ((contextfn (or contextfn 'python-syntax-context-type))
1207 (start-pos (point-marker))
1208 (prev-pos))
1209 (catch 'found
1210 (while t
1211 (let* ((newpos
1212 (and (funcall fn) (point-marker)))
1213 (context (funcall contextfn)))
1214 (cond ((and (not context) newpos
1215 (or (and (not prev-pos) newpos)
1216 (and prev-pos newpos
1217 (funcall poscompfn newpos prev-pos))))
1218 (throw 'found (point-marker)))
1219 ((and newpos context)
1220 (setq prev-pos (point)))
1221 (t (when (not newpos) (goto-char start-pos))
1222 (throw 'found nil))))))))
1223
1224 (defun python-nav--forward-defun (arg)
1225 "Internal implementation of python-nav-{backward,forward}-defun.
1226 Uses ARG to define which function to call, and how many times
1227 repeat it."
1228 (let ((found))
1229 (while (and (> arg 0)
1230 (setq found
1231 (python-nav--syntactically
1232 (lambda ()
1233 (re-search-forward
1234 python-nav-beginning-of-defun-regexp nil t))
1235 '>)))
1236 (setq arg (1- arg)))
1237 (while (and (< arg 0)
1238 (setq found
1239 (python-nav--syntactically
1240 (lambda ()
1241 (re-search-backward
1242 python-nav-beginning-of-defun-regexp nil t))
1243 '<)))
1244 (setq arg (1+ arg)))
1245 found))
1246
1247 (defun python-nav-backward-defun (&optional arg)
1248 "Navigate to closer defun backward ARG times.
1249 Unlikely `python-nav-beginning-of-defun' this doesn't care about
1250 nested definitions."
1251 (interactive "^p")
1252 (python-nav--forward-defun (- (or arg 1))))
1253
1254 (defun python-nav-forward-defun (&optional arg)
1255 "Navigate to closer defun forward ARG times.
1256 Unlikely `python-nav-beginning-of-defun' this doesn't care about
1257 nested definitions."
1258 (interactive "^p")
1259 (python-nav--forward-defun (or arg 1)))
1260
1261 (defun python-nav-beginning-of-statement ()
1262 "Move to start of current statement."
1263 (interactive "^")
1264 (while (and (or (back-to-indentation) t)
1265 (not (bobp))
1266 (when (or
1267 (save-excursion
1268 (forward-line -1)
1269 (python-info-line-ends-backslash-p))
1270 (python-syntax-context 'string)
1271 (python-syntax-context 'paren))
1272 (forward-line -1))))
1273 (point-marker))
1274
1275 (defun python-nav-end-of-statement (&optional noend)
1276 "Move to end of current statement.
1277 Optional argument NOEND is internal and makes the logic to not
1278 jump to the end of line when moving forward searching for the end
1279 of the statement."
1280 (interactive "^")
1281 (let (string-start bs-pos)
1282 (while (and (or noend (goto-char (line-end-position)))
1283 (not (eobp))
1284 (cond ((setq string-start (python-syntax-context 'string))
1285 (goto-char string-start)
1286 (if (python-syntax-context 'paren)
1287 ;; Ended up inside a paren, roll again.
1288 (python-nav-end-of-statement t)
1289 ;; This is not inside a paren, move to the
1290 ;; end of this string.
1291 (goto-char (+ (point)
1292 (python-syntax-count-quotes
1293 (char-after (point)) (point))))
1294 (or (re-search-forward (rx (syntax string-delimiter)) nil t)
1295 (goto-char (point-max)))))
1296 ((python-syntax-context 'paren)
1297 ;; The statement won't end before we've escaped
1298 ;; at least one level of parenthesis.
1299 (condition-case err
1300 (goto-char (scan-lists (point) 1 -1))
1301 (scan-error (goto-char (nth 3 err)))))
1302 ((setq bs-pos (python-info-line-ends-backslash-p))
1303 (goto-char bs-pos)
1304 (forward-line 1))))))
1305 (point-marker))
1306
1307 (defun python-nav-backward-statement (&optional arg)
1308 "Move backward to previous statement.
1309 With ARG, repeat. See `python-nav-forward-statement'."
1310 (interactive "^p")
1311 (or arg (setq arg 1))
1312 (python-nav-forward-statement (- arg)))
1313
1314 (defun python-nav-forward-statement (&optional arg)
1315 "Move forward to next statement.
1316 With ARG, repeat. With negative argument, move ARG times
1317 backward to previous statement."
1318 (interactive "^p")
1319 (or arg (setq arg 1))
1320 (while (> arg 0)
1321 (python-nav-end-of-statement)
1322 (python-util-forward-comment)
1323 (python-nav-beginning-of-statement)
1324 (setq arg (1- arg)))
1325 (while (< arg 0)
1326 (python-nav-beginning-of-statement)
1327 (python-util-forward-comment -1)
1328 (python-nav-beginning-of-statement)
1329 (setq arg (1+ arg))))
1330
1331 (defun python-nav-beginning-of-block ()
1332 "Move to start of current block."
1333 (interactive "^")
1334 (let ((starting-pos (point))
1335 (block-regexp (python-rx
1336 line-start (* whitespace) block-start)))
1337 (if (progn
1338 (python-nav-beginning-of-statement)
1339 (looking-at (python-rx block-start)))
1340 (point-marker)
1341 ;; Go to first line beginning a statement
1342 (while (and (not (bobp))
1343 (or (and (python-nav-beginning-of-statement) nil)
1344 (python-info-current-line-comment-p)
1345 (python-info-current-line-empty-p)))
1346 (forward-line -1))
1347 (let ((block-matching-indent
1348 (- (current-indentation) python-indent-offset)))
1349 (while
1350 (and (python-nav-backward-block)
1351 (> (current-indentation) block-matching-indent)))
1352 (if (and (looking-at (python-rx block-start))
1353 (= (current-indentation) block-matching-indent))
1354 (point-marker)
1355 (and (goto-char starting-pos) nil))))))
1356
1357 (defun python-nav-end-of-block ()
1358 "Move to end of current block."
1359 (interactive "^")
1360 (when (python-nav-beginning-of-block)
1361 (let ((block-indentation (current-indentation)))
1362 (python-nav-end-of-statement)
1363 (while (and (forward-line 1)
1364 (not (eobp))
1365 (or (and (> (current-indentation) block-indentation)
1366 (or (python-nav-end-of-statement) t))
1367 (python-info-current-line-comment-p)
1368 (python-info-current-line-empty-p))))
1369 (python-util-forward-comment -1)
1370 (point-marker))))
1371
1372 (defun python-nav-backward-block (&optional arg)
1373 "Move backward to previous block of code.
1374 With ARG, repeat. See `python-nav-forward-block'."
1375 (interactive "^p")
1376 (or arg (setq arg 1))
1377 (python-nav-forward-block (- arg)))
1378
1379 (defun python-nav-forward-block (&optional arg)
1380 "Move forward to next block of code.
1381 With ARG, repeat. With negative argument, move ARG times
1382 backward to previous block."
1383 (interactive "^p")
1384 (or arg (setq arg 1))
1385 (let ((block-start-regexp
1386 (python-rx line-start (* whitespace) block-start))
1387 (starting-pos (point)))
1388 (while (> arg 0)
1389 (python-nav-end-of-statement)
1390 (while (and
1391 (re-search-forward block-start-regexp nil t)
1392 (python-syntax-context-type)))
1393 (setq arg (1- arg)))
1394 (while (< arg 0)
1395 (python-nav-beginning-of-statement)
1396 (while (and
1397 (re-search-backward block-start-regexp nil t)
1398 (python-syntax-context-type)))
1399 (setq arg (1+ arg)))
1400 (python-nav-beginning-of-statement)
1401 (if (not (looking-at (python-rx block-start)))
1402 (and (goto-char starting-pos) nil)
1403 (and (not (= (point) starting-pos)) (point-marker)))))
1404
1405 (defun python-nav-lisp-forward-sexp-safe (&optional arg)
1406 "Safe version of standard `forward-sexp'.
1407 When ARG > 0 move forward, else if ARG is < 0."
1408 (or arg (setq arg 1))
1409 (let ((forward-sexp-function)
1410 (paren-regexp
1411 (if (> arg 0) (python-rx close-paren) (python-rx open-paren)))
1412 (search-fn
1413 (if (> arg 0) #'re-search-forward #'re-search-backward)))
1414 (condition-case nil
1415 (forward-sexp arg)
1416 (error
1417 (while (and (funcall search-fn paren-regexp nil t)
1418 (python-syntax-context 'paren)))))))
1419
1420 (defun python-nav--forward-sexp (&optional dir)
1421 "Move to forward sexp.
1422 With positive Optional argument DIR direction move forward, else
1423 backwards."
1424 (setq dir (or dir 1))
1425 (unless (= dir 0)
1426 (let* ((forward-p (if (> dir 0)
1427 (and (setq dir 1) t)
1428 (and (setq dir -1) nil)))
1429 (re-search-fn (if forward-p
1430 're-search-forward
1431 're-search-backward))
1432 (context-type (python-syntax-context-type)))
1433 (cond
1434 ((memq context-type '(string comment))
1435 ;; Inside of a string, get out of it.
1436 (let ((forward-sexp-function))
1437 (forward-sexp dir)))
1438 ((or (eq context-type 'paren)
1439 (and forward-p (looking-at (python-rx open-paren)))
1440 (and (not forward-p)
1441 (eq (syntax-class (syntax-after (1- (point))))
1442 (car (string-to-syntax ")")))))
1443 ;; Inside a paren or looking at it, lisp knows what to do.
1444 (python-nav-lisp-forward-sexp-safe dir))
1445 (t
1446 ;; This part handles the lispy feel of
1447 ;; `python-nav-forward-sexp'. Knowing everything about the
1448 ;; current context and the context of the next sexp tries to
1449 ;; follow the lisp sexp motion commands in a symmetric manner.
1450 (let* ((context
1451 (cond
1452 ((python-info-beginning-of-block-p) 'block-start)
1453 ((python-info-end-of-block-p) 'block-end)
1454 ((python-info-beginning-of-statement-p) 'statement-start)
1455 ((python-info-end-of-statement-p) 'statement-end)))
1456 (next-sexp-pos
1457 (save-excursion
1458 (python-nav-lisp-forward-sexp-safe dir)
1459 (point)))
1460 (next-sexp-context
1461 (save-excursion
1462 (goto-char next-sexp-pos)
1463 (cond
1464 ((python-info-beginning-of-block-p) 'block-start)
1465 ((python-info-end-of-block-p) 'block-end)
1466 ((python-info-beginning-of-statement-p) 'statement-start)
1467 ((python-info-end-of-statement-p) 'statement-end)
1468 ((python-info-statement-starts-block-p) 'starts-block)
1469 ((python-info-statement-ends-block-p) 'ends-block)))))
1470 (if forward-p
1471 (cond ((and (not (eobp))
1472 (python-info-current-line-empty-p))
1473 (python-util-forward-comment dir)
1474 (python-nav--forward-sexp dir))
1475 ((eq context 'block-start)
1476 (python-nav-end-of-block))
1477 ((eq context 'statement-start)
1478 (python-nav-end-of-statement))
1479 ((and (memq context '(statement-end block-end))
1480 (eq next-sexp-context 'ends-block))
1481 (goto-char next-sexp-pos)
1482 (python-nav-end-of-block))
1483 ((and (memq context '(statement-end block-end))
1484 (eq next-sexp-context 'starts-block))
1485 (goto-char next-sexp-pos)
1486 (python-nav-end-of-block))
1487 ((memq context '(statement-end block-end))
1488 (goto-char next-sexp-pos)
1489 (python-nav-end-of-statement))
1490 (t (goto-char next-sexp-pos)))
1491 (cond ((and (not (bobp))
1492 (python-info-current-line-empty-p))
1493 (python-util-forward-comment dir)
1494 (python-nav--forward-sexp dir))
1495 ((eq context 'block-end)
1496 (python-nav-beginning-of-block))
1497 ((eq context 'statement-end)
1498 (python-nav-beginning-of-statement))
1499 ((and (memq context '(statement-start block-start))
1500 (eq next-sexp-context 'starts-block))
1501 (goto-char next-sexp-pos)
1502 (python-nav-beginning-of-block))
1503 ((and (memq context '(statement-start block-start))
1504 (eq next-sexp-context 'ends-block))
1505 (goto-char next-sexp-pos)
1506 (python-nav-beginning-of-block))
1507 ((memq context '(statement-start block-start))
1508 (goto-char next-sexp-pos)
1509 (python-nav-beginning-of-statement))
1510 (t (goto-char next-sexp-pos))))))))))
1511
1512 (defun python-nav--backward-sexp ()
1513 "Move to backward sexp."
1514 (python-nav--forward-sexp -1))
1515
1516 (defun python-nav-forward-sexp (&optional arg)
1517 "Move forward across one block of code.
1518 With ARG, do it that many times. Negative arg -N means
1519 move backward N times."
1520 (interactive "^p")
1521 (or arg (setq arg 1))
1522 (while (> arg 0)
1523 (python-nav--forward-sexp)
1524 (setq arg (1- arg)))
1525 (while (< arg 0)
1526 (python-nav--backward-sexp)
1527 (setq arg (1+ arg))))
1528
1529 (defun python-nav--up-list (&optional dir)
1530 "Internal implementation of `python-nav-up-list'.
1531 DIR is always 1 or -1 and comes sanitized from
1532 `python-nav-up-list' calls."
1533 (let ((context (python-syntax-context-type))
1534 (forward-p (> dir 0)))
1535 (cond
1536 ((memq context '(string comment)))
1537 ((eq context 'paren)
1538 (let ((forward-sexp-function))
1539 (up-list dir)))
1540 ((and forward-p (python-info-end-of-block-p))
1541 (let ((parent-end-pos
1542 (save-excursion
1543 (let ((indentation (and
1544 (python-nav-beginning-of-block)
1545 (current-indentation))))
1546 (while (and indentation
1547 (> indentation 0)
1548 (>= (current-indentation) indentation)
1549 (python-nav-backward-block)))
1550 (python-nav-end-of-block)))))
1551 (and (> (or parent-end-pos (point)) (point))
1552 (goto-char parent-end-pos))))
1553 (forward-p (python-nav-end-of-block))
1554 ((and (not forward-p)
1555 (> (current-indentation) 0)
1556 (python-info-beginning-of-block-p))
1557 (let ((prev-block-pos
1558 (save-excursion
1559 (let ((indentation (current-indentation)))
1560 (while (and (python-nav-backward-block)
1561 (>= (current-indentation) indentation))))
1562 (point))))
1563 (and (> (point) prev-block-pos)
1564 (goto-char prev-block-pos))))
1565 ((not forward-p) (python-nav-beginning-of-block)))))
1566
1567 (defun python-nav-up-list (&optional arg)
1568 "Move forward out of one level of parentheses (or blocks).
1569 With ARG, do this that many times.
1570 A negative argument means move backward but still to a less deep spot.
1571 This command assumes point is not in a string or comment."
1572 (interactive "^p")
1573 (or arg (setq arg 1))
1574 (while (> arg 0)
1575 (python-nav--up-list 1)
1576 (setq arg (1- arg)))
1577 (while (< arg 0)
1578 (python-nav--up-list -1)
1579 (setq arg (1+ arg))))
1580
1581 (defun python-nav-backward-up-list (&optional arg)
1582 "Move backward out of one level of parentheses (or blocks).
1583 With ARG, do this that many times.
1584 A negative argument means move backward but still to a less deep spot.
1585 This command assumes point is not in a string or comment."
1586 (interactive "^p")
1587 (or arg (setq arg 1))
1588 (python-nav-up-list (- arg)))
1589
1590 \f
1591 ;;; Shell integration
1592
1593 (defcustom python-shell-buffer-name "Python"
1594 "Default buffer name for Python interpreter."
1595 :type 'string
1596 :group 'python
1597 :safe 'stringp)
1598
1599 (defcustom python-shell-interpreter "python"
1600 "Default Python interpreter for shell."
1601 :type 'string
1602 :group 'python)
1603
1604 (defcustom python-shell-internal-buffer-name "Python Internal"
1605 "Default buffer name for the Internal Python interpreter."
1606 :type 'string
1607 :group 'python
1608 :safe 'stringp)
1609
1610 (defcustom python-shell-interpreter-args "-i"
1611 "Default arguments for the Python interpreter."
1612 :type 'string
1613 :group 'python)
1614
1615 (defcustom python-shell-prompt-regexp ">>> "
1616 "Regular Expression matching top\-level input prompt of python shell.
1617 It should not contain a caret (^) at the beginning."
1618 :type 'string
1619 :group 'python
1620 :safe 'stringp)
1621
1622 (defcustom python-shell-prompt-block-regexp "[.][.][.] "
1623 "Regular Expression matching block input prompt of python shell.
1624 It should not contain a caret (^) at the beginning."
1625 :type 'string
1626 :group 'python
1627 :safe 'stringp)
1628
1629 (defcustom python-shell-prompt-output-regexp ""
1630 "Regular Expression matching output prompt of python shell.
1631 It should not contain a caret (^) at the beginning."
1632 :type 'string
1633 :group 'python
1634 :safe 'stringp)
1635
1636 (defcustom python-shell-prompt-pdb-regexp "[(<]*[Ii]?[Pp]db[>)]+ "
1637 "Regular Expression matching pdb input prompt of python shell.
1638 It should not contain a caret (^) at the beginning."
1639 :type 'string
1640 :group 'python
1641 :safe 'stringp)
1642
1643 (defcustom python-shell-enable-font-lock t
1644 "Should syntax highlighting be enabled in the python shell buffer?
1645 Restart the python shell after changing this variable for it to take effect."
1646 :type 'boolean
1647 :group 'python
1648 :safe 'booleanp)
1649
1650 (defcustom python-shell-process-environment nil
1651 "List of environment variables for Python shell.
1652 This variable follows the same rules as `process-environment'
1653 since it merges with it before the process creation routines are
1654 called. When this variable is nil, the Python shell is run with
1655 the default `process-environment'."
1656 :type '(repeat string)
1657 :group 'python
1658 :safe 'listp)
1659
1660 (defcustom python-shell-extra-pythonpaths nil
1661 "List of extra pythonpaths for Python shell.
1662 The values of this variable are added to the existing value of
1663 PYTHONPATH in the `process-environment' variable."
1664 :type '(repeat string)
1665 :group 'python
1666 :safe 'listp)
1667
1668 (defcustom python-shell-exec-path nil
1669 "List of path to search for binaries.
1670 This variable follows the same rules as `exec-path' since it
1671 merges with it before the process creation routines are called.
1672 When this variable is nil, the Python shell is run with the
1673 default `exec-path'."
1674 :type '(repeat string)
1675 :group 'python
1676 :safe 'listp)
1677
1678 (defcustom python-shell-virtualenv-path nil
1679 "Path to virtualenv root.
1680 This variable, when set to a string, makes the values stored in
1681 `python-shell-process-environment' and `python-shell-exec-path'
1682 to be modified properly so shells are started with the specified
1683 virtualenv."
1684 :type '(choice (const nil) string)
1685 :group 'python
1686 :safe 'stringp)
1687
1688 (defcustom python-shell-setup-codes '(python-shell-completion-setup-code
1689 python-ffap-setup-code
1690 python-eldoc-setup-code)
1691 "List of code run by `python-shell-send-setup-codes'."
1692 :type '(repeat symbol)
1693 :group 'python
1694 :safe 'listp)
1695
1696 (defcustom python-shell-compilation-regexp-alist
1697 `((,(rx line-start (1+ (any " \t")) "File \""
1698 (group (1+ (not (any "\"<")))) ; avoid `<stdin>' &c
1699 "\", line " (group (1+ digit)))
1700 1 2)
1701 (,(rx " in file " (group (1+ not-newline)) " on line "
1702 (group (1+ digit)))
1703 1 2)
1704 (,(rx line-start "> " (group (1+ (not (any "(\"<"))))
1705 "(" (group (1+ digit)) ")" (1+ (not (any "("))) "()")
1706 1 2))
1707 "`compilation-error-regexp-alist' for inferior Python."
1708 :type '(alist string)
1709 :group 'python)
1710
1711 (defun python-shell-get-process-name (dedicated)
1712 "Calculate the appropriate process name for inferior Python process.
1713 If DEDICATED is t and the variable `buffer-file-name' is non-nil
1714 returns a string with the form
1715 `python-shell-buffer-name'[variable `buffer-file-name'] else
1716 returns the value of `python-shell-buffer-name'."
1717 (let ((process-name
1718 (if (and dedicated
1719 buffer-file-name)
1720 (format "%s[%s]" python-shell-buffer-name buffer-file-name)
1721 (format "%s" python-shell-buffer-name))))
1722 process-name))
1723
1724 (defun python-shell-internal-get-process-name ()
1725 "Calculate the appropriate process name for Internal Python process.
1726 The name is calculated from `python-shell-global-buffer-name' and
1727 a hash of all relevant global shell settings in order to ensure
1728 uniqueness for different types of configurations."
1729 (format "%s [%s]"
1730 python-shell-internal-buffer-name
1731 (md5
1732 (concat
1733 (python-shell-parse-command)
1734 python-shell-prompt-regexp
1735 python-shell-prompt-block-regexp
1736 python-shell-prompt-output-regexp
1737 (mapconcat #'symbol-value python-shell-setup-codes "")
1738 (mapconcat #'identity python-shell-process-environment "")
1739 (mapconcat #'identity python-shell-extra-pythonpaths "")
1740 (mapconcat #'identity python-shell-exec-path "")
1741 (or python-shell-virtualenv-path "")
1742 (mapconcat #'identity python-shell-exec-path "")))))
1743
1744 (defun python-shell-parse-command ()
1745 "Calculate the string used to execute the inferior Python process."
1746 (let ((process-environment (python-shell-calculate-process-environment))
1747 (exec-path (python-shell-calculate-exec-path)))
1748 (format "%s %s"
1749 (executable-find python-shell-interpreter)
1750 python-shell-interpreter-args)))
1751
1752 (defun python-shell-calculate-process-environment ()
1753 "Calculate process environment given `python-shell-virtualenv-path'."
1754 (let ((process-environment (append
1755 python-shell-process-environment
1756 process-environment nil))
1757 (virtualenv (if python-shell-virtualenv-path
1758 (directory-file-name python-shell-virtualenv-path)
1759 nil)))
1760 (when python-shell-extra-pythonpaths
1761 (setenv "PYTHONPATH"
1762 (format "%s%s%s"
1763 (mapconcat 'identity
1764 python-shell-extra-pythonpaths
1765 path-separator)
1766 path-separator
1767 (or (getenv "PYTHONPATH") ""))))
1768 (if (not virtualenv)
1769 process-environment
1770 (setenv "PYTHONHOME" nil)
1771 (setenv "PATH" (format "%s/bin%s%s"
1772 virtualenv path-separator
1773 (or (getenv "PATH") "")))
1774 (setenv "VIRTUAL_ENV" virtualenv))
1775 process-environment))
1776
1777 (defun python-shell-calculate-exec-path ()
1778 "Calculate exec path given `python-shell-virtualenv-path'."
1779 (let ((path (append python-shell-exec-path
1780 exec-path nil)))
1781 (if (not python-shell-virtualenv-path)
1782 path
1783 (cons (format "%s/bin"
1784 (directory-file-name python-shell-virtualenv-path))
1785 path))))
1786
1787 (defun python-comint-output-filter-function (output)
1788 "Hook run after content is put into comint buffer.
1789 OUTPUT is a string with the contents of the buffer."
1790 (ansi-color-filter-apply output))
1791
1792 (defvar python-shell--parent-buffer nil)
1793
1794 (defvar python-shell-output-syntax-table
1795 (let ((table (make-syntax-table python-dotty-syntax-table)))
1796 (modify-syntax-entry ?\' "." table)
1797 (modify-syntax-entry ?\" "." table)
1798 (modify-syntax-entry ?\( "." table)
1799 (modify-syntax-entry ?\[ "." table)
1800 (modify-syntax-entry ?\{ "." table)
1801 (modify-syntax-entry ?\) "." table)
1802 (modify-syntax-entry ?\] "." table)
1803 (modify-syntax-entry ?\} "." table)
1804 table)
1805 "Syntax table for shell output.
1806 It makes parens and quotes be treated as punctuation chars.")
1807
1808 (define-derived-mode inferior-python-mode comint-mode "Inferior Python"
1809 "Major mode for Python inferior process.
1810 Runs a Python interpreter as a subprocess of Emacs, with Python
1811 I/O through an Emacs buffer. Variables
1812 `python-shell-interpreter' and `python-shell-interpreter-args'
1813 controls which Python interpreter is run. Variables
1814 `python-shell-prompt-regexp',
1815 `python-shell-prompt-output-regexp',
1816 `python-shell-prompt-block-regexp',
1817 `python-shell-enable-font-lock',
1818 `python-shell-completion-setup-code',
1819 `python-shell-completion-string-code',
1820 `python-shell-completion-module-string-code',
1821 `python-eldoc-setup-code', `python-eldoc-string-code',
1822 `python-ffap-setup-code' and `python-ffap-string-code' can
1823 customize this mode for different Python interpreters.
1824
1825 You can also add additional setup code to be run at
1826 initialization of the interpreter via `python-shell-setup-codes'
1827 variable.
1828
1829 \(Type \\[describe-mode] in the process buffer for a list of commands.)"
1830 (and python-shell--parent-buffer
1831 (python-util-clone-local-variables python-shell--parent-buffer))
1832 (setq comint-prompt-regexp (format "^\\(?:%s\\|%s\\|%s\\)"
1833 python-shell-prompt-regexp
1834 python-shell-prompt-block-regexp
1835 python-shell-prompt-pdb-regexp))
1836 (setq mode-line-process '(":%s"))
1837 (make-local-variable 'comint-output-filter-functions)
1838 (add-hook 'comint-output-filter-functions
1839 'python-comint-output-filter-function)
1840 (add-hook 'comint-output-filter-functions
1841 'python-pdbtrack-comint-output-filter-function)
1842 (set (make-local-variable 'compilation-error-regexp-alist)
1843 python-shell-compilation-regexp-alist)
1844 (define-key inferior-python-mode-map [remap complete-symbol]
1845 'completion-at-point)
1846 (add-hook 'completion-at-point-functions
1847 'python-shell-completion-complete-at-point nil 'local)
1848 (add-to-list (make-local-variable 'comint-dynamic-complete-functions)
1849 'python-shell-completion-complete-at-point)
1850 (define-key inferior-python-mode-map "\t"
1851 'python-shell-completion-complete-or-indent)
1852 (make-local-variable 'python-pdbtrack-buffers-to-kill)
1853 (make-local-variable 'python-pdbtrack-tracked-buffer)
1854 (make-local-variable 'python-shell-internal-last-output)
1855 (when python-shell-enable-font-lock
1856 (set-syntax-table python-mode-syntax-table)
1857 (set (make-local-variable 'font-lock-defaults)
1858 '(python-font-lock-keywords nil nil nil nil))
1859 (set (make-local-variable 'syntax-propertize-function)
1860 (eval
1861 ;; XXX: Unfortunately eval is needed here to make use of the
1862 ;; dynamic value of `comint-prompt-regexp'.
1863 `(syntax-propertize-rules
1864 (,comint-prompt-regexp
1865 (0 (ignore
1866 (put-text-property
1867 comint-last-input-start end 'syntax-table
1868 python-shell-output-syntax-table)
1869 ;; XXX: This might look weird, but it is the easiest
1870 ;; way to ensure font lock gets cleaned up before the
1871 ;; current prompt, which is needed for unclosed
1872 ;; strings to not mess up with current input.
1873 (font-lock-unfontify-region comint-last-input-start end))))
1874 (,(python-rx string-delimiter)
1875 (0 (ignore
1876 (and (not (eq (get-text-property start 'field) 'output))
1877 (python-syntax-stringify)))))))))
1878 (compilation-shell-minor-mode 1))
1879
1880 (defun python-shell-make-comint (cmd proc-name &optional pop internal)
1881 "Create a python shell comint buffer.
1882 CMD is the python command to be executed and PROC-NAME is the
1883 process name the comint buffer will get. After the comint buffer
1884 is created the `inferior-python-mode' is activated. When
1885 optional argument POP is non-nil the buffer is shown. When
1886 optional argument INTERNAL is non-nil this process is run on a
1887 buffer with a name that starts with a space, following the Emacs
1888 convention for temporary/internal buffers, and also makes sure
1889 the user is not queried for confirmation when the process is
1890 killed."
1891 (save-excursion
1892 (let* ((proc-buffer-name
1893 (format (if (not internal) "*%s*" " *%s*") proc-name))
1894 (process-environment (python-shell-calculate-process-environment))
1895 (exec-path (python-shell-calculate-exec-path)))
1896 (when (not (comint-check-proc proc-buffer-name))
1897 (let* ((cmdlist (split-string-and-unquote cmd))
1898 (buffer (apply #'make-comint-in-buffer proc-name proc-buffer-name
1899 (car cmdlist) nil (cdr cmdlist)))
1900 (python-shell--parent-buffer (current-buffer))
1901 (process (get-buffer-process buffer)))
1902 (with-current-buffer buffer
1903 (inferior-python-mode))
1904 (accept-process-output process)
1905 (and pop (pop-to-buffer buffer t))
1906 (and internal (set-process-query-on-exit-flag process nil))))
1907 proc-buffer-name)))
1908
1909 ;;;###autoload
1910 (defun run-python (cmd &optional dedicated show)
1911 "Run an inferior Python process.
1912 Input and output via buffer named after
1913 `python-shell-buffer-name'. If there is a process already
1914 running in that buffer, just switch to it.
1915
1916 With argument, allows you to define CMD so you can edit the
1917 command used to call the interpreter and define DEDICATED, so a
1918 dedicated process for the current buffer is open. When numeric
1919 prefix arg is other than 0 or 4 do not SHOW.
1920
1921 Runs the hook `inferior-python-mode-hook' (after the
1922 `comint-mode-hook' is run). \(Type \\[describe-mode] in the
1923 process buffer for a list of commands.)"
1924 (interactive
1925 (if current-prefix-arg
1926 (list
1927 (read-string "Run Python: " (python-shell-parse-command))
1928 (y-or-n-p "Make dedicated process? ")
1929 (= (prefix-numeric-value current-prefix-arg) 4))
1930 (list (python-shell-parse-command) nil t)))
1931 (python-shell-make-comint
1932 cmd (python-shell-get-process-name dedicated) show)
1933 dedicated)
1934
1935 (defun run-python-internal ()
1936 "Run an inferior Internal Python process.
1937 Input and output via buffer named after
1938 `python-shell-internal-buffer-name' and what
1939 `python-shell-internal-get-process-name' returns.
1940
1941 This new kind of shell is intended to be used for generic
1942 communication related to defined configurations, the main
1943 difference with global or dedicated shells is that these ones are
1944 attached to a configuration, not a buffer. This means that can
1945 be used for example to retrieve the sys.path and other stuff,
1946 without messing with user shells. Note that
1947 `python-shell-enable-font-lock' and `inferior-python-mode-hook'
1948 are set to nil for these shells, so setup codes are not sent at
1949 startup."
1950 (let ((python-shell-enable-font-lock nil)
1951 (inferior-python-mode-hook nil))
1952 (get-buffer-process
1953 (python-shell-make-comint
1954 (python-shell-parse-command)
1955 (python-shell-internal-get-process-name) nil t))))
1956
1957 (defun python-shell-get-process ()
1958 "Get inferior Python process for current buffer and return it."
1959 (let* ((dedicated-proc-name (python-shell-get-process-name t))
1960 (dedicated-proc-buffer-name (format "*%s*" dedicated-proc-name))
1961 (global-proc-name (python-shell-get-process-name nil))
1962 (global-proc-buffer-name (format "*%s*" global-proc-name))
1963 (dedicated-running (comint-check-proc dedicated-proc-buffer-name))
1964 (global-running (comint-check-proc global-proc-buffer-name)))
1965 ;; Always prefer dedicated
1966 (get-buffer-process (or (and dedicated-running dedicated-proc-buffer-name)
1967 (and global-running global-proc-buffer-name)))))
1968
1969 (defun python-shell-get-or-create-process ()
1970 "Get or create an inferior Python process for current buffer and return it."
1971 (let* ((dedicated-proc-name (python-shell-get-process-name t))
1972 (dedicated-proc-buffer-name (format "*%s*" dedicated-proc-name))
1973 (global-proc-name (python-shell-get-process-name nil))
1974 (global-proc-buffer-name (format "*%s*" global-proc-name))
1975 (dedicated-running (comint-check-proc dedicated-proc-buffer-name))
1976 (global-running (comint-check-proc global-proc-buffer-name))
1977 (current-prefix-arg 16))
1978 (when (and (not dedicated-running) (not global-running))
1979 (if (call-interactively 'run-python)
1980 (setq dedicated-running t)
1981 (setq global-running t)))
1982 ;; Always prefer dedicated
1983 (get-buffer-process (if dedicated-running
1984 dedicated-proc-buffer-name
1985 global-proc-buffer-name))))
1986
1987 (defvar python-shell-internal-buffer nil
1988 "Current internal shell buffer for the current buffer.
1989 This is really not necessary at all for the code to work but it's
1990 there for compatibility with CEDET.")
1991
1992 (defvar python-shell-internal-last-output nil
1993 "Last output captured by the internal shell.
1994 This is really not necessary at all for the code to work but it's
1995 there for compatibility with CEDET.")
1996
1997 (defun python-shell-internal-get-or-create-process ()
1998 "Get or create an inferior Internal Python process."
1999 (let* ((proc-name (python-shell-internal-get-process-name))
2000 (proc-buffer-name (format " *%s*" proc-name)))
2001 (when (not (process-live-p proc-name))
2002 (run-python-internal)
2003 (setq python-shell-internal-buffer proc-buffer-name)
2004 ;; XXX: Why is this `sit-for' needed?
2005 ;; `python-shell-make-comint' calls `accept-process-output'
2006 ;; already but it is not helping to get proper output on
2007 ;; 'gnu/linux when the internal shell process is not running and
2008 ;; a call to `python-shell-internal-send-string' is issued.
2009 (sit-for 0.1 t))
2010 (get-buffer-process proc-buffer-name)))
2011
2012 (define-obsolete-function-alias
2013 'python-proc 'python-shell-internal-get-or-create-process "24.3")
2014
2015 (define-obsolete-variable-alias
2016 'python-buffer 'python-shell-internal-buffer "24.3")
2017
2018 (define-obsolete-variable-alias
2019 'python-preoutput-result 'python-shell-internal-last-output "24.3")
2020
2021 (defun python-shell-send-string (string &optional process msg)
2022 "Send STRING to inferior Python PROCESS.
2023 When MSG is non-nil messages the first line of STRING."
2024 (interactive "sPython command: ")
2025 (let ((process (or process (python-shell-get-or-create-process)))
2026 (lines (split-string string "\n" t)))
2027 (and msg (message "Sent: %s..." (nth 0 lines)))
2028 (if (> (length lines) 1)
2029 (let* ((temporary-file-directory
2030 (if (file-remote-p default-directory)
2031 (concat (file-remote-p default-directory) "/tmp")
2032 temporary-file-directory))
2033 (temp-file-name (make-temp-file "py"))
2034 (file-name (or (buffer-file-name) temp-file-name)))
2035 (with-temp-file temp-file-name
2036 (insert string)
2037 (delete-trailing-whitespace))
2038 (python-shell-send-file file-name process temp-file-name))
2039 (comint-send-string process string)
2040 (when (or (not (string-match "\n$" string))
2041 (string-match "\n[ \t].*\n?$" string))
2042 (comint-send-string process "\n")))))
2043
2044 (defvar python-shell-output-filter-in-progress nil)
2045 (defvar python-shell-output-filter-buffer nil)
2046
2047 (defun python-shell-output-filter (string)
2048 "Filter used in `python-shell-send-string-no-output' to grab output.
2049 STRING is the output received to this point from the process.
2050 This filter saves received output from the process in
2051 `python-shell-output-filter-buffer' and stops receiving it after
2052 detecting a prompt at the end of the buffer."
2053 (setq
2054 string (ansi-color-filter-apply string)
2055 python-shell-output-filter-buffer
2056 (concat python-shell-output-filter-buffer string))
2057 (when (string-match
2058 ;; XXX: It seems on OSX an extra carriage return is attached
2059 ;; at the end of output, this handles that too.
2060 (format "\r?\n\\(?:%s\\|%s\\|%s\\)$"
2061 python-shell-prompt-regexp
2062 python-shell-prompt-block-regexp
2063 python-shell-prompt-pdb-regexp)
2064 python-shell-output-filter-buffer)
2065 ;; Output ends when `python-shell-output-filter-buffer' contains
2066 ;; the prompt attached at the end of it.
2067 (setq python-shell-output-filter-in-progress nil
2068 python-shell-output-filter-buffer
2069 (substring python-shell-output-filter-buffer
2070 0 (match-beginning 0)))
2071 (when (and (> (length python-shell-prompt-output-regexp) 0)
2072 (string-match (concat "^" python-shell-prompt-output-regexp)
2073 python-shell-output-filter-buffer))
2074 ;; Some shells, like iPython might append a prompt before the
2075 ;; output, clean that.
2076 (setq python-shell-output-filter-buffer
2077 (substring python-shell-output-filter-buffer (match-end 0)))))
2078 "")
2079
2080 (defun python-shell-send-string-no-output (string &optional process msg)
2081 "Send STRING to PROCESS and inhibit output.
2082 When MSG is non-nil messages the first line of STRING. Return
2083 the output."
2084 (let ((process (or process (python-shell-get-or-create-process)))
2085 (comint-preoutput-filter-functions
2086 '(python-shell-output-filter))
2087 (python-shell-output-filter-in-progress t)
2088 (inhibit-quit t))
2089 (or
2090 (with-local-quit
2091 (python-shell-send-string string process msg)
2092 (while python-shell-output-filter-in-progress
2093 ;; `python-shell-output-filter' takes care of setting
2094 ;; `python-shell-output-filter-in-progress' to NIL after it
2095 ;; detects end of output.
2096 (accept-process-output process))
2097 (prog1
2098 python-shell-output-filter-buffer
2099 (setq python-shell-output-filter-buffer nil)))
2100 (with-current-buffer (process-buffer process)
2101 (comint-interrupt-subjob)))))
2102
2103 (defun python-shell-internal-send-string (string)
2104 "Send STRING to the Internal Python interpreter.
2105 Returns the output. See `python-shell-send-string-no-output'."
2106 ;; XXX Remove `python-shell-internal-last-output' once CEDET is
2107 ;; updated to support this new mode.
2108 (setq python-shell-internal-last-output
2109 (python-shell-send-string-no-output
2110 ;; Makes this function compatible with the old
2111 ;; python-send-receive. (At least for CEDET).
2112 (replace-regexp-in-string "_emacs_out +" "" string)
2113 (python-shell-internal-get-or-create-process) nil)))
2114
2115 (define-obsolete-function-alias
2116 'python-send-receive 'python-shell-internal-send-string "24.3")
2117
2118 (define-obsolete-function-alias
2119 'python-send-string 'python-shell-internal-send-string "24.3")
2120
2121 (defun python-shell-send-region (start end)
2122 "Send the region delimited by START and END to inferior Python process."
2123 (interactive "r")
2124 (python-shell-send-string
2125 (concat
2126 (let ((line-num (line-number-at-pos start)))
2127 ;; When sending a region, add blank lines for non sent code so
2128 ;; backtraces remain correct.
2129 (make-string (1- line-num) ?\n))
2130 (buffer-substring start end))
2131 nil t))
2132
2133 (defun python-shell-send-buffer (&optional arg)
2134 "Send the entire buffer to inferior Python process.
2135 With prefix ARG allow execution of code inside blocks delimited
2136 by \"if __name__== '__main__':\""
2137 (interactive "P")
2138 (save-restriction
2139 (widen)
2140 (let ((str (buffer-substring (point-min) (point-max))))
2141 (and
2142 (not arg)
2143 (setq str (replace-regexp-in-string
2144 (python-rx if-name-main)
2145 "if __name__ == '__main__ ':" str)))
2146 (python-shell-send-string str))))
2147
2148 (defun python-shell-send-defun (arg)
2149 "Send the current defun to inferior Python process.
2150 When argument ARG is non-nil do not include decorators."
2151 (interactive "P")
2152 (save-excursion
2153 (python-shell-send-region
2154 (progn
2155 (end-of-line 1)
2156 (while (and (or (python-nav-beginning-of-defun)
2157 (beginning-of-line 1))
2158 (> (current-indentation) 0)))
2159 (when (not arg)
2160 (while (and (forward-line -1)
2161 (looking-at (python-rx decorator))))
2162 (forward-line 1))
2163 (point-marker))
2164 (progn
2165 (or (python-nav-end-of-defun)
2166 (end-of-line 1))
2167 (point-marker)))))
2168
2169 (defun python-shell-send-file (file-name &optional process temp-file-name)
2170 "Send FILE-NAME to inferior Python PROCESS.
2171 If TEMP-FILE-NAME is passed then that file is used for processing
2172 instead, while internally the shell will continue to use
2173 FILE-NAME."
2174 (interactive "fFile to send: ")
2175 (let* ((process (or process (python-shell-get-or-create-process)))
2176 (temp-file-name (when temp-file-name
2177 (expand-file-name
2178 (or (file-remote-p temp-file-name 'localname)
2179 temp-file-name))))
2180 (file-name (or (when file-name
2181 (expand-file-name
2182 (or (file-remote-p file-name 'localname)
2183 file-name)))
2184 temp-file-name)))
2185 (when (not file-name)
2186 (error "If FILE-NAME is nil then TEMP-FILE-NAME must be non-nil"))
2187 (python-shell-send-string
2188 (format
2189 (concat "__pyfile = open('''%s''');"
2190 "exec(compile(__pyfile.read(), '''%s''', 'exec'));"
2191 "__pyfile.close()")
2192 (or temp-file-name file-name) file-name)
2193 process)))
2194
2195 (defun python-shell-switch-to-shell ()
2196 "Switch to inferior Python process buffer."
2197 (interactive)
2198 (pop-to-buffer (process-buffer (python-shell-get-or-create-process)) t))
2199
2200 (defun python-shell-send-setup-code ()
2201 "Send all setup code for shell.
2202 This function takes the list of setup code to send from the
2203 `python-shell-setup-codes' list."
2204 (let ((process (get-buffer-process (current-buffer))))
2205 (dolist (code python-shell-setup-codes)
2206 (when code
2207 (message "Sent %s" code)
2208 (python-shell-send-string
2209 (symbol-value code) process)))))
2210
2211 (add-hook 'inferior-python-mode-hook
2212 #'python-shell-send-setup-code)
2213
2214 \f
2215 ;;; Shell completion
2216
2217 (defcustom python-shell-completion-setup-code
2218 "try:
2219 import readline
2220 except ImportError:
2221 def __COMPLETER_all_completions(text): []
2222 else:
2223 import rlcompleter
2224 readline.set_completer(rlcompleter.Completer().complete)
2225 def __COMPLETER_all_completions(text):
2226 import sys
2227 completions = []
2228 try:
2229 i = 0
2230 while True:
2231 res = readline.get_completer()(text, i)
2232 if not res: break
2233 i += 1
2234 completions.append(res)
2235 except NameError:
2236 pass
2237 return completions"
2238 "Code used to setup completion in inferior Python processes."
2239 :type 'string
2240 :group 'python)
2241
2242 (defcustom python-shell-completion-string-code
2243 "';'.join(__COMPLETER_all_completions('''%s'''))\n"
2244 "Python code used to get a string of completions separated by semicolons."
2245 :type 'string
2246 :group 'python)
2247
2248 (defcustom python-shell-completion-module-string-code ""
2249 "Python code used to get completions separated by semicolons for imports.
2250
2251 For IPython v0.11, add the following line to
2252 `python-shell-completion-setup-code':
2253
2254 from IPython.core.completerlib import module_completion
2255
2256 and use the following as the value of this variable:
2257
2258 ';'.join(module_completion('''%s'''))\n"
2259 :type 'string
2260 :group 'python)
2261
2262 (defcustom python-shell-completion-pdb-string-code
2263 "';'.join(globals().keys() + locals().keys())"
2264 "Python code used to get completions separated by semicolons for [i]pdb."
2265 :type 'string
2266 :group 'python)
2267
2268 (defun python-shell-completion-get-completions (process line input)
2269 "Do completion at point for PROCESS.
2270 LINE is used to detect the context on how to complete given
2271 INPUT."
2272 (let* ((prompt
2273 ;; Get the last prompt for the inferior process
2274 ;; buffer. This is used for the completion code selection
2275 ;; heuristic.
2276 (with-current-buffer (process-buffer process)
2277 (buffer-substring-no-properties
2278 (overlay-start comint-last-prompt-overlay)
2279 (overlay-end comint-last-prompt-overlay))))
2280 (completion-context
2281 ;; Check whether a prompt matches a pdb string, an import
2282 ;; statement or just the standard prompt and use the
2283 ;; correct python-shell-completion-*-code string
2284 (cond ((and (> (length python-shell-completion-pdb-string-code) 0)
2285 (string-match
2286 (concat "^" python-shell-prompt-pdb-regexp) prompt))
2287 'pdb)
2288 ((and (>
2289 (length python-shell-completion-module-string-code) 0)
2290 (string-match
2291 (concat "^" python-shell-prompt-regexp) prompt)
2292 (string-match "^[ \t]*\\(from\\|import\\)[ \t]" line))
2293 'import)
2294 ((string-match
2295 (concat "^" python-shell-prompt-regexp) prompt)
2296 'default)
2297 (t nil)))
2298 (completion-code
2299 (pcase completion-context
2300 (`pdb python-shell-completion-pdb-string-code)
2301 (`import python-shell-completion-module-string-code)
2302 (`default python-shell-completion-string-code)
2303 (_ nil)))
2304 (input
2305 (if (eq completion-context 'import)
2306 (replace-regexp-in-string "^[ \t]+" "" line)
2307 input)))
2308 (and completion-code
2309 (> (length input) 0)
2310 (with-current-buffer (process-buffer process)
2311 (let ((completions (python-shell-send-string-no-output
2312 (format completion-code input) process)))
2313 (and (> (length completions) 2)
2314 (split-string completions
2315 "^'\\|^\"\\|;\\|'$\\|\"$" t)))))))
2316
2317 (defun python-shell-completion-complete-at-point (&optional process)
2318 "Perform completion at point in inferior Python.
2319 Optional argument PROCESS forces completions to be retrieved
2320 using that one instead of current buffer's process."
2321 (setq process (or process (get-buffer-process (current-buffer))))
2322 (let* ((start
2323 (save-excursion
2324 (with-syntax-table python-dotty-syntax-table
2325 (let* ((paren-depth (car (syntax-ppss)))
2326 (syntax-string "w_")
2327 (syntax-list (string-to-syntax syntax-string)))
2328 ;; Stop scanning for the beginning of the completion
2329 ;; subject after the char before point matches a
2330 ;; delimiter
2331 (while (member
2332 (car (syntax-after (1- (point)))) syntax-list)
2333 (skip-syntax-backward syntax-string)
2334 (when (or (equal (char-before) ?\))
2335 (equal (char-before) ?\"))
2336 (forward-char -1))
2337 (while (or
2338 ;; honor initial paren depth
2339 (> (car (syntax-ppss)) paren-depth)
2340 (python-syntax-context 'string))
2341 (forward-char -1)))
2342 (point)))))
2343 (end (point)))
2344 (list start end
2345 (completion-table-dynamic
2346 (apply-partially
2347 #'python-shell-completion-get-completions
2348 process (buffer-substring-no-properties
2349 (line-beginning-position) end))))))
2350
2351 (defun python-shell-completion-complete-or-indent ()
2352 "Complete or indent depending on the context.
2353 If content before pointer is all whitespace indent. If not try
2354 to complete."
2355 (interactive)
2356 (if (string-match "^[[:space:]]*$"
2357 (buffer-substring (comint-line-beginning-position)
2358 (point-marker)))
2359 (indent-for-tab-command)
2360 (completion-at-point)))
2361
2362 \f
2363 ;;; PDB Track integration
2364
2365 (defcustom python-pdbtrack-activate t
2366 "Non-nil makes python shell enable pdbtracking."
2367 :type 'boolean
2368 :group 'python
2369 :safe 'booleanp)
2370
2371 (defcustom python-pdbtrack-stacktrace-info-regexp
2372 "^> \\([^\"(<]+\\)(\\([0-9]+\\))\\([?a-zA-Z0-9_<>]+\\)()"
2373 "Regular Expression matching stacktrace information.
2374 Used to extract the current line and module being inspected."
2375 :type 'string
2376 :group 'python
2377 :safe 'stringp)
2378
2379 (defvar python-pdbtrack-tracked-buffer nil
2380 "Variable containing the value of the current tracked buffer.
2381 Never set this variable directly, use
2382 `python-pdbtrack-set-tracked-buffer' instead.")
2383
2384 (defvar python-pdbtrack-buffers-to-kill nil
2385 "List of buffers to be deleted after tracking finishes.")
2386
2387 (defun python-pdbtrack-set-tracked-buffer (file-name)
2388 "Set the buffer for FILE-NAME as the tracked buffer.
2389 Internally it uses the `python-pdbtrack-tracked-buffer' variable.
2390 Returns the tracked buffer."
2391 (let ((file-buffer (get-file-buffer file-name)))
2392 (if file-buffer
2393 (setq python-pdbtrack-tracked-buffer file-buffer)
2394 (setq file-buffer (find-file-noselect file-name))
2395 (when (not (member file-buffer python-pdbtrack-buffers-to-kill))
2396 (add-to-list 'python-pdbtrack-buffers-to-kill file-buffer)))
2397 file-buffer))
2398
2399 (defun python-pdbtrack-comint-output-filter-function (output)
2400 "Move overlay arrow to current pdb line in tracked buffer.
2401 Argument OUTPUT is a string with the output from the comint process."
2402 (when (and python-pdbtrack-activate (not (string= output "")))
2403 (let* ((full-output (ansi-color-filter-apply
2404 (buffer-substring comint-last-input-end (point-max))))
2405 (line-number)
2406 (file-name
2407 (with-temp-buffer
2408 (insert full-output)
2409 ;; When the debugger encounters a pdb.set_trace()
2410 ;; command, it prints a single stack frame. Sometimes
2411 ;; it prints a bit of extra information about the
2412 ;; arguments of the present function. When ipdb
2413 ;; encounters an exception, it prints the _entire_ stack
2414 ;; trace. To handle all of these cases, we want to find
2415 ;; the _last_ stack frame printed in the most recent
2416 ;; batch of output, then jump to the corresponding
2417 ;; file/line number.
2418 (goto-char (point-max))
2419 (when (re-search-backward python-pdbtrack-stacktrace-info-regexp nil t)
2420 (setq line-number (string-to-number
2421 (match-string-no-properties 2)))
2422 (match-string-no-properties 1)))))
2423 (if (and file-name line-number)
2424 (let* ((tracked-buffer
2425 (python-pdbtrack-set-tracked-buffer file-name))
2426 (shell-buffer (current-buffer))
2427 (tracked-buffer-window (get-buffer-window tracked-buffer))
2428 (tracked-buffer-line-pos))
2429 (with-current-buffer tracked-buffer
2430 (set (make-local-variable 'overlay-arrow-string) "=>")
2431 (set (make-local-variable 'overlay-arrow-position) (make-marker))
2432 (setq tracked-buffer-line-pos (progn
2433 (goto-char (point-min))
2434 (forward-line (1- line-number))
2435 (point-marker)))
2436 (when tracked-buffer-window
2437 (set-window-point
2438 tracked-buffer-window tracked-buffer-line-pos))
2439 (set-marker overlay-arrow-position tracked-buffer-line-pos))
2440 (pop-to-buffer tracked-buffer)
2441 (switch-to-buffer-other-window shell-buffer))
2442 (when python-pdbtrack-tracked-buffer
2443 (with-current-buffer python-pdbtrack-tracked-buffer
2444 (set-marker overlay-arrow-position nil))
2445 (mapc #'(lambda (buffer)
2446 (ignore-errors (kill-buffer buffer)))
2447 python-pdbtrack-buffers-to-kill)
2448 (setq python-pdbtrack-tracked-buffer nil
2449 python-pdbtrack-buffers-to-kill nil)))))
2450 output)
2451
2452 \f
2453 ;;; Symbol completion
2454
2455 (defun python-completion-complete-at-point ()
2456 "Complete current symbol at point.
2457 For this to work the best as possible you should call
2458 `python-shell-send-buffer' from time to time so context in
2459 inferior python process is updated properly."
2460 (let ((process (python-shell-get-process)))
2461 (if (not process)
2462 (error "Completion needs an inferior Python process running")
2463 (python-shell-completion-complete-at-point process))))
2464
2465 (add-to-list 'debug-ignored-errors
2466 "^Completion needs an inferior Python process running.")
2467
2468 \f
2469 ;;; Fill paragraph
2470
2471 (defcustom python-fill-comment-function 'python-fill-comment
2472 "Function to fill comments.
2473 This is the function used by `python-fill-paragraph' to
2474 fill comments."
2475 :type 'symbol
2476 :group 'python)
2477
2478 (defcustom python-fill-string-function 'python-fill-string
2479 "Function to fill strings.
2480 This is the function used by `python-fill-paragraph' to
2481 fill strings."
2482 :type 'symbol
2483 :group 'python)
2484
2485 (defcustom python-fill-decorator-function 'python-fill-decorator
2486 "Function to fill decorators.
2487 This is the function used by `python-fill-paragraph' to
2488 fill decorators."
2489 :type 'symbol
2490 :group 'python)
2491
2492 (defcustom python-fill-paren-function 'python-fill-paren
2493 "Function to fill parens.
2494 This is the function used by `python-fill-paragraph' to
2495 fill parens."
2496 :type 'symbol
2497 :group 'python)
2498
2499 (defcustom python-fill-docstring-style 'pep-257
2500 "Style used to fill docstrings.
2501 This affects `python-fill-string' behavior with regards to
2502 triple quotes positioning.
2503
2504 Possible values are DJANGO, ONETWO, PEP-257, PEP-257-NN,
2505 SYMMETRIC, and NIL. A value of NIL won't care about quotes
2506 position and will treat docstrings a normal string, any other
2507 value may result in one of the following docstring styles:
2508
2509 DJANGO:
2510
2511 \"\"\"
2512 Process foo, return bar.
2513 \"\"\"
2514
2515 \"\"\"
2516 Process foo, return bar.
2517
2518 If processing fails throw ProcessingError.
2519 \"\"\"
2520
2521 ONETWO:
2522
2523 \"\"\"Process foo, return bar.\"\"\"
2524
2525 \"\"\"
2526 Process foo, return bar.
2527
2528 If processing fails throw ProcessingError.
2529
2530 \"\"\"
2531
2532 PEP-257:
2533
2534 \"\"\"Process foo, return bar.\"\"\"
2535
2536 \"\"\"Process foo, return bar.
2537
2538 If processing fails throw ProcessingError.
2539
2540 \"\"\"
2541
2542 PEP-257-NN:
2543
2544 \"\"\"Process foo, return bar.\"\"\"
2545
2546 \"\"\"Process foo, return bar.
2547
2548 If processing fails throw ProcessingError.
2549 \"\"\"
2550
2551 SYMMETRIC:
2552
2553 \"\"\"Process foo, return bar.\"\"\"
2554
2555 \"\"\"
2556 Process foo, return bar.
2557
2558 If processing fails throw ProcessingError.
2559 \"\"\""
2560 :type '(choice
2561 (const :tag "Don't format docstrings" nil)
2562 (const :tag "Django's coding standards style." django)
2563 (const :tag "One newline and start and Two at end style." onetwo)
2564 (const :tag "PEP-257 with 2 newlines at end of string." pep-257)
2565 (const :tag "PEP-257 with 1 newline at end of string." pep-257-nn)
2566 (const :tag "Symmetric style." symmetric))
2567 :group 'python
2568 :safe (lambda (val)
2569 (memq val '(django onetwo pep-257 pep-257-nn symmetric nil))))
2570
2571 (defun python-fill-paragraph (&optional justify)
2572 "`fill-paragraph-function' handling multi-line strings and possibly comments.
2573 If any of the current line is in or at the end of a multi-line string,
2574 fill the string or the paragraph of it that point is in, preserving
2575 the string's indentation.
2576 Optional argument JUSTIFY defines if the paragraph should be justified."
2577 (interactive "P")
2578 (save-excursion
2579 (cond
2580 ;; Comments
2581 ((python-syntax-context 'comment)
2582 (funcall python-fill-comment-function justify))
2583 ;; Strings/Docstrings
2584 ((save-excursion (or (python-syntax-context 'string)
2585 (equal (string-to-syntax "|")
2586 (syntax-after (point)))))
2587 (funcall python-fill-string-function justify))
2588 ;; Decorators
2589 ((equal (char-after (save-excursion
2590 (python-nav-beginning-of-statement))) ?@)
2591 (funcall python-fill-decorator-function justify))
2592 ;; Parens
2593 ((or (python-syntax-context 'paren)
2594 (looking-at (python-rx open-paren))
2595 (save-excursion
2596 (skip-syntax-forward "^(" (line-end-position))
2597 (looking-at (python-rx open-paren))))
2598 (funcall python-fill-paren-function justify))
2599 (t t))))
2600
2601 (defun python-fill-comment (&optional justify)
2602 "Comment fill function for `python-fill-paragraph'.
2603 JUSTIFY should be used (if applicable) as in `fill-paragraph'."
2604 (fill-comment-paragraph justify))
2605
2606 (defun python-fill-string (&optional justify)
2607 "String fill function for `python-fill-paragraph'.
2608 JUSTIFY should be used (if applicable) as in `fill-paragraph'."
2609 (let* ((marker (point-marker))
2610 (str-start-pos
2611 (set-marker
2612 (make-marker)
2613 (or (python-syntax-context 'string)
2614 (and (equal (string-to-syntax "|")
2615 (syntax-after (point)))
2616 (point)))))
2617 (num-quotes (python-syntax-count-quotes
2618 (char-after str-start-pos) str-start-pos))
2619 (str-end-pos
2620 (save-excursion
2621 (goto-char (+ str-start-pos num-quotes))
2622 (or (re-search-forward (rx (syntax string-delimiter)) nil t)
2623 (goto-char (point-max)))
2624 (point-marker)))
2625 (multi-line-p
2626 ;; Docstring styles may vary for oneliners and multi-liners.
2627 (> (count-matches "\n" str-start-pos str-end-pos) 0))
2628 (delimiters-style
2629 (pcase python-fill-docstring-style
2630 ;; delimiters-style is a cons cell with the form
2631 ;; (START-NEWLINES . END-NEWLINES). When any of the sexps
2632 ;; is NIL means to not add any newlines for start or end
2633 ;; of docstring. See `python-fill-docstring-style' for a
2634 ;; graphic idea of each style.
2635 (`django (cons 1 1))
2636 (`onetwo (and multi-line-p (cons 1 2)))
2637 (`pep-257 (and multi-line-p (cons nil 2)))
2638 (`pep-257-nn (and multi-line-p (cons nil 1)))
2639 (`symmetric (and multi-line-p (cons 1 1)))))
2640 (docstring-p (save-excursion
2641 ;; Consider docstrings those strings which
2642 ;; start on a line by themselves.
2643 (python-nav-beginning-of-statement)
2644 (and (= (point) str-start-pos))))
2645 (fill-paragraph-function))
2646 (save-restriction
2647 (narrow-to-region str-start-pos str-end-pos)
2648 (fill-paragraph justify))
2649 (save-excursion
2650 (when (and docstring-p python-fill-docstring-style)
2651 ;; Add the number of newlines indicated by the selected style
2652 ;; at the start of the docstring.
2653 (goto-char (+ str-start-pos num-quotes))
2654 (delete-region (point) (progn
2655 (skip-syntax-forward "> ")
2656 (point)))
2657 (and (car delimiters-style)
2658 (or (newline (car delimiters-style)) t)
2659 ;; Indent only if a newline is added.
2660 (indent-according-to-mode))
2661 ;; Add the number of newlines indicated by the selected style
2662 ;; at the end of the docstring.
2663 (goto-char (if (not (= str-end-pos (point-max)))
2664 (- str-end-pos num-quotes)
2665 str-end-pos))
2666 (delete-region (point) (progn
2667 (skip-syntax-backward "> ")
2668 (point)))
2669 (and (cdr delimiters-style)
2670 ;; Add newlines only if string ends.
2671 (not (= str-end-pos (point-max)))
2672 (or (newline (cdr delimiters-style)) t)
2673 ;; Again indent only if a newline is added.
2674 (indent-according-to-mode))))) t)
2675
2676 (defun python-fill-decorator (&optional justify)
2677 "Decorator fill function for `python-fill-paragraph'.
2678 JUSTIFY should be used (if applicable) as in `fill-paragraph'."
2679 t)
2680
2681 (defun python-fill-paren (&optional justify)
2682 "Paren fill function for `python-fill-paragraph'.
2683 JUSTIFY should be used (if applicable) as in `fill-paragraph'."
2684 (save-restriction
2685 (narrow-to-region (progn
2686 (while (python-syntax-context 'paren)
2687 (goto-char (1- (point-marker))))
2688 (point-marker)
2689 (line-beginning-position))
2690 (progn
2691 (when (not (python-syntax-context 'paren))
2692 (end-of-line)
2693 (when (not (python-syntax-context 'paren))
2694 (skip-syntax-backward "^)")))
2695 (while (python-syntax-context 'paren)
2696 (goto-char (1+ (point-marker))))
2697 (point-marker)))
2698 (let ((paragraph-start "\f\\|[ \t]*$")
2699 (paragraph-separate ",")
2700 (fill-paragraph-function))
2701 (goto-char (point-min))
2702 (fill-paragraph justify))
2703 (while (not (eobp))
2704 (forward-line 1)
2705 (python-indent-line)
2706 (goto-char (line-end-position)))) t)
2707
2708 \f
2709 ;;; Skeletons
2710
2711 (defcustom python-skeleton-autoinsert nil
2712 "Non-nil means template skeletons will be automagically inserted.
2713 This happens when pressing \"if<SPACE>\", for example, to prompt for
2714 the if condition."
2715 :type 'boolean
2716 :group 'python
2717 :safe 'booleanp)
2718
2719 (define-obsolete-variable-alias
2720 'python-use-skeletons 'python-skeleton-autoinsert "24.3")
2721
2722 (defvar python-skeleton-available '()
2723 "Internal list of available skeletons.")
2724
2725 (define-abbrev-table 'python-mode-skeleton-abbrev-table ()
2726 "Abbrev table for Python mode skeletons."
2727 :case-fixed t
2728 ;; Allow / inside abbrevs.
2729 :regexp "\\(?:^\\|[^/]\\)\\<\\([[:word:]/]+\\)\\W*"
2730 ;; Only expand in code.
2731 :enable-function (lambda ()
2732 (and
2733 (not (python-syntax-comment-or-string-p))
2734 python-skeleton-autoinsert)))
2735
2736 (defmacro python-skeleton-define (name doc &rest skel)
2737 "Define a `python-mode' skeleton using NAME DOC and SKEL.
2738 The skeleton will be bound to python-skeleton-NAME and will
2739 be added to `python-mode-skeleton-abbrev-table'."
2740 (declare (indent 2))
2741 (let* ((name (symbol-name name))
2742 (function-name (intern (concat "python-skeleton-" name))))
2743 `(progn
2744 (define-abbrev python-mode-skeleton-abbrev-table
2745 ,name "" ',function-name :system t)
2746 (setq python-skeleton-available
2747 (cons ',function-name python-skeleton-available))
2748 (define-skeleton ,function-name
2749 ,(or doc
2750 (format "Insert %s statement." name))
2751 ,@skel))))
2752
2753 (define-abbrev-table 'python-mode-abbrev-table ()
2754 "Abbrev table for Python mode."
2755 :parents (list python-mode-skeleton-abbrev-table))
2756
2757 (defmacro python-define-auxiliary-skeleton (name doc &optional &rest skel)
2758 "Define a `python-mode' auxiliary skeleton using NAME DOC and SKEL.
2759 The skeleton will be bound to python-skeleton-NAME."
2760 (declare (indent 2))
2761 (let* ((name (symbol-name name))
2762 (function-name (intern (concat "python-skeleton--" name)))
2763 (msg (format
2764 "Add '%s' clause? " name)))
2765 (when (not skel)
2766 (setq skel
2767 `(< ,(format "%s:" name) \n \n
2768 > _ \n)))
2769 `(define-skeleton ,function-name
2770 ,(or doc
2771 (format "Auxiliary skeleton for %s statement." name))
2772 nil
2773 (unless (y-or-n-p ,msg)
2774 (signal 'quit t))
2775 ,@skel)))
2776
2777 (python-define-auxiliary-skeleton else nil)
2778
2779 (python-define-auxiliary-skeleton except nil)
2780
2781 (python-define-auxiliary-skeleton finally nil)
2782
2783 (python-skeleton-define if nil
2784 "Condition: "
2785 "if " str ":" \n
2786 _ \n
2787 ("other condition, %s: "
2788 <
2789 "elif " str ":" \n
2790 > _ \n nil)
2791 '(python-skeleton--else) | ^)
2792
2793 (python-skeleton-define while nil
2794 "Condition: "
2795 "while " str ":" \n
2796 > _ \n
2797 '(python-skeleton--else) | ^)
2798
2799 (python-skeleton-define for nil
2800 "Iteration spec: "
2801 "for " str ":" \n
2802 > _ \n
2803 '(python-skeleton--else) | ^)
2804
2805 (python-skeleton-define try nil
2806 nil
2807 "try:" \n
2808 > _ \n
2809 ("Exception, %s: "
2810 <
2811 "except " str ":" \n
2812 > _ \n nil)
2813 resume:
2814 '(python-skeleton--except)
2815 '(python-skeleton--else)
2816 '(python-skeleton--finally) | ^)
2817
2818 (python-skeleton-define def nil
2819 "Function name: "
2820 "def " str "(" ("Parameter, %s: "
2821 (unless (equal ?\( (char-before)) ", ")
2822 str) "):" \n
2823 "\"\"\"" - "\"\"\"" \n
2824 > _ \n)
2825
2826 (python-skeleton-define class nil
2827 "Class name: "
2828 "class " str "(" ("Inheritance, %s: "
2829 (unless (equal ?\( (char-before)) ", ")
2830 str)
2831 & ")" | -2
2832 ":" \n
2833 "\"\"\"" - "\"\"\"" \n
2834 > _ \n)
2835
2836 (defun python-skeleton-add-menu-items ()
2837 "Add menu items to Python->Skeletons menu."
2838 (let ((skeletons (sort python-skeleton-available 'string<))
2839 (items))
2840 (dolist (skeleton skeletons)
2841 (easy-menu-add-item
2842 nil '("Python" "Skeletons")
2843 `[,(format
2844 "Insert %s" (nth 2 (split-string (symbol-name skeleton) "-")))
2845 ,skeleton t]))))
2846 \f
2847 ;;; FFAP
2848
2849 (defcustom python-ffap-setup-code
2850 "def __FFAP_get_module_path(module):
2851 try:
2852 import os
2853 path = __import__(module).__file__
2854 if path[-4:] == '.pyc' and os.path.exists(path[0:-1]):
2855 path = path[:-1]
2856 return path
2857 except:
2858 return ''"
2859 "Python code to get a module path."
2860 :type 'string
2861 :group 'python)
2862
2863 (defcustom python-ffap-string-code
2864 "__FFAP_get_module_path('''%s''')\n"
2865 "Python code used to get a string with the path of a module."
2866 :type 'string
2867 :group 'python)
2868
2869 (defun python-ffap-module-path (module)
2870 "Function for `ffap-alist' to return path for MODULE."
2871 (let ((process (or
2872 (and (eq major-mode 'inferior-python-mode)
2873 (get-buffer-process (current-buffer)))
2874 (python-shell-get-process))))
2875 (if (not process)
2876 nil
2877 (let ((module-file
2878 (python-shell-send-string-no-output
2879 (format python-ffap-string-code module) process)))
2880 (when module-file
2881 (substring-no-properties module-file 1 -1))))))
2882
2883 (eval-after-load "ffap"
2884 '(progn
2885 (push '(python-mode . python-ffap-module-path) ffap-alist)
2886 (push '(inferior-python-mode . python-ffap-module-path) ffap-alist)))
2887
2888 \f
2889 ;;; Code check
2890
2891 (defcustom python-check-command
2892 "pyflakes"
2893 "Command used to check a Python file."
2894 :type 'string
2895 :group 'python)
2896
2897 (defcustom python-check-buffer-name
2898 "*Python check: %s*"
2899 "Buffer name used for check commands."
2900 :type 'string
2901 :group 'python)
2902
2903 (defvar python-check-custom-command nil
2904 "Internal use.")
2905
2906 (defun python-check (command)
2907 "Check a Python file (default current buffer's file).
2908 Runs COMMAND, a shell command, as if by `compile'. See
2909 `python-check-command' for the default."
2910 (interactive
2911 (list (read-string "Check command: "
2912 (or python-check-custom-command
2913 (concat python-check-command " "
2914 (shell-quote-argument
2915 (or
2916 (let ((name (buffer-file-name)))
2917 (and name
2918 (file-name-nondirectory name)))
2919 "")))))))
2920 (setq python-check-custom-command command)
2921 (save-some-buffers (not compilation-ask-about-save) nil)
2922 (let ((process-environment (python-shell-calculate-process-environment))
2923 (exec-path (python-shell-calculate-exec-path)))
2924 (compilation-start command nil
2925 (lambda (mode-name)
2926 (format python-check-buffer-name command)))))
2927
2928 \f
2929 ;;; Eldoc
2930
2931 (defcustom python-eldoc-setup-code
2932 "def __PYDOC_get_help(obj):
2933 try:
2934 import inspect
2935 if hasattr(obj, 'startswith'):
2936 obj = eval(obj, globals())
2937 doc = inspect.getdoc(obj)
2938 if not doc and callable(obj):
2939 target = None
2940 if inspect.isclass(obj) and hasattr(obj, '__init__'):
2941 target = obj.__init__
2942 objtype = 'class'
2943 else:
2944 target = obj
2945 objtype = 'def'
2946 if target:
2947 args = inspect.formatargspec(
2948 *inspect.getargspec(target)
2949 )
2950 name = obj.__name__
2951 doc = '{objtype} {name}{args}'.format(
2952 objtype=objtype, name=name, args=args
2953 )
2954 else:
2955 doc = doc.splitlines()[0]
2956 except:
2957 doc = ''
2958 try:
2959 exec('print doc')
2960 except SyntaxError:
2961 print(doc)"
2962 "Python code to setup documentation retrieval."
2963 :type 'string
2964 :group 'python)
2965
2966 (defcustom python-eldoc-string-code
2967 "__PYDOC_get_help('''%s''')\n"
2968 "Python code used to get a string with the documentation of an object."
2969 :type 'string
2970 :group 'python)
2971
2972 (defun python-eldoc--get-doc-at-point (&optional force-input force-process)
2973 "Internal implementation to get documentation at point.
2974 If not FORCE-INPUT is passed then what
2975 `python-info-current-symbol' returns will be used. If not
2976 FORCE-PROCESS is passed what `python-shell-get-process' returns
2977 is used."
2978 (let ((process (or force-process (python-shell-get-process))))
2979 (if (not process)
2980 (error "Eldoc needs an inferior Python process running")
2981 (let ((input (or force-input
2982 (python-info-current-symbol t))))
2983 (and input
2984 (python-shell-send-string-no-output
2985 (format python-eldoc-string-code input)
2986 process))))))
2987
2988 (defun python-eldoc-function ()
2989 "`eldoc-documentation-function' for Python.
2990 For this to work the best as possible you should call
2991 `python-shell-send-buffer' from time to time so context in
2992 inferior python process is updated properly."
2993 (python-eldoc--get-doc-at-point))
2994
2995 (defun python-eldoc-at-point (symbol)
2996 "Get help on SYMBOL using `help'.
2997 Interactively, prompt for symbol."
2998 (interactive
2999 (let ((symbol (python-info-current-symbol t))
3000 (enable-recursive-minibuffers t))
3001 (list (read-string (if symbol
3002 (format "Describe symbol (default %s): " symbol)
3003 "Describe symbol: ")
3004 nil nil symbol))))
3005 (message (python-eldoc--get-doc-at-point symbol)))
3006
3007 (add-to-list 'debug-ignored-errors
3008 "^Eldoc needs an inferior Python process running.")
3009
3010 \f
3011 ;;; Imenu
3012
3013 (defvar python-imenu-format-item-label-function
3014 'python-imenu-format-item-label
3015 "Imenu function used to format an item label.
3016 It must be a function with two arguments: TYPE and NAME.")
3017
3018 (defvar python-imenu-format-parent-item-label-function
3019 'python-imenu-format-parent-item-label
3020 "Imenu function used to format a parent item label.
3021 It must be a function with two arguments: TYPE and NAME.")
3022
3023 (defvar python-imenu-format-parent-item-jump-label-function
3024 'python-imenu-format-parent-item-jump-label
3025 "Imenu function used to format a parent jump item label.
3026 It must be a function with two arguments: TYPE and NAME.")
3027
3028 (defun python-imenu-format-item-label (type name)
3029 "Return imenu label for single node using TYPE and NAME."
3030 (format "%s (%s)" name type))
3031
3032 (defun python-imenu-format-parent-item-label (type name)
3033 "Return imenu label for parent node using TYPE and NAME."
3034 (format "%s..." (python-imenu-format-item-label type name)))
3035
3036 (defun python-imenu-format-parent-item-jump-label (type name)
3037 "Return imenu label for parent node jump using TYPE and NAME."
3038 (if (string= type "class")
3039 "*class definition*"
3040 "*function definition*"))
3041
3042 (defun python-imenu--put-parent (type name pos num-children tree &optional root)
3043 "Add the parent with TYPE, NAME, POS and NUM-CHILDREN to TREE.
3044 Optional Argument ROOT must be non-nil when the node being
3045 processed is the root of the TREE."
3046 (let ((label
3047 (funcall python-imenu-format-item-label-function type name))
3048 (jump-label
3049 (funcall python-imenu-format-parent-item-jump-label-function type name)))
3050 (if root
3051 ;; This is the root, everything is a children.
3052 (cons label (cons (cons jump-label pos) tree))
3053 ;; This is node a which may contain some children.
3054 (cons
3055 (cons label (cons (cons jump-label pos)
3056 ;; Append all the children
3057 (python-util-popn tree num-children)))
3058 ;; All previous non-children nodes.
3059 (nthcdr num-children tree)))))
3060
3061 (defun python-imenu--build-tree (&optional min-indent prev-indent num-children tree)
3062 "Recursively build the tree of nested definitions of a node.
3063 Arguments MIN-INDENT PREV-INDENT NUM-CHILDREN and TREE are
3064 internal and should not be passed explicitly unless you know what
3065 you are doing."
3066 (setq num-children (or num-children 0)
3067 min-indent (or min-indent 0))
3068 (let* ((pos (python-nav-backward-defun))
3069 (type)
3070 (name (when (and pos (looking-at python-nav-beginning-of-defun-regexp))
3071 (let ((split (split-string (match-string-no-properties 0))))
3072 (setq type (car split))
3073 (cadr split))))
3074 (label (when name
3075 (funcall python-imenu-format-item-label-function type name)))
3076 (indent (current-indentation)))
3077 (cond ((not pos)
3078 ;; No defun found, nothing to add.
3079 tree)
3080 ((equal indent 0)
3081 (if (> num-children 0)
3082 ;; Append it as the parent of everything collected to
3083 ;; this point.
3084 (python-imenu--put-parent type name pos num-children tree t)
3085 ;; There are no children, this is a lonely defun.
3086 (cons label pos)))
3087 ((equal min-indent indent)
3088 ;; Stop collecting nodes after moving to a position with
3089 ;; indentation equaling min-indent. This is specially
3090 ;; useful for navigating nested definitions recursively.
3091 tree)
3092 (t
3093 (python-imenu--build-tree
3094 min-indent
3095 indent
3096 ;; Add another children, either when this is the
3097 ;; first call or when indentation is
3098 ;; less-or-equal than previous. And do not
3099 ;; discard the number of children, because the
3100 ;; way code is scanned, all children are
3101 ;; collected until a root node yet to be found
3102 ;; appears.
3103 (if (or (not prev-indent)
3104 (and
3105 (> indent min-indent)
3106 (<= indent prev-indent)))
3107 (1+ num-children)
3108 num-children)
3109 (cond ((not prev-indent)
3110 ;; First call to the function: append this
3111 ;; defun to the index.
3112 (list (cons label pos)))
3113 ((= indent prev-indent)
3114 ;; Add another defun with the same depth
3115 ;; as the previous.
3116 (cons (cons label pos) tree))
3117 ((and (< indent prev-indent)
3118 (< 0 num-children))
3119 ;; There are children to be appended and
3120 ;; the previous defun had more
3121 ;; indentation, the current one must be a
3122 ;; parent.
3123 (python-imenu--put-parent type name pos num-children tree))
3124 ((> indent prev-indent)
3125 ;; There are children defuns deeper than
3126 ;; current depth. Fear not, we already
3127 ;; know how to treat them.
3128 (cons
3129 (prog1
3130 (python-imenu--build-tree
3131 prev-indent indent 1 (list (cons label pos)))
3132 ;; Adjustment: after scanning backwards
3133 ;; for all deeper children, we need to
3134 ;; continue our scan for a parent from
3135 ;; the current defun we are looking at.
3136 (python-nav-forward-defun))
3137 tree))))))))
3138
3139 (defun python-imenu-create-index ()
3140 "Return tree Imenu alist for the current python buffer.
3141 Change `python-imenu-format-item-label-function',
3142 `python-imenu-format-parent-item-label-function',
3143 `python-imenu-format-parent-item-jump-label-function' to
3144 customize how labels are formatted."
3145 (goto-char (point-max))
3146 (let ((index)
3147 (tree))
3148 (while (setq tree (python-imenu--build-tree))
3149 (setq index (cons tree index)))
3150 index))
3151
3152 (defun python-imenu-create-flat-index (&optional alist prefix)
3153 "Return flat outline of the current python buffer for Imenu.
3154 Optional Argument ALIST is the tree to be flattened, when nil
3155 `python-imenu-build-index' is used with
3156 `python-imenu-format-parent-item-jump-label-function'
3157 `python-imenu-format-parent-item-label-function'
3158 `python-imenu-format-item-label-function' set to (lambda (type
3159 name) name). Optional Argument PREFIX is used in recursive calls
3160 and should not be passed explicitly.
3161
3162 Converts this:
3163
3164 \((\"Foo\" . 103)
3165 (\"Bar\" . 138)
3166 (\"decorator\"
3167 (\"decorator\" . 173)
3168 (\"wrap\"
3169 (\"wrap\" . 353)
3170 (\"wrapped_f\" . 393))))
3171
3172 To this:
3173
3174 \((\"Foo\" . 103)
3175 (\"Bar\" . 138)
3176 (\"decorator\" . 173)
3177 (\"decorator.wrap\" . 353)
3178 (\"decorator.wrapped_f\" . 393))"
3179 ;; Inspired by imenu--flatten-index-alist removed in revno 21853.
3180 (apply
3181 'nconc
3182 (mapcar
3183 (lambda (item)
3184 (let ((name (if prefix
3185 (concat prefix "." (car item))
3186 (car item)))
3187 (pos (cdr item)))
3188 (cond ((or (numberp pos) (markerp pos))
3189 (list (cons name pos)))
3190 ((listp pos)
3191 (cons
3192 (cons name (cdar pos))
3193 (python-imenu-create-flat-index (cddr item) name))))))
3194 (or alist
3195 (let* ((fn (lambda (type name) name))
3196 (python-imenu-format-item-label-function fn)
3197 (python-imenu-format-parent-item-label-function fn)
3198 (python-imenu-format-parent-item-jump-label-function fn))
3199 (python-imenu-create-index))))))
3200
3201 \f
3202 ;;; Misc helpers
3203
3204 (defun python-info-current-defun (&optional include-type)
3205 "Return name of surrounding function with Python compatible dotty syntax.
3206 Optional argument INCLUDE-TYPE indicates to include the type of the defun.
3207 This function is compatible to be used as
3208 `add-log-current-defun-function' since it returns nil if point is
3209 not inside a defun."
3210 (save-restriction
3211 (widen)
3212 (save-excursion
3213 (end-of-line 1)
3214 (let ((names)
3215 (starting-indentation (current-indentation))
3216 (starting-pos (point))
3217 (first-run t)
3218 (last-indent)
3219 (type))
3220 (catch 'exit
3221 (while (python-nav-beginning-of-defun 1)
3222 (when (save-match-data
3223 (and
3224 (or (not last-indent)
3225 (< (current-indentation) last-indent))
3226 (or
3227 (and first-run
3228 (save-excursion
3229 ;; If this is the first run, we may add
3230 ;; the current defun at point.
3231 (setq first-run nil)
3232 (goto-char starting-pos)
3233 (python-nav-beginning-of-statement)
3234 (beginning-of-line 1)
3235 (looking-at-p
3236 python-nav-beginning-of-defun-regexp)))
3237 (< starting-pos
3238 (save-excursion
3239 (let ((min-indent
3240 (+ (current-indentation)
3241 python-indent-offset)))
3242 (if (< starting-indentation min-indent)
3243 ;; If the starting indentation is not
3244 ;; within the min defun indent make the
3245 ;; check fail.
3246 starting-pos
3247 ;; Else go to the end of defun and add
3248 ;; up the current indentation to the
3249 ;; ending position.
3250 (python-nav-end-of-defun)
3251 (+ (point)
3252 (if (>= (current-indentation) min-indent)
3253 (1+ (current-indentation))
3254 0)))))))))
3255 (save-match-data (setq last-indent (current-indentation)))
3256 (if (or (not include-type) type)
3257 (setq names (cons (match-string-no-properties 1) names))
3258 (let ((match (split-string (match-string-no-properties 0))))
3259 (setq type (car match))
3260 (setq names (cons (cadr match) names)))))
3261 ;; Stop searching ASAP.
3262 (and (= (current-indentation) 0) (throw 'exit t))))
3263 (and names
3264 (concat (and type (format "%s " type))
3265 (mapconcat 'identity names ".")))))))
3266
3267 (defun python-info-current-symbol (&optional replace-self)
3268 "Return current symbol using dotty syntax.
3269 With optional argument REPLACE-SELF convert \"self\" to current
3270 parent defun name."
3271 (let ((name
3272 (and (not (python-syntax-comment-or-string-p))
3273 (with-syntax-table python-dotty-syntax-table
3274 (let ((sym (symbol-at-point)))
3275 (and sym
3276 (substring-no-properties (symbol-name sym))))))))
3277 (when name
3278 (if (not replace-self)
3279 name
3280 (let ((current-defun (python-info-current-defun)))
3281 (if (not current-defun)
3282 name
3283 (replace-regexp-in-string
3284 (python-rx line-start word-start "self" word-end ?.)
3285 (concat
3286 (mapconcat 'identity
3287 (butlast (split-string current-defun "\\."))
3288 ".") ".")
3289 name)))))))
3290
3291 (defun python-info-statement-starts-block-p ()
3292 "Return non-nil if current statement opens a block."
3293 (save-excursion
3294 (python-nav-beginning-of-statement)
3295 (looking-at (python-rx block-start))))
3296
3297 (defun python-info-statement-ends-block-p ()
3298 "Return non-nil if point is at end of block."
3299 (let ((end-of-block-pos (save-excursion
3300 (python-nav-end-of-block)))
3301 (end-of-statement-pos (save-excursion
3302 (python-nav-end-of-statement))))
3303 (and end-of-block-pos end-of-statement-pos
3304 (= end-of-block-pos end-of-statement-pos))))
3305
3306 (defun python-info-beginning-of-statement-p ()
3307 "Return non-nil if point is at beginning of statement."
3308 (= (point) (save-excursion
3309 (python-nav-beginning-of-statement)
3310 (point))))
3311
3312 (defun python-info-end-of-statement-p ()
3313 "Return non-nil if point is at end of statement."
3314 (= (point) (save-excursion
3315 (python-nav-end-of-statement)
3316 (point))))
3317
3318 (defun python-info-beginning-of-block-p ()
3319 "Return non-nil if point is at beginning of block."
3320 (and (python-info-beginning-of-statement-p)
3321 (python-info-statement-starts-block-p)))
3322
3323 (defun python-info-end-of-block-p ()
3324 "Return non-nil if point is at end of block."
3325 (and (python-info-end-of-statement-p)
3326 (python-info-statement-ends-block-p)))
3327
3328 (defun python-info-closing-block ()
3329 "Return the point of the block the current line closes."
3330 (let ((closing-word (save-excursion
3331 (back-to-indentation)
3332 (current-word)))
3333 (indentation (current-indentation)))
3334 (when (member closing-word python-indent-dedenters)
3335 (save-excursion
3336 (forward-line -1)
3337 (while (and (> (current-indentation) indentation)
3338 (not (bobp))
3339 (not (back-to-indentation))
3340 (forward-line -1)))
3341 (back-to-indentation)
3342 (cond
3343 ((not (equal indentation (current-indentation))) nil)
3344 ((string= closing-word "elif")
3345 (when (member (current-word) '("if" "elif"))
3346 (point-marker)))
3347 ((string= closing-word "else")
3348 (when (member (current-word) '("if" "elif" "except" "for" "while"))
3349 (point-marker)))
3350 ((string= closing-word "except")
3351 (when (member (current-word) '("try"))
3352 (point-marker)))
3353 ((string= closing-word "finally")
3354 (when (member (current-word) '("except" "else"))
3355 (point-marker))))))))
3356
3357 (defun python-info-closing-block-message (&optional closing-block-point)
3358 "Message the contents of the block the current line closes.
3359 With optional argument CLOSING-BLOCK-POINT use that instead of
3360 recalculating it calling `python-info-closing-block'."
3361 (let ((point (or closing-block-point (python-info-closing-block))))
3362 (when point
3363 (save-restriction
3364 (widen)
3365 (message "Closes %s" (save-excursion
3366 (goto-char point)
3367 (back-to-indentation)
3368 (buffer-substring
3369 (point) (line-end-position))))))))
3370
3371 (defun python-info-line-ends-backslash-p (&optional line-number)
3372 "Return non-nil if current line ends with backslash.
3373 With optional argument LINE-NUMBER, check that line instead."
3374 (save-excursion
3375 (save-restriction
3376 (widen)
3377 (when line-number
3378 (python-util-goto-line line-number))
3379 (while (and (not (eobp))
3380 (goto-char (line-end-position))
3381 (python-syntax-context 'paren)
3382 (not (equal (char-before (point)) ?\\)))
3383 (forward-line 1))
3384 (when (equal (char-before) ?\\)
3385 (point-marker)))))
3386
3387 (defun python-info-beginning-of-backslash (&optional line-number)
3388 "Return the point where the backslashed line start.
3389 Optional argument LINE-NUMBER forces the line number to check against."
3390 (save-excursion
3391 (save-restriction
3392 (widen)
3393 (when line-number
3394 (python-util-goto-line line-number))
3395 (when (python-info-line-ends-backslash-p)
3396 (while (save-excursion
3397 (goto-char (line-beginning-position))
3398 (python-syntax-context 'paren))
3399 (forward-line -1))
3400 (back-to-indentation)
3401 (point-marker)))))
3402
3403 (defun python-info-continuation-line-p ()
3404 "Check if current line is continuation of another.
3405 When current line is continuation of another return the point
3406 where the continued line ends."
3407 (save-excursion
3408 (save-restriction
3409 (widen)
3410 (let* ((context-type (progn
3411 (back-to-indentation)
3412 (python-syntax-context-type)))
3413 (line-start (line-number-at-pos))
3414 (context-start (when context-type
3415 (python-syntax-context context-type))))
3416 (cond ((equal context-type 'paren)
3417 ;; Lines inside a paren are always a continuation line
3418 ;; (except the first one).
3419 (python-util-forward-comment -1)
3420 (point-marker))
3421 ((member context-type '(string comment))
3422 ;; move forward an roll again
3423 (goto-char context-start)
3424 (python-util-forward-comment)
3425 (python-info-continuation-line-p))
3426 (t
3427 ;; Not within a paren, string or comment, the only way
3428 ;; we are dealing with a continuation line is that
3429 ;; previous line contains a backslash, and this can
3430 ;; only be the previous line from current
3431 (back-to-indentation)
3432 (python-util-forward-comment -1)
3433 (when (and (equal (1- line-start) (line-number-at-pos))
3434 (python-info-line-ends-backslash-p))
3435 (point-marker))))))))
3436
3437 (defun python-info-block-continuation-line-p ()
3438 "Return non-nil if current line is a continuation of a block."
3439 (save-excursion
3440 (when (python-info-continuation-line-p)
3441 (forward-line -1)
3442 (back-to-indentation)
3443 (when (looking-at (python-rx block-start))
3444 (point-marker)))))
3445
3446 (defun python-info-assignment-continuation-line-p ()
3447 "Check if current line is a continuation of an assignment.
3448 When current line is continuation of another with an assignment
3449 return the point of the first non-blank character after the
3450 operator."
3451 (save-excursion
3452 (when (python-info-continuation-line-p)
3453 (forward-line -1)
3454 (back-to-indentation)
3455 (when (and (not (looking-at (python-rx block-start)))
3456 (and (re-search-forward (python-rx not-simple-operator
3457 assignment-operator
3458 not-simple-operator)
3459 (line-end-position) t)
3460 (not (python-syntax-context-type))))
3461 (skip-syntax-forward "\s")
3462 (point-marker)))))
3463
3464 (defun python-info-looking-at-beginning-of-defun (&optional syntax-ppss)
3465 "Check if point is at `beginning-of-defun' using SYNTAX-PPSS."
3466 (and (not (python-syntax-context-type (or syntax-ppss (syntax-ppss))))
3467 (save-excursion
3468 (beginning-of-line 1)
3469 (looking-at python-nav-beginning-of-defun-regexp))))
3470
3471 (defun python-info-current-line-comment-p ()
3472 "Check if current line is a comment line."
3473 (char-equal
3474 (or (char-after (+ (line-beginning-position) (current-indentation))) ?_)
3475 ?#))
3476
3477 (defun python-info-current-line-empty-p ()
3478 "Check if current line is empty, ignoring whitespace."
3479 (save-excursion
3480 (beginning-of-line 1)
3481 (looking-at
3482 (python-rx line-start (* whitespace)
3483 (group (* not-newline))
3484 (* whitespace) line-end))
3485 (string-equal "" (match-string-no-properties 1))))
3486
3487 \f
3488 ;;; Utility functions
3489
3490 (defun python-util-goto-line (line-number)
3491 "Move point to LINE-NUMBER."
3492 (goto-char (point-min))
3493 (forward-line (1- line-number)))
3494
3495 ;; Stolen from org-mode
3496 (defun python-util-clone-local-variables (from-buffer &optional regexp)
3497 "Clone local variables from FROM-BUFFER.
3498 Optional argument REGEXP selects variables to clone and defaults
3499 to \"^python-\"."
3500 (mapc
3501 (lambda (pair)
3502 (and (symbolp (car pair))
3503 (string-match (or regexp "^python-")
3504 (symbol-name (car pair)))
3505 (set (make-local-variable (car pair))
3506 (cdr pair))))
3507 (buffer-local-variables from-buffer)))
3508
3509 (defun python-util-forward-comment (&optional direction)
3510 "Python mode specific version of `forward-comment'.
3511 Optional argument DIRECTION defines the direction to move to."
3512 (let ((comment-start (python-syntax-context 'comment))
3513 (factor (if (< (or direction 0) 0)
3514 -99999
3515 99999)))
3516 (when comment-start
3517 (goto-char comment-start))
3518 (forward-comment factor)))
3519
3520 (defun python-util-popn (lst n)
3521 "Return LST first N elements.
3522 N should be an integer, when it's a natural negative number its
3523 opposite is used. When N is bigger than the length of LST, the
3524 list is returned as is."
3525 (let* ((n (min (abs n)))
3526 (len (length lst))
3527 (acc))
3528 (if (> n len)
3529 lst
3530 (while (< 0 n)
3531 (setq acc (cons (car lst) acc)
3532 lst (cdr lst)
3533 n (1- n)))
3534 (reverse acc))))
3535
3536 \f
3537 ;;;###autoload
3538 (define-derived-mode python-mode prog-mode "Python"
3539 "Major mode for editing Python files.
3540
3541 \\{python-mode-map}
3542 Entry to this mode calls the value of `python-mode-hook'
3543 if that value is non-nil."
3544 (set (make-local-variable 'tab-width) 8)
3545 (set (make-local-variable 'indent-tabs-mode) nil)
3546
3547 (set (make-local-variable 'comment-start) "# ")
3548 (set (make-local-variable 'comment-start-skip) "#+\\s-*")
3549
3550 (set (make-local-variable 'parse-sexp-lookup-properties) t)
3551 (set (make-local-variable 'parse-sexp-ignore-comments) t)
3552
3553 (set (make-local-variable 'forward-sexp-function)
3554 'python-nav-forward-sexp)
3555
3556 (set (make-local-variable 'font-lock-defaults)
3557 '(python-font-lock-keywords nil nil nil nil))
3558
3559 (set (make-local-variable 'syntax-propertize-function)
3560 python-syntax-propertize-function)
3561
3562 (set (make-local-variable 'indent-line-function)
3563 #'python-indent-line-function)
3564 (set (make-local-variable 'indent-region-function) #'python-indent-region)
3565
3566 (set (make-local-variable 'paragraph-start) "\\s-*$")
3567 (set (make-local-variable 'fill-paragraph-function)
3568 'python-fill-paragraph)
3569
3570 (set (make-local-variable 'beginning-of-defun-function)
3571 #'python-nav-beginning-of-defun)
3572 (set (make-local-variable 'end-of-defun-function)
3573 #'python-nav-end-of-defun)
3574
3575 (add-hook 'completion-at-point-functions
3576 'python-completion-complete-at-point nil 'local)
3577
3578 (add-hook 'post-self-insert-hook
3579 'python-indent-post-self-insert-function nil 'local)
3580
3581 (set (make-local-variable 'imenu-create-index-function)
3582 #'python-imenu-create-index)
3583
3584 (set (make-local-variable 'add-log-current-defun-function)
3585 #'python-info-current-defun)
3586
3587 (add-hook 'which-func-functions #'python-info-current-defun nil t)
3588
3589 (set (make-local-variable 'skeleton-further-elements)
3590 '((abbrev-mode nil)
3591 (< '(backward-delete-char-untabify (min python-indent-offset
3592 (current-column))))
3593 (^ '(- (1+ (current-indentation))))))
3594
3595 (set (make-local-variable 'eldoc-documentation-function)
3596 #'python-eldoc-function)
3597
3598 (add-to-list 'hs-special-modes-alist
3599 `(python-mode "^\\s-*\\(?:def\\|class\\)\\>" nil "#"
3600 ,(lambda (arg)
3601 (python-nav-end-of-defun)) nil))
3602
3603 (set (make-local-variable 'mode-require-final-newline) t)
3604
3605 (set (make-local-variable 'outline-regexp)
3606 (python-rx (* space) block-start))
3607 (set (make-local-variable 'outline-heading-end-regexp) ":\\s-*\n")
3608 (set (make-local-variable 'outline-level)
3609 #'(lambda ()
3610 "`outline-level' function for Python mode."
3611 (1+ (/ (current-indentation) python-indent-offset))))
3612
3613 (python-skeleton-add-menu-items)
3614
3615 (make-local-variable 'python-shell-internal-buffer)
3616
3617 (when python-indent-guess-indent-offset
3618 (python-indent-guess-indent-offset)))
3619
3620
3621 (provide 'python)
3622
3623 ;; Local Variables:
3624 ;; coding: utf-8
3625 ;; indent-tabs-mode: nil
3626 ;; End:
3627
3628 ;;; python.el ends here