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