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