Merge from emacs--rel--22
[bpt/emacs.git] / lisp / progmodes / python.el
1 ;;; python.el --- silly walks for Python -*- coding: iso-8859-1 -*-
2
3 ;; Copyright (C) 2003, 2004, 2005, 2006, 2007, 2008 Free Software Foundation, Inc.
4
5 ;; Author: Dave Love <fx@gnu.org>
6 ;; Maintainer: FSF
7 ;; Created: Nov 2003
8 ;; Keywords: languages
9
10 ;; This file is part of GNU Emacs.
11
12 ;; GNU Emacs is free software; you can redistribute it and/or modify
13 ;; it under the terms of the GNU General Public License as published by
14 ;; the Free Software Foundation; either version 3, or (at your option)
15 ;; any later version.
16
17 ;; GNU Emacs is distributed in the hope that it will be useful,
18 ;; but WITHOUT ANY WARRANTY; without even the implied warranty of
19 ;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 ;; GNU General Public License for more details.
21
22 ;; You should have received a copy of the GNU General Public License
23 ;; along with GNU Emacs; see the file COPYING. If not, write to the
24 ;; Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
25 ;; Boston, MA 02110-1301, USA.
26
27 ;;; Commentary:
28
29 ;; Major mode for editing Python, with support for inferior processes.
30
31 ;; There is another Python mode, python-mode.el, used by XEmacs and
32 ;; maintained with Python. That isn't covered by an FSF copyright
33 ;; assignment, unlike this code, and seems not to be well-maintained
34 ;; for Emacs (though I've submitted fixes). This mode is rather
35 ;; simpler and is better in other ways. In particular, using the
36 ;; syntax functions with text properties maintained by font-lock makes
37 ;; it more correct with arbitrary string and comment contents.
38
39 ;; This doesn't implement all the facilities of python-mode.el. Some
40 ;; just need doing, e.g. catching exceptions in the inferior Python
41 ;; buffer (but see M-x pdb for debugging). [Actually, the use of
42 ;; `compilation-shell-minor-mode' now is probably enough for that.]
43 ;; Others don't seem appropriate. For instance,
44 ;; `forward-into-nomenclature' should be done separately, since it's
45 ;; not specific to Python, and I've installed a minor mode to do the
46 ;; job properly in Emacs 23. [CC mode 5.31 contains an incompatible
47 ;; feature, `c-subword-mode' which is intended to have a similar
48 ;; effect, but actually only affects word-oriented keybindings.]
49
50 ;; Other things seem more natural or canonical here, e.g. the
51 ;; {beginning,end}-of-defun implementation dealing with nested
52 ;; definitions, and the inferior mode following `cmuscheme'. (The
53 ;; inferior mode can find the source of errors from
54 ;; `python-send-region' & al via `compilation-shell-minor-mode'.)
55 ;; There is (limited) symbol completion using lookup in Python and
56 ;; Eldoc support also using the inferior process. Successive TABs
57 ;; cycle between possible indentations for the line.
58
59 ;; Even where it has similar facilities, this mode is incompatible
60 ;; with python-mode.el in some respects. For instance, various key
61 ;; bindings are changed to obey Emacs conventions.
62
63 ;; TODO: See various Fixmes below.
64
65 ;; Fixme: This doesn't support (the nascent) Python 3 .
66
67 ;;; Code:
68
69 (require 'comint)
70
71 (eval-when-compile
72 (require 'compile)
73 (require 'hippie-exp))
74
75 (require 'sym-comp)
76 (autoload 'comint-mode "comint")
77
78 (defgroup python nil
79 "Silly walks in the Python language."
80 :group 'languages
81 :version "22.1"
82 :link '(emacs-commentary-link "python"))
83 \f
84 ;;;###autoload
85 (add-to-list 'interpreter-mode-alist '("jython" . jython-mode))
86 ;;;###autoload
87 (add-to-list 'interpreter-mode-alist '("python" . python-mode))
88 ;;;###autoload
89 (add-to-list 'auto-mode-alist '("\\.py\\'" . python-mode))
90 (add-to-list 'same-window-buffer-names "*Python*")
91 \f
92 ;;;; Font lock
93
94 (defvar python-font-lock-keywords
95 `(,(rx symbol-start
96 ;; From v 2.5 reference, § keywords.
97 ;; def and class dealt with separately below
98 (or "and" "as" "assert" "break" "continue" "del" "elif" "else"
99 "except" "exec" "finally" "for" "from" "global" "if"
100 "import" "in" "is" "lambda" "not" "or" "pass" "print"
101 "raise" "return" "try" "while" "with" "yield"
102 ;; Not real keywords, but close enough to be fontified as such
103 "self" "True" "False")
104 symbol-end)
105 (,(rx symbol-start "None" symbol-end) ; see § Keywords in 2.5 manual
106 . font-lock-constant-face)
107 ;; Definitions
108 (,(rx symbol-start (group "class") (1+ space) (group (1+ (or word ?_))))
109 (1 font-lock-keyword-face) (2 font-lock-type-face))
110 (,(rx symbol-start (group "def") (1+ space) (group (1+ (or word ?_))))
111 (1 font-lock-keyword-face) (2 font-lock-function-name-face))
112 ;; Top-level assignments are worth highlighting.
113 (,(rx line-start (group (1+ (or word ?_))) (0+ space) "=")
114 (1 font-lock-variable-name-face))
115 (,(rx line-start (* (any " \t")) (group "@" (1+ (or word ?_)))) ; decorators
116 (1 font-lock-type-face))
117 ;; Built-ins. (The next three blocks are from
118 ;; `__builtin__.__dict__.keys()' in Python 2.5.1.) These patterns
119 ;; are debateable, but they at least help to spot possible
120 ;; shadowing of builtins.
121 (,(rx symbol-start (or
122 ;; exceptions
123 "ArithmeticError" "AssertionError" "AttributeError"
124 "BaseException" "DeprecationWarning" "EOFError"
125 "EnvironmentError" "Exception" "FloatingPointError"
126 "FutureWarning" "GeneratorExit" "IOError" "ImportError"
127 "ImportWarning" "IndentationError" "IndexError" "KeyError"
128 "KeyboardInterrupt" "LookupError" "MemoryError" "NameError"
129 "NotImplemented" "NotImplementedError" "OSError"
130 "OverflowError" "PendingDeprecationWarning" "ReferenceError"
131 "RuntimeError" "RuntimeWarning" "StandardError"
132 "StopIteration" "SyntaxError" "SyntaxWarning" "SystemError"
133 "SystemExit" "TabError" "TypeError" "UnboundLocalError"
134 "UnicodeDecodeError" "UnicodeEncodeError" "UnicodeError"
135 "UnicodeTranslateError" "UnicodeWarning" "UserWarning"
136 "ValueError" "Warning" "ZeroDivisionError") symbol-end)
137 . font-lock-type-face)
138 (,(rx (or line-start (not (any ". \t"))) (* (any " \t")) symbol-start
139 (group (or
140 ;; callable built-ins, fontified when not appearing as
141 ;; object attributes
142 "abs" "all" "any" "apply" "basestring" "bool" "buffer" "callable"
143 "chr" "classmethod" "cmp" "coerce" "compile" "complex"
144 "copyright" "credits" "delattr" "dict" "dir" "divmod"
145 "enumerate" "eval" "execfile" "exit" "file" "filter" "float"
146 "frozenset" "getattr" "globals" "hasattr" "hash" "help"
147 "hex" "id" "input" "int" "intern" "isinstance" "issubclass"
148 "iter" "len" "license" "list" "locals" "long" "map" "max"
149 "min" "object" "oct" "open" "ord" "pow" "property" "quit"
150 "range" "raw_input" "reduce" "reload" "repr" "reversed"
151 "round" "set" "setattr" "slice" "sorted" "staticmethod"
152 "str" "sum" "super" "tuple" "type" "unichr" "unicode" "vars"
153 "xrange" "zip")) symbol-end)
154 (1 font-lock-builtin-face))
155 (,(rx symbol-start (or
156 ;; other built-ins
157 "True" "False" "None" "Ellipsis"
158 "_" "__debug__" "__doc__" "__import__" "__name__") symbol-end)
159 . font-lock-builtin-face)))
160
161 (defconst python-font-lock-syntactic-keywords
162 ;; Make outer chars of matching triple-quote sequences into generic
163 ;; string delimiters. Fixme: Is there a better way?
164 ;; First avoid a sequence preceded by an odd number of backslashes.
165 `((,(rx (not (any ?\\))
166 ?\\ (* (and ?\\ ?\\))
167 (group (syntax string-quote))
168 (backref 1)
169 (group (backref 1)))
170 (2 ,(string-to-syntax "\""))) ; dummy
171 (,(rx (group (optional (any "uUrR"))) ; prefix gets syntax property
172 (optional (any "rR")) ; possible second prefix
173 (group (syntax string-quote)) ; maybe gets property
174 (backref 2) ; per first quote
175 (group (backref 2))) ; maybe gets property
176 (1 (python-quote-syntax 1))
177 (2 (python-quote-syntax 2))
178 (3 (python-quote-syntax 3)))
179 ;; This doesn't really help.
180 ;;; (,(rx (and ?\\ (group ?\n))) (1 " "))
181 ))
182
183 (defun python-quote-syntax (n)
184 "Put `syntax-table' property correctly on triple quote.
185 Used for syntactic keywords. N is the match number (1, 2 or 3)."
186 ;; Given a triple quote, we have to check the context to know
187 ;; whether this is an opening or closing triple or whether it's
188 ;; quoted anyhow, and should be ignored. (For that we need to do
189 ;; the same job as `syntax-ppss' to be correct and it seems to be OK
190 ;; to use it here despite initial worries.) We also have to sort
191 ;; out a possible prefix -- well, we don't _have_ to, but I think it
192 ;; should be treated as part of the string.
193
194 ;; Test cases:
195 ;; ur"""ar""" x='"' # """
196 ;; x = ''' """ ' a
197 ;; '''
198 ;; x '"""' x """ \"""" x
199 (save-excursion
200 (goto-char (match-beginning 0))
201 (cond
202 ;; Consider property for the last char if in a fenced string.
203 ((= n 3)
204 (let* ((font-lock-syntactic-keywords nil)
205 (syntax (syntax-ppss)))
206 (when (eq t (nth 3 syntax)) ; after unclosed fence
207 (goto-char (nth 8 syntax)) ; fence position
208 (skip-chars-forward "uUrR") ; skip any prefix
209 ;; Is it a matching sequence?
210 (if (eq (char-after) (char-after (match-beginning 2)))
211 (eval-when-compile (string-to-syntax "|"))))))
212 ;; Consider property for initial char, accounting for prefixes.
213 ((or (and (= n 2) ; leading quote (not prefix)
214 (= (match-beginning 1) (match-end 1))) ; prefix is null
215 (and (= n 1) ; prefix
216 (/= (match-beginning 1) (match-end 1)))) ; non-empty
217 (let ((font-lock-syntactic-keywords nil))
218 (unless (eq 'string (syntax-ppss-context (syntax-ppss)))
219 (eval-when-compile (string-to-syntax "|")))))
220 ;; Otherwise (we're in a non-matching string) the property is
221 ;; nil, which is OK.
222 )))
223
224 ;; This isn't currently in `font-lock-defaults' as probably not worth
225 ;; it -- we basically only mess with a few normally-symbol characters.
226
227 ;; (defun python-font-lock-syntactic-face-function (state)
228 ;; "`font-lock-syntactic-face-function' for Python mode.
229 ;; Returns the string or comment face as usual, with side effect of putting
230 ;; a `syntax-table' property on the inside of the string or comment which is
231 ;; the standard syntax table."
232 ;; (if (nth 3 state)
233 ;; (save-excursion
234 ;; (goto-char (nth 8 state))
235 ;; (condition-case nil
236 ;; (forward-sexp)
237 ;; (error nil))
238 ;; (put-text-property (1+ (nth 8 state)) (1- (point))
239 ;; 'syntax-table (standard-syntax-table))
240 ;; 'font-lock-string-face)
241 ;; (put-text-property (1+ (nth 8 state)) (line-end-position)
242 ;; 'syntax-table (standard-syntax-table))
243 ;; 'font-lock-comment-face))
244 \f
245 ;;;; Keymap and syntax
246
247 (defvar python-mode-map
248 (let ((map (make-sparse-keymap)))
249 ;; Mostly taken from python-mode.el.
250 (define-key map ":" 'python-electric-colon)
251 (define-key map "\177" 'python-backspace)
252 (define-key map "\C-c<" 'python-shift-left)
253 (define-key map "\C-c>" 'python-shift-right)
254 (define-key map "\C-c\C-k" 'python-mark-block)
255 (define-key map "\C-c\C-d" 'python-pdbtrack-toggle-stack-tracking)
256 (define-key map "\C-c\C-n" 'python-next-statement)
257 (define-key map "\C-c\C-p" 'python-previous-statement)
258 (define-key map "\C-c\C-u" 'python-beginning-of-block)
259 (define-key map "\C-c\C-f" 'python-describe-symbol)
260 (define-key map "\C-c\C-w" 'python-check)
261 (define-key map "\C-c\C-v" 'python-check) ; a la sgml-mode
262 (define-key map "\C-c\C-s" 'python-send-string)
263 (define-key map [?\C-\M-x] 'python-send-defun)
264 (define-key map "\C-c\C-r" 'python-send-region)
265 (define-key map "\C-c\M-r" 'python-send-region-and-go)
266 (define-key map "\C-c\C-c" 'python-send-buffer)
267 (define-key map "\C-c\C-z" 'python-switch-to-python)
268 (define-key map "\C-c\C-m" 'python-load-file)
269 (define-key map "\C-c\C-l" 'python-load-file) ; a la cmuscheme
270 (substitute-key-definition 'complete-symbol 'symbol-complete
271 map global-map)
272 (define-key map "\C-c\C-i" 'python-find-imports)
273 (define-key map "\C-c\C-t" 'python-expand-template)
274 (easy-menu-define python-menu map "Python Mode menu"
275 `("Python"
276 :help "Python-specific Features"
277 ["Shift region left" python-shift-left :active mark-active
278 :help "Shift by a single indentation step"]
279 ["Shift region right" python-shift-right :active mark-active
280 :help "Shift by a single indentation step"]
281 "-"
282 ["Mark block" python-mark-block
283 :help "Mark innermost block around point"]
284 ["Mark def/class" mark-defun
285 :help "Mark innermost definition around point"]
286 "-"
287 ["Start of block" python-beginning-of-block
288 :help "Go to start of innermost definition around point"]
289 ["End of block" python-end-of-block
290 :help "Go to end of innermost definition around point"]
291 ["Start of def/class" beginning-of-defun
292 :help "Go to start of innermost definition around point"]
293 ["End of def/class" end-of-defun
294 :help "Go to end of innermost definition around point"]
295 "-"
296 ("Templates..."
297 :help "Expand templates for compound statements"
298 :filter (lambda (&rest junk)
299 (mapcar (lambda (elt)
300 (vector (car elt) (cdr elt) t))
301 python-skeletons))) ; defined later
302 "-"
303 ["Start interpreter" python-shell
304 :help "Run `inferior' Python in separate buffer"]
305 ["Import/reload file" python-load-file
306 :help "Load into inferior Python session"]
307 ["Eval buffer" python-send-buffer
308 :help "Evaluate buffer en bloc in inferior Python session"]
309 ["Eval region" python-send-region :active mark-active
310 :help "Evaluate region en bloc in inferior Python session"]
311 ["Eval def/class" python-send-defun
312 :help "Evaluate current definition in inferior Python session"]
313 ["Switch to interpreter" python-switch-to-python
314 :help "Switch to inferior Python buffer"]
315 ["Set default process" python-set-proc
316 :help "Make buffer's inferior process the default"
317 :active (buffer-live-p python-buffer)]
318 ["Check file" python-check :help "Run pychecker"]
319 ["Debugger" pdb :help "Run pdb under GUD"]
320 "-"
321 ["Help on symbol" python-describe-symbol
322 :help "Use pydoc on symbol at point"]
323 ["Complete symbol" symbol-complete
324 :help "Complete (qualified) symbol before point"]
325 ["Find function" python-find-function
326 :help "Try to find source definition of function at point"]
327 ["Update imports" python-find-imports
328 :help "Update list of top-level imports for completion"]))
329 map))
330 ;; Fixme: add toolbar stuff for useful things like symbol help, send
331 ;; region, at least. (Shouldn't be specific to Python, obviously.)
332 ;; eric has items including: (un)indent, (un)comment, restart script,
333 ;; run script, debug script; also things for profiling, unit testing.
334
335 (defvar python-shell-map
336 (let ((map (copy-keymap comint-mode-map)))
337 (define-key map [tab] 'tab-to-tab-stop)
338 (define-key map "\C-c-" 'py-up-exception)
339 (define-key map "\C-c=" 'py-down-exception)
340 map)
341 "Keymap used in *Python* shell buffers.")
342
343 (defvar python-mode-syntax-table
344 (let ((table (make-syntax-table)))
345 ;; Give punctuation syntax to ASCII that normally has symbol
346 ;; syntax or has word syntax and isn't a letter.
347 (let ((symbol (string-to-syntax "_"))
348 (sst (standard-syntax-table)))
349 (dotimes (i 128)
350 (unless (= i ?_)
351 (if (equal symbol (aref sst i))
352 (modify-syntax-entry i "." table)))))
353 (modify-syntax-entry ?$ "." table)
354 (modify-syntax-entry ?% "." table)
355 ;; exceptions
356 (modify-syntax-entry ?# "<" table)
357 (modify-syntax-entry ?\n ">" table)
358 (modify-syntax-entry ?' "\"" table)
359 (modify-syntax-entry ?` "$" table)
360 table))
361 \f
362 ;;;; Utility stuff
363
364 (defsubst python-in-string/comment ()
365 "Return non-nil if point is in a Python literal (a comment or string)."
366 ;; We don't need to save the match data.
367 (nth 8 (syntax-ppss)))
368
369 (defconst python-space-backslash-table
370 (let ((table (copy-syntax-table python-mode-syntax-table)))
371 (modify-syntax-entry ?\\ " " table)
372 table)
373 "`python-mode-syntax-table' with backslash given whitespace syntax.")
374
375 (defun python-skip-comments/blanks (&optional backward)
376 "Skip comments and blank lines.
377 BACKWARD non-nil means go backwards, otherwise go forwards.
378 Backslash is treated as whitespace so that continued blank lines
379 are skipped. Doesn't move out of comments -- should be outside
380 or at end of line."
381 (let ((arg (if backward
382 ;; If we're in a comment (including on the trailing
383 ;; newline), forward-comment doesn't move backwards out
384 ;; of it. Don't set the syntax table round this bit!
385 (let ((syntax (syntax-ppss)))
386 (if (nth 4 syntax)
387 (goto-char (nth 8 syntax)))
388 (- (point-max)))
389 (point-max))))
390 (with-syntax-table python-space-backslash-table
391 (forward-comment arg))))
392
393 (defun python-backslash-continuation-line-p ()
394 "Non-nil if preceding line ends with backslash that is not in a comment."
395 (and (eq ?\\ (char-before (line-end-position 0)))
396 (not (syntax-ppss-context (syntax-ppss)))))
397
398 (defun python-continuation-line-p ()
399 "Return non-nil if current line continues a previous one.
400 The criteria are that the previous line ends in a backslash outside
401 comments and strings, or that point is within brackets/parens."
402 (or (python-backslash-continuation-line-p)
403 (let ((depth (syntax-ppss-depth
404 (save-excursion ; syntax-ppss with arg changes point
405 (syntax-ppss (line-beginning-position))))))
406 (or (> depth 0)
407 (if (< depth 0) ; Unbalanced brackets -- act locally
408 (save-excursion
409 (condition-case ()
410 (progn (backward-up-list) t) ; actually within brackets
411 (error nil))))))))
412
413 (defun python-comment-line-p ()
414 "Return non-nil iff current line has only a comment."
415 (save-excursion
416 (end-of-line)
417 (when (eq 'comment (syntax-ppss-context (syntax-ppss)))
418 (back-to-indentation)
419 (looking-at (rx (or (syntax comment-start) line-end))))))
420
421 (defun python-blank-line-p ()
422 "Return non-nil iff current line is blank."
423 (save-excursion
424 (beginning-of-line)
425 (looking-at "\\s-*$")))
426
427 (defun python-beginning-of-string ()
428 "Go to beginning of string around point.
429 Do nothing if not in string."
430 (let ((state (syntax-ppss)))
431 (when (eq 'string (syntax-ppss-context state))
432 (goto-char (nth 8 state)))))
433
434 (defun python-open-block-statement-p (&optional bos)
435 "Return non-nil if statement at point opens a block.
436 BOS non-nil means point is known to be at beginning of statement."
437 (save-excursion
438 (unless bos (python-beginning-of-statement))
439 (looking-at (rx (and (or "if" "else" "elif" "while" "for" "def"
440 "class" "try" "except" "finally" "with")
441 symbol-end)))))
442
443 (defun python-close-block-statement-p (&optional bos)
444 "Return non-nil if current line is a statement closing a block.
445 BOS non-nil means point is at beginning of statement.
446 The criteria are that the line isn't a comment or in string and
447 starts with keyword `raise', `break', `continue' or `pass'."
448 (save-excursion
449 (unless bos (python-beginning-of-statement))
450 (back-to-indentation)
451 (looking-at (rx (or "return" "raise" "break" "continue" "pass")
452 symbol-end))))
453
454 (defun python-outdent-p ()
455 "Return non-nil if current line should outdent a level."
456 (save-excursion
457 (back-to-indentation)
458 (and (looking-at (rx (and (or "else" "finally" "except" "elif")
459 symbol-end)))
460 (not (python-in-string/comment))
461 ;; Ensure there's a previous statement and move to it.
462 (zerop (python-previous-statement))
463 (not (python-close-block-statement-p t))
464 ;; Fixme: check this
465 (not (python-open-block-statement-p)))))
466 \f
467 ;;;; Indentation.
468
469 (defcustom python-indent 4
470 "Number of columns for a unit of indentation in Python mode.
471 See also `\\[python-guess-indent]'"
472 :group 'python
473 :type 'integer)
474 (put 'python-indent 'safe-local-variable 'integerp)
475
476 (defcustom python-guess-indent t
477 "Non-nil means Python mode guesses `python-indent' for the buffer."
478 :type 'boolean
479 :group 'python)
480
481 (defcustom python-indent-string-contents t
482 "Non-nil means indent contents of multi-line strings together.
483 This means indent them the same as the preceding non-blank line.
484 Otherwise preserve their indentation.
485
486 This only applies to `doc' strings, i.e. those that form statements;
487 the indentation is preserved in others."
488 :type '(choice (const :tag "Align with preceding" t)
489 (const :tag "Preserve indentation" nil))
490 :group 'python)
491
492 (defcustom python-honour-comment-indentation nil
493 "Non-nil means indent relative to preceding comment line.
494 Only do this for comments where the leading comment character is
495 followed by space. This doesn't apply to comment lines, which
496 are always indented in lines with preceding comments."
497 :type 'boolean
498 :group 'python)
499
500 (defcustom python-continuation-offset 4
501 "Number of columns of additional indentation for continuation lines.
502 Continuation lines follow a backslash-terminated line starting a
503 statement."
504 :group 'python
505 :type 'integer)
506
507
508 (defcustom python-default-interpreter 'cpython
509 "*Which Python interpreter is used by default.
510 The value for this variable can be either `cpython' or `jpython'.
511
512 When the value is `cpython', the variables `python-python-command' and
513 `python-python-command-args' are consulted to determine the interpreter
514 and arguments to use.
515
516 When the value is `jpython', the variables `python-jpython-command' and
517 `python-jpython-command-args' are consulted to determine the interpreter
518 and arguments to use.
519
520 Note that this variable is consulted only the first time that a Python
521 mode buffer is visited during an Emacs session. After that, use
522 \\[python-toggle-shells] to change the interpreter shell."
523 :type '(choice (const :tag "Python (a.k.a. CPython)" cpython)
524 (const :tag "JPython" jpython))
525 :group 'python)
526
527 (defcustom python-python-command-args '("-i")
528 "*List of string arguments to be used when starting a Python shell."
529 :type '(repeat string)
530 :group 'python)
531
532 (defcustom python-jython-command-args '("-i")
533 "*List of string arguments to be used when starting a Jython shell."
534 :type '(repeat string)
535 :group 'python
536 :tag "JPython Command Args")
537
538 ;; for toggling between CPython and JPython
539 (defvar python-which-shell nil)
540 (defvar python-which-args python-python-command-args)
541 (defvar python-which-bufname "Python")
542 (make-variable-buffer-local 'python-which-shell)
543 (make-variable-buffer-local 'python-which-args)
544 (make-variable-buffer-local 'python-which-bufname)
545
546 (defcustom python-pdbtrack-do-tracking-p t
547 "*Controls whether the pdbtrack feature is enabled or not.
548
549 When non-nil, pdbtrack is enabled in all comint-based buffers,
550 e.g. shell interaction buffers and the *Python* buffer.
551
552 When using pdb to debug a Python program, pdbtrack notices the
553 pdb prompt and presents the line in the source file where the
554 program is stopped in a pop-up buffer. It's similar to what
555 gud-mode does for debugging C programs with gdb, but without
556 having to restart the program."
557 :type 'boolean
558 :group 'python)
559 (make-variable-buffer-local 'python-pdbtrack-do-tracking-p)
560
561 (defcustom python-pdbtrack-minor-mode-string " PDB"
562 "*Minor-mode sign to be displayed when pdbtrack is active."
563 :type 'string
564 :group 'python)
565
566 ;; Add a designator to the minor mode strings
567 (or (assq 'python-pdbtrack-is-tracking-p minor-mode-alist)
568 (push '(python-pdbtrack-is-tracking-p python-pdbtrack-minor-mode-string)
569 minor-mode-alist))
570
571 ;; Bind python-file-queue before installing the kill-emacs-hook.
572 (defvar python-file-queue nil
573 "Queue of Python temp files awaiting execution.
574 Currently-active file is at the head of the list.")
575
576 (defvar python-pdbtrack-is-tracking-p nil)
577
578 (defconst python-pdbtrack-stack-entry-regexp
579 "^> \\(.*\\)(\\([0-9]+\\))\\([?a-zA-Z0-9_]+\\)()"
580 "Regular expression pdbtrack uses to find a stack trace entry.")
581
582 (defconst python-pdbtrack-input-prompt "\n[(<]*[Pp]db[>)]+ "
583 "Regular expression pdbtrack uses to recognize a pdb prompt.")
584
585 (defconst python-pdbtrack-track-range 10000
586 "Max number of characters from end of buffer to search for stack entry.")
587
588 (defun python-guess-indent ()
589 "Guess step for indentation of current buffer.
590 Set `python-indent' locally to the value guessed."
591 (interactive)
592 (save-excursion
593 (save-restriction
594 (widen)
595 (goto-char (point-min))
596 (let (done indent)
597 (while (and (not done) (not (eobp)))
598 (when (and (re-search-forward (rx ?: (0+ space)
599 (or (syntax comment-start)
600 line-end))
601 nil 'move)
602 (python-open-block-statement-p))
603 (save-excursion
604 (python-beginning-of-statement)
605 (let ((initial (current-indentation)))
606 (if (zerop (python-next-statement))
607 (setq indent (- (current-indentation) initial)))
608 (if (and indent (>= indent 2) (<= indent 8)) ; sanity check
609 (setq done t))))))
610 (when done
611 (when (/= indent (default-value 'python-indent))
612 (set (make-local-variable 'python-indent) indent)
613 (unless (= tab-width python-indent)
614 (setq indent-tabs-mode nil)))
615 indent)))))
616
617 ;; Alist of possible indentations and start of statement they would
618 ;; close. Used in indentation cycling (below).
619 (defvar python-indent-list nil
620 "Internal use.")
621 ;; Length of the above
622 (defvar python-indent-list-length nil
623 "Internal use.")
624 ;; Current index into the alist.
625 (defvar python-indent-index nil
626 "Internal use.")
627
628 (defun python-calculate-indentation ()
629 "Calculate Python indentation for line at point."
630 (setq python-indent-list nil
631 python-indent-list-length 1)
632 (save-excursion
633 (beginning-of-line)
634 (let ((syntax (syntax-ppss))
635 start)
636 (cond
637 ((eq 'string (syntax-ppss-context syntax)) ; multi-line string
638 (if (not python-indent-string-contents)
639 (current-indentation)
640 ;; Only respect `python-indent-string-contents' in doc
641 ;; strings (defined as those which form statements).
642 (if (not (save-excursion
643 (python-beginning-of-statement)
644 (looking-at (rx (or (syntax string-delimiter)
645 (syntax string-quote))))))
646 (current-indentation)
647 ;; Find indentation of preceding non-blank line within string.
648 (setq start (nth 8 syntax))
649 (forward-line -1)
650 (while (and (< start (point)) (looking-at "\\s-*$"))
651 (forward-line -1))
652 (current-indentation))))
653 ((python-continuation-line-p) ; after backslash, or bracketed
654 (let ((point (point))
655 (open-start (cadr syntax))
656 (backslash (python-backslash-continuation-line-p))
657 (colon (eq ?: (char-before (1- (line-beginning-position))))))
658 (if open-start
659 ;; Inside bracketed expression.
660 (progn
661 (goto-char (1+ open-start))
662 ;; Look for first item in list (preceding point) and
663 ;; align with it, if found.
664 (if (with-syntax-table python-space-backslash-table
665 (let ((parse-sexp-ignore-comments t))
666 (condition-case ()
667 (progn (forward-sexp)
668 (backward-sexp)
669 (< (point) point))
670 (error nil))))
671 ;; Extra level if we're backslash-continued or
672 ;; following a key.
673 (if (or backslash colon)
674 (+ python-indent (current-column))
675 (current-column))
676 ;; Otherwise indent relative to statement start, one
677 ;; level per bracketing level.
678 (goto-char (1+ open-start))
679 (python-beginning-of-statement)
680 (+ (current-indentation) (* (car syntax) python-indent))))
681 ;; Otherwise backslash-continued.
682 (forward-line -1)
683 (if (python-continuation-line-p)
684 ;; We're past first continuation line. Align with
685 ;; previous line.
686 (current-indentation)
687 ;; First continuation line. Indent one step, with an
688 ;; extra one if statement opens a block.
689 (python-beginning-of-statement)
690 (+ (current-indentation) python-continuation-offset
691 (if (python-open-block-statement-p t)
692 python-indent
693 0))))))
694 ((bobp) 0)
695 ;; Fixme: Like python-mode.el; not convinced by this.
696 ((looking-at (rx (0+ space) (syntax comment-start)
697 (not (any " \t\n")))) ; non-indentable comment
698 (current-indentation))
699 ((and python-honour-comment-indentation
700 ;; Back over whitespace, newlines, non-indentable comments.
701 (catch 'done
702 (while (cond ((bobp) nil)
703 ((not (forward-comment -1))
704 nil) ; not at comment start
705 ;; Now at start of comment -- trailing one?
706 ((/= (current-column) (current-indentation))
707 nil)
708 ;; Indentable comment, like python-mode.el?
709 ((and (looking-at (rx (syntax comment-start)
710 (or space line-end)))
711 (/= 0 (current-column)))
712 (throw 'done (current-column)))
713 ;; Else skip it (loop).
714 (t))))))
715 (t
716 (python-indentation-levels)
717 ;; Prefer to indent comments with an immediately-following
718 ;; statement, e.g.
719 ;; ...
720 ;; # ...
721 ;; def ...
722 (when (and (> python-indent-list-length 1)
723 (python-comment-line-p))
724 (forward-line)
725 (unless (python-comment-line-p)
726 (let ((elt (assq (current-indentation) python-indent-list)))
727 (setq python-indent-list
728 (nconc (delete elt python-indent-list)
729 (list elt))))))
730 (caar (last python-indent-list)))))))
731
732 ;;;; Cycling through the possible indentations with successive TABs.
733
734 ;; These don't need to be buffer-local since they're only relevant
735 ;; during a cycle.
736
737 (defun python-initial-text ()
738 "Text of line following indentation and ignoring any trailing comment."
739 (save-excursion
740 (buffer-substring (progn
741 (back-to-indentation)
742 (point))
743 (progn
744 (end-of-line)
745 (forward-comment -1)
746 (point)))))
747
748 (defconst python-block-pairs
749 '(("else" "if" "elif" "while" "for" "try" "except")
750 ("elif" "if" "elif")
751 ("except" "try" "except")
752 ("finally" "try"))
753 "Alist of keyword matches.
754 The car of an element is a keyword introducing a statement which
755 can close a block opened by a keyword in the cdr.")
756
757 (defun python-first-word ()
758 "Return first word (actually symbol) on the line."
759 (save-excursion
760 (back-to-indentation)
761 (current-word t)))
762
763 (defun python-indentation-levels ()
764 "Return a list of possible indentations for this line.
765 It is assumed not to be a continuation line or in a multi-line string.
766 Includes the default indentation and those which would close all
767 enclosing blocks. Elements of the list are actually pairs:
768 \(INDENTATION . TEXT), where TEXT is the initial text of the
769 corresponding block opening (or nil)."
770 (save-excursion
771 (let ((initial "")
772 levels indent)
773 ;; Only one possibility immediately following a block open
774 ;; statement, assuming it doesn't have a `suite' on the same line.
775 (cond
776 ((save-excursion (and (python-previous-statement)
777 (python-open-block-statement-p t)
778 (setq indent (current-indentation))
779 ;; Check we don't have something like:
780 ;; if ...: ...
781 (if (progn (python-end-of-statement)
782 (python-skip-comments/blanks t)
783 (eq ?: (char-before)))
784 (setq indent (+ python-indent indent)))))
785 (push (cons indent initial) levels))
786 ;; Only one possibility for comment line immediately following
787 ;; another.
788 ((save-excursion
789 (when (python-comment-line-p)
790 (forward-line -1)
791 (if (python-comment-line-p)
792 (push (cons (current-indentation) initial) levels)))))
793 ;; Fixme: Maybe have a case here which indents (only) first
794 ;; line after a lambda.
795 (t
796 (let ((start (car (assoc (python-first-word) python-block-pairs))))
797 (python-previous-statement)
798 ;; Is this a valid indentation for the line of interest?
799 (unless (or (if start ; potentially only outdentable
800 ;; Check for things like:
801 ;; if ...: ...
802 ;; else ...:
803 ;; where the second line need not be outdented.
804 (not (member (python-first-word)
805 (cdr (assoc start
806 python-block-pairs)))))
807 ;; Not sensible to indent to the same level as
808 ;; previous `return' &c.
809 (python-close-block-statement-p))
810 (push (cons (current-indentation) (python-initial-text))
811 levels))
812 (while (python-beginning-of-block)
813 (when (or (not start)
814 (member (python-first-word)
815 (cdr (assoc start python-block-pairs))))
816 (push (cons (current-indentation) (python-initial-text))
817 levels))))))
818 (prog1 (or levels (setq levels '((0 . ""))))
819 (setq python-indent-list levels
820 python-indent-list-length (length python-indent-list))))))
821
822 ;; This is basically what `python-indent-line' would be if we didn't
823 ;; do the cycling.
824 (defun python-indent-line-1 (&optional leave)
825 "Subroutine of `python-indent-line'.
826 Does non-repeated indentation. LEAVE non-nil means leave
827 indentation if it is valid, i.e. one of the positions returned by
828 `python-calculate-indentation'."
829 (let ((target (python-calculate-indentation))
830 (pos (- (point-max) (point))))
831 (if (or (= target (current-indentation))
832 ;; Maybe keep a valid indentation.
833 (and leave python-indent-list
834 (assq (current-indentation) python-indent-list)))
835 (if (< (current-column) (current-indentation))
836 (back-to-indentation))
837 (beginning-of-line)
838 (delete-horizontal-space)
839 (indent-to target)
840 (if (> (- (point-max) pos) (point))
841 (goto-char (- (point-max) pos))))))
842
843 (defun python-indent-line ()
844 "Indent current line as Python code.
845 When invoked via `indent-for-tab-command', cycle through possible
846 indentations for current line. The cycle is broken by a command
847 different from `indent-for-tab-command', i.e. successive TABs do
848 the cycling."
849 (interactive)
850 (if (and (eq this-command 'indent-for-tab-command)
851 (eq last-command this-command))
852 (if (= 1 python-indent-list-length)
853 (message "Sole indentation")
854 (progn (setq python-indent-index
855 (% (1+ python-indent-index) python-indent-list-length))
856 (beginning-of-line)
857 (delete-horizontal-space)
858 (indent-to (car (nth python-indent-index python-indent-list)))
859 (if (python-block-end-p)
860 (let ((text (cdr (nth python-indent-index
861 python-indent-list))))
862 (if text
863 (message "Closes: %s" text))))))
864 (python-indent-line-1)
865 (setq python-indent-index (1- python-indent-list-length))))
866
867 (defun python-indent-region (start end)
868 "`indent-region-function' for Python.
869 Leaves validly-indented lines alone, i.e. doesn't indent to
870 another valid position."
871 (save-excursion
872 (goto-char end)
873 (setq end (point-marker))
874 (goto-char start)
875 (or (bolp) (forward-line 1))
876 (while (< (point) end)
877 (or (and (bolp) (eolp))
878 (python-indent-line-1 t))
879 (forward-line 1))
880 (move-marker end nil)))
881
882 (defun python-block-end-p ()
883 "Non-nil if this is a line in a statement closing a block,
884 or a blank line indented to where it would close a block."
885 (and (not (python-comment-line-p))
886 (or (python-close-block-statement-p t)
887 (< (current-indentation)
888 (save-excursion
889 (python-previous-statement)
890 (current-indentation))))))
891 \f
892 ;;;; Movement.
893
894 ;; Fixme: Define {for,back}ward-sexp-function? Maybe skip units like
895 ;; block, statement, depending on context.
896
897 (defun python-beginning-of-defun ()
898 "`beginning-of-defun-function' for Python.
899 Finds beginning of innermost nested class or method definition.
900 Returns the name of the definition found at the end, or nil if
901 reached start of buffer."
902 (let ((ci (current-indentation))
903 (def-re (rx line-start (0+ space) (or "def" "class") (1+ space)
904 (group (1+ (or word (syntax symbol))))))
905 found lep) ;; def-line
906 (if (python-comment-line-p)
907 (setq ci most-positive-fixnum))
908 (while (and (not (bobp)) (not found))
909 ;; Treat bol at beginning of function as outside function so
910 ;; that successive C-M-a makes progress backwards.
911 ;;(setq def-line (looking-at def-re))
912 (unless (bolp) (end-of-line))
913 (setq lep (line-end-position))
914 (if (and (re-search-backward def-re nil 'move)
915 ;; Must be less indented or matching top level, or
916 ;; equally indented if we started on a definition line.
917 (let ((in (current-indentation)))
918 (or (and (zerop ci) (zerop in))
919 (= lep (line-end-position)) ; on initial line
920 ;; Not sure why it was like this -- fails in case of
921 ;; last internal function followed by first
922 ;; non-def statement of the main body.
923 ;; (and def-line (= in ci))
924 (= in ci)
925 (< in ci)))
926 (not (python-in-string/comment)))
927 (setq found t)))
928 found))
929
930 (defun python-end-of-defun ()
931 "`end-of-defun-function' for Python.
932 Finds end of innermost nested class or method definition."
933 (let ((orig (point))
934 (pattern (rx line-start (0+ space) (or "def" "class") space)))
935 ;; Go to start of current block and check whether it's at top
936 ;; level. If it is, and not a block start, look forward for
937 ;; definition statement.
938 (when (python-comment-line-p)
939 (end-of-line)
940 (forward-comment most-positive-fixnum))
941 (if (not (python-open-block-statement-p))
942 (python-beginning-of-block))
943 (if (zerop (current-indentation))
944 (unless (python-open-block-statement-p)
945 (while (and (re-search-forward pattern nil 'move)
946 (python-in-string/comment))) ; just loop
947 (unless (eobp)
948 (beginning-of-line)))
949 ;; Don't move before top-level statement that would end defun.
950 (end-of-line)
951 (python-beginning-of-defun))
952 ;; If we got to the start of buffer, look forward for
953 ;; definition statement.
954 (if (and (bobp) (not (looking-at "def\\|class")))
955 (while (and (not (eobp))
956 (re-search-forward pattern nil 'move)
957 (python-in-string/comment)))) ; just loop
958 ;; We're at a definition statement (or end-of-buffer).
959 (unless (eobp)
960 (python-end-of-block)
961 ;; Count trailing space in defun (but not trailing comments).
962 (skip-syntax-forward " >")
963 (unless (eobp) ; e.g. missing final newline
964 (beginning-of-line)))
965 ;; Catch pathological cases like this, where the beginning-of-defun
966 ;; skips to a definition we're not in:
967 ;; if ...:
968 ;; ...
969 ;; else:
970 ;; ... # point here
971 ;; ...
972 ;; def ...
973 (if (< (point) orig)
974 (goto-char (point-max)))))
975
976 (defun python-beginning-of-statement ()
977 "Go to start of current statement.
978 Accounts for continuation lines, multi-line strings, and
979 multi-line bracketed expressions."
980 (beginning-of-line)
981 (python-beginning-of-string)
982 (let ((point (point)))
983 (while (and (python-continuation-line-p)
984 (> point (setq point (point))))
985 (beginning-of-line)
986 (if (python-backslash-continuation-line-p)
987 (progn
988 (forward-line -1)
989 (while (python-backslash-continuation-line-p)
990 (forward-line -1)))
991 (python-beginning-of-string)
992 (python-skip-out))))
993 (back-to-indentation))
994
995 (defun python-skip-out (&optional forward syntax)
996 "Skip out of any nested brackets.
997 Skip forward if FORWARD is non-nil, else backward.
998 If SYNTAX is non-nil it is the state returned by `syntax-ppss' at point.
999 Return non-nil iff skipping was done."
1000 (let ((depth (syntax-ppss-depth (or syntax (syntax-ppss))))
1001 (forward (if forward -1 1)))
1002 (unless (zerop depth)
1003 (if (> depth 0)
1004 ;; Skip forward out of nested brackets.
1005 (condition-case () ; beware invalid syntax
1006 (progn (backward-up-list (* forward depth)) t)
1007 (error nil))
1008 ;; Invalid syntax (too many closed brackets).
1009 ;; Skip out of as many as possible.
1010 (let (done)
1011 (while (condition-case ()
1012 (progn (backward-up-list forward)
1013 (setq done t))
1014 (error nil)))
1015 done)))))
1016
1017 (defun python-end-of-statement ()
1018 "Go to the end of the current statement and return point.
1019 Usually this is the start of the next line, but if this is a
1020 multi-line statement we need to skip over the continuation lines.
1021 On a comment line, go to end of line."
1022 (end-of-line)
1023 (while (let (comment)
1024 ;; Move past any enclosing strings and sexps, or stop if
1025 ;; we're in a comment.
1026 (while (let ((s (syntax-ppss)))
1027 (cond ((eq 'comment (syntax-ppss-context s))
1028 (setq comment t)
1029 nil)
1030 ((eq 'string (syntax-ppss-context s))
1031 ;; Go to start of string and skip it.
1032 (let ((pos (point)))
1033 (goto-char (nth 8 s))
1034 (condition-case () ; beware invalid syntax
1035 (progn (forward-sexp) t)
1036 ;; If there's a mismatched string, make sure
1037 ;; we still overall move *forward*.
1038 (error (goto-char pos) (end-of-line)))))
1039 ((python-skip-out t s))))
1040 (end-of-line))
1041 (unless comment
1042 (eq ?\\ (char-before)))) ; Line continued?
1043 (end-of-line 2)) ; Try next line.
1044 (point))
1045
1046 (defun python-previous-statement (&optional count)
1047 "Go to start of previous statement.
1048 With argument COUNT, do it COUNT times. Stop at beginning of buffer.
1049 Return count of statements left to move."
1050 (interactive "p")
1051 (unless count (setq count 1))
1052 (if (< count 0)
1053 (python-next-statement (- count))
1054 (python-beginning-of-statement)
1055 (while (and (> count 0) (not (bobp)))
1056 (python-skip-comments/blanks t)
1057 (python-beginning-of-statement)
1058 (unless (bobp) (setq count (1- count))))
1059 count))
1060
1061 (defun python-next-statement (&optional count)
1062 "Go to start of next statement.
1063 With argument COUNT, do it COUNT times. Stop at end of buffer.
1064 Return count of statements left to move."
1065 (interactive "p")
1066 (unless count (setq count 1))
1067 (if (< count 0)
1068 (python-previous-statement (- count))
1069 (beginning-of-line)
1070 (let (bogus)
1071 (while (and (> count 0) (not (eobp)) (not bogus))
1072 (python-end-of-statement)
1073 (python-skip-comments/blanks)
1074 (if (eq 'string (syntax-ppss-context (syntax-ppss)))
1075 (setq bogus t)
1076 (unless (eobp)
1077 (setq count (1- count))))))
1078 count))
1079
1080 (defun python-beginning-of-block (&optional arg)
1081 "Go to start of current block.
1082 With numeric arg, do it that many times. If ARG is negative, call
1083 `python-end-of-block' instead.
1084 If point is on the first line of a block, use its outer block.
1085 If current statement is in column zero, don't move and return nil.
1086 Otherwise return non-nil."
1087 (interactive "p")
1088 (unless arg (setq arg 1))
1089 (cond
1090 ((zerop arg))
1091 ((< arg 0) (python-end-of-block (- arg)))
1092 (t
1093 (let ((point (point)))
1094 (if (or (python-comment-line-p)
1095 (python-blank-line-p))
1096 (python-skip-comments/blanks t))
1097 (python-beginning-of-statement)
1098 (let ((ci (current-indentation)))
1099 (if (zerop ci)
1100 (not (goto-char point)) ; return nil
1101 ;; Look upwards for less indented statement.
1102 (if (catch 'done
1103 ;;; This is slower than the below.
1104 ;;; (while (zerop (python-previous-statement))
1105 ;;; (when (and (< (current-indentation) ci)
1106 ;;; (python-open-block-statement-p t))
1107 ;;; (beginning-of-line)
1108 ;;; (throw 'done t)))
1109 (while (and (zerop (forward-line -1)))
1110 (when (and (< (current-indentation) ci)
1111 (not (python-comment-line-p))
1112 ;; Move to beginning to save effort in case
1113 ;; this is in string.
1114 (progn (python-beginning-of-statement) t)
1115 (python-open-block-statement-p t))
1116 (beginning-of-line)
1117 (throw 'done t)))
1118 (not (goto-char point))) ; Failed -- return nil
1119 (python-beginning-of-block (1- arg)))))))))
1120
1121 (defun python-end-of-block (&optional arg)
1122 "Go to end of current block.
1123 With numeric arg, do it that many times. If ARG is negative,
1124 call `python-beginning-of-block' instead.
1125 If current statement is in column zero and doesn't open a block,
1126 don't move and return nil. Otherwise return t."
1127 (interactive "p")
1128 (unless arg (setq arg 1))
1129 (if (< arg 0)
1130 (python-beginning-of-block (- arg))
1131 (while (and (> arg 0)
1132 (let* ((point (point))
1133 (_ (if (python-comment-line-p)
1134 (python-skip-comments/blanks t)))
1135 (ci (current-indentation))
1136 (open (python-open-block-statement-p)))
1137 (if (and (zerop ci) (not open))
1138 (not (goto-char point))
1139 (catch 'done
1140 (while (zerop (python-next-statement))
1141 (when (or (and open (<= (current-indentation) ci))
1142 (< (current-indentation) ci))
1143 (python-skip-comments/blanks t)
1144 (beginning-of-line 2)
1145 (throw 'done t)))))))
1146 (setq arg (1- arg)))
1147 (zerop arg)))
1148
1149 (defvar python-which-func-length-limit 40
1150 "Non-strict length limit for `python-which-func' output.")
1151
1152 (defun python-which-func ()
1153 (let ((function-name (python-current-defun python-which-func-length-limit)))
1154 (set-text-properties 0 (length function-name) nil function-name)
1155 function-name))
1156
1157 \f
1158 ;;;; Imenu.
1159
1160 ;; For possibily speeding this up, here's the top of the ELP profile
1161 ;; for rescanning pydoc.py (2.2k lines, 90kb):
1162 ;; Function Name Call Count Elapsed Time Average Time
1163 ;; ==================================== ========== ============= ============
1164 ;; python-imenu-create-index 156 2.430906 0.0155827307
1165 ;; python-end-of-defun 155 1.2718260000 0.0082053290
1166 ;; python-end-of-block 155 1.1898689999 0.0076765741
1167 ;; python-next-statement 2970 1.024717 0.0003450225
1168 ;; python-end-of-statement 2970 0.4332190000 0.0001458649
1169 ;; python-beginning-of-defun 265 0.0918479999 0.0003465962
1170 ;; python-skip-comments/blanks 3125 0.0753319999 2.410...e-05
1171
1172 (defvar python-recursing)
1173 (defun python-imenu-create-index ()
1174 "`imenu-create-index-function' for Python.
1175
1176 Makes nested Imenu menus from nested `class' and `def' statements.
1177 The nested menus are headed by an item referencing the outer
1178 definition; it has a space prepended to the name so that it sorts
1179 first with `imenu--sort-by-name' (though, unfortunately, sub-menus
1180 precede it)."
1181 (unless (boundp 'python-recursing) ; dynamically bound below
1182 ;; Normal call from Imenu.
1183 (goto-char (point-min))
1184 ;; Without this, we can get an infloop if the buffer isn't all
1185 ;; fontified. I guess this is really a bug in syntax.el. OTOH,
1186 ;; _with_ this, imenu doesn't immediately work; I can't figure out
1187 ;; what's going on, but it must be something to do with timers in
1188 ;; font-lock.
1189 ;; This can't be right, especially not when jit-lock is not used. --Stef
1190 ;; (unless (get-text-property (1- (point-max)) 'fontified)
1191 ;; (font-lock-fontify-region (point-min) (point-max)))
1192 )
1193 (let (index-alist) ; accumulated value to return
1194 (while (re-search-forward
1195 (rx line-start (0+ space) ; leading space
1196 (or (group "def") (group "class")) ; type
1197 (1+ space) (group (1+ (or word ?_)))) ; name
1198 nil t)
1199 (unless (python-in-string/comment)
1200 (let ((pos (match-beginning 0))
1201 (name (match-string-no-properties 3)))
1202 (if (match-beginning 2) ; def or class?
1203 (setq name (concat "class " name)))
1204 (save-restriction
1205 (narrow-to-defun)
1206 (let* ((python-recursing t)
1207 (sublist (python-imenu-create-index)))
1208 (if sublist
1209 (progn (push (cons (concat " " name) pos) sublist)
1210 (push (cons name sublist) index-alist))
1211 (push (cons name pos) index-alist)))))))
1212 (unless (boundp 'python-recursing)
1213 ;; Look for module variables.
1214 (let (vars)
1215 (goto-char (point-min))
1216 (while (re-search-forward
1217 (rx line-start (group (1+ (or word ?_))) (0+ space) "=")
1218 nil t)
1219 (unless (python-in-string/comment)
1220 (push (cons (match-string 1) (match-beginning 1))
1221 vars)))
1222 (setq index-alist (nreverse index-alist))
1223 (if vars
1224 (push (cons "Module variables"
1225 (nreverse vars))
1226 index-alist))))
1227 index-alist))
1228 \f
1229 ;;;; `Electric' commands.
1230
1231 (defun python-electric-colon (arg)
1232 "Insert a colon and maybe outdent the line if it is a statement like `else'.
1233 With numeric ARG, just insert that many colons. With \\[universal-argument],
1234 just insert a single colon."
1235 (interactive "*P")
1236 (self-insert-command (if (not (integerp arg)) 1 arg))
1237 (and (not arg)
1238 (eolp)
1239 (python-outdent-p)
1240 (not (python-in-string/comment))
1241 (> (current-indentation) (python-calculate-indentation))
1242 (python-indent-line))) ; OK, do it
1243 (put 'python-electric-colon 'delete-selection t)
1244
1245 (defun python-backspace (arg)
1246 "Maybe delete a level of indentation on the current line.
1247 Do so if point is at the end of the line's indentation outside
1248 strings and comments.
1249 Otherwise just call `backward-delete-char-untabify'.
1250 Repeat ARG times."
1251 (interactive "*p")
1252 (if (or (/= (current-indentation) (current-column))
1253 (bolp)
1254 (python-continuation-line-p)
1255 (python-in-string/comment))
1256 (backward-delete-char-untabify arg)
1257 ;; Look for the largest valid indentation which is smaller than
1258 ;; the current indentation.
1259 (let ((indent 0)
1260 (ci (current-indentation))
1261 (indents (python-indentation-levels))
1262 initial)
1263 (dolist (x indents)
1264 (if (< (car x) ci)
1265 (setq indent (max indent (car x)))))
1266 (setq initial (cdr (assq indent indents)))
1267 (if (> (length initial) 0)
1268 (message "Closes %s" initial))
1269 (delete-horizontal-space)
1270 (indent-to indent))))
1271 (put 'python-backspace 'delete-selection 'supersede)
1272 \f
1273 ;;;; pychecker
1274
1275 (defcustom python-check-command "pychecker --stdlib"
1276 "Command used to check a Python file."
1277 :type 'string
1278 :group 'python)
1279
1280 (defvar python-saved-check-command nil
1281 "Internal use.")
1282
1283 ;; After `sgml-validate-command'.
1284 (defun python-check (command)
1285 "Check a Python file (default current buffer's file).
1286 Runs COMMAND, a shell command, as if by `compile'.
1287 See `python-check-command' for the default."
1288 (interactive
1289 (list (read-string "Checker command: "
1290 (or python-saved-check-command
1291 (concat python-check-command " "
1292 (let ((name (buffer-file-name)))
1293 (if name
1294 (file-name-nondirectory name))))))))
1295 (setq python-saved-check-command command)
1296 (require 'compile) ;To define compilation-* variables.
1297 (save-some-buffers (not compilation-ask-about-save) nil)
1298 (let ((compilation-error-regexp-alist
1299 (cons '("(\\([^,]+\\), line \\([0-9]+\\))" 1 2)
1300 compilation-error-regexp-alist)))
1301 (compilation-start command)))
1302 \f
1303 ;;;; Inferior mode stuff (following cmuscheme).
1304
1305 ;; Fixme: Make sure we can work with IPython.
1306
1307 (defcustom python-python-command "python"
1308 "Shell command to run Python interpreter.
1309 Any arguments can't contain whitespace.
1310 Note that IPython may not work properly; it must at least be used
1311 with the `-cl' flag, i.e. use `ipython -cl'."
1312 :group 'python
1313 :type 'string)
1314
1315 (defcustom python-jython-command "jython"
1316 "Shell command to run Jython interpreter.
1317 Any arguments can't contain whitespace."
1318 :group 'python
1319 :type 'string)
1320
1321 (defvar python-command python-python-command
1322 "Actual command used to run Python.
1323 May be `python-python-command' or `python-jython-command', possibly
1324 modified by the user. Additional arguments are added when the command
1325 is used by `run-python' et al.")
1326
1327 (defvar python-buffer nil
1328 "*The current Python process buffer.
1329
1330 Commands that send text from source buffers to Python processes have
1331 to choose a process to send to. This is determined by buffer-local
1332 value of `python-buffer'. If its value in the current buffer,
1333 i.e. both any local value and the default one, is nil, `run-python'
1334 and commands that send to the Python process will start a new process.
1335
1336 Whenever \\[run-python] starts a new process, it resets the default
1337 value of `python-buffer' to be the new process's buffer and sets the
1338 buffer-local value similarly if the current buffer is in Python mode
1339 or Inferior Python mode, so that source buffer stays associated with a
1340 specific sub-process.
1341
1342 Use \\[python-set-proc] to set the default value from a buffer with a
1343 local value.")
1344 (make-variable-buffer-local 'python-buffer)
1345
1346 (defconst python-compilation-regexp-alist
1347 ;; FIXME: maybe these should move to compilation-error-regexp-alist-alist.
1348 ;; The first already is (for CAML), but the second isn't. Anyhow,
1349 ;; these are specific to the inferior buffer. -- fx
1350 `((,(rx line-start (1+ (any " \t")) "File \""
1351 (group (1+ (not (any "\"<")))) ; avoid `<stdin>' &c
1352 "\", line " (group (1+ digit)))
1353 1 2)
1354 (,(rx " in file " (group (1+ not-newline)) " on line "
1355 (group (1+ digit)))
1356 1 2)
1357 ;; pdb stack trace
1358 (,(rx line-start "> " (group (1+ (not (any "(\"<"))))
1359 "(" (group (1+ digit)) ")" (1+ (not (any "("))) "()")
1360 1 2))
1361 "`compilation-error-regexp-alist' for inferior Python.")
1362
1363 (defvar inferior-python-mode-map
1364 (let ((map (make-sparse-keymap)))
1365 ;; This will inherit from comint-mode-map.
1366 (define-key map "\C-c\C-l" 'python-load-file)
1367 (define-key map "\C-c\C-v" 'python-check)
1368 ;; Note that we _can_ still use these commands which send to the
1369 ;; Python process even at the prompt iff we have a normal prompt,
1370 ;; i.e. '>>> ' and not '... '. See the comment before
1371 ;; python-send-region. Fixme: uncomment these if we address that.
1372
1373 ;; (define-key map [(meta ?\t)] 'python-complete-symbol)
1374 ;; (define-key map "\C-c\C-f" 'python-describe-symbol)
1375 map))
1376
1377 (defvar inferior-python-mode-syntax-table
1378 (let ((st (make-syntax-table python-mode-syntax-table)))
1379 ;; Don't get confused by apostrophes in the process's output (e.g. if
1380 ;; you execute "help(os)").
1381 (modify-syntax-entry ?\' "." st)
1382 ;; Maybe we should do the same for double quotes?
1383 ;; (modify-syntax-entry ?\" "." st)
1384 st))
1385
1386 ;; Autoloaded.
1387 (declare-function compilation-shell-minor-mode "compile" (&optional arg))
1388
1389 ;; Fixme: This should inherit some stuff from `python-mode', but I'm
1390 ;; not sure how much: at least some keybindings, like C-c C-f;
1391 ;; syntax?; font-locking, e.g. for triple-quoted strings?
1392 (define-derived-mode inferior-python-mode comint-mode "Inferior Python"
1393 "Major mode for interacting with an inferior Python process.
1394 A Python process can be started with \\[run-python].
1395
1396 Hooks `comint-mode-hook' and `inferior-python-mode-hook' are run in
1397 that order.
1398
1399 You can send text to the inferior Python process from other buffers
1400 containing Python source.
1401 * \\[python-switch-to-python] switches the current buffer to the Python
1402 process buffer.
1403 * \\[python-send-region] sends the current region to the Python process.
1404 * \\[python-send-region-and-go] switches to the Python process buffer
1405 after sending the text.
1406 For running multiple processes in multiple buffers, see `run-python' and
1407 `python-buffer'.
1408
1409 \\{inferior-python-mode-map}"
1410 :group 'python
1411 (setq mode-line-process '(":%s"))
1412 (set (make-local-variable 'comint-input-filter) 'python-input-filter)
1413 (add-hook 'comint-preoutput-filter-functions #'python-preoutput-filter
1414 nil t)
1415 ;; Still required by `comint-redirect-send-command', for instance
1416 ;; (and we need to match things like `>>> ... >>> '):
1417 (set (make-local-variable 'comint-prompt-regexp)
1418 (rx line-start (1+ (and (or (repeat 3 (any ">.")) "(Pdb)") " "))))
1419 (set (make-local-variable 'compilation-error-regexp-alist)
1420 python-compilation-regexp-alist)
1421 (compilation-shell-minor-mode 1))
1422
1423 (defcustom inferior-python-filter-regexp "\\`\\s-*\\S-?\\S-?\\s-*\\'"
1424 "Input matching this regexp is not saved on the history list.
1425 Default ignores all inputs of 0, 1, or 2 non-blank characters."
1426 :type 'regexp
1427 :group 'python)
1428
1429 (defun python-input-filter (str)
1430 "`comint-input-filter' function for inferior Python.
1431 Don't save anything for STR matching `inferior-python-filter-regexp'."
1432 (not (string-match inferior-python-filter-regexp str)))
1433
1434 ;; Fixme: Loses with quoted whitespace.
1435 (defun python-args-to-list (string)
1436 (let ((where (string-match "[ \t]" string)))
1437 (cond ((null where) (list string))
1438 ((not (= where 0))
1439 (cons (substring string 0 where)
1440 (python-args-to-list (substring string (+ 1 where)))))
1441 (t (let ((pos (string-match "[^ \t]" string)))
1442 (if pos (python-args-to-list (substring string pos))))))))
1443
1444 (defvar python-preoutput-result nil
1445 "Data from last `_emacs_out' line seen by the preoutput filter.")
1446
1447 (defvar python-preoutput-continuation nil
1448 "If non-nil, funcall this when `python-preoutput-filter' sees `_emacs_ok'.")
1449
1450 (defvar python-preoutput-leftover nil)
1451 (defvar python-preoutput-skip-next-prompt nil)
1452
1453 ;; Using this stops us getting lines in the buffer like
1454 ;; >>> ... ... >>>
1455 ;; Also look for (and delete) an `_emacs_ok' string and call
1456 ;; `python-preoutput-continuation' if we get it.
1457 (defun python-preoutput-filter (s)
1458 "`comint-preoutput-filter-functions' function: ignore prompts not at bol."
1459 (when python-preoutput-leftover
1460 (setq s (concat python-preoutput-leftover s))
1461 (setq python-preoutput-leftover nil))
1462 (let ((start 0)
1463 (res ""))
1464 ;; First process whole lines.
1465 (while (string-match "\n" s start)
1466 (let ((line (substring s start (setq start (match-end 0)))))
1467 ;; Skip prompt if needed.
1468 (when (and python-preoutput-skip-next-prompt
1469 (string-match comint-prompt-regexp line))
1470 (setq python-preoutput-skip-next-prompt nil)
1471 (setq line (substring line (match-end 0))))
1472 ;; Recognize special _emacs_out lines.
1473 (if (and (string-match "\\`_emacs_out \\(.*\\)\n\\'" line)
1474 (local-variable-p 'python-preoutput-result))
1475 (progn
1476 (setq python-preoutput-result (match-string 1 line))
1477 (set (make-local-variable 'python-preoutput-skip-next-prompt) t))
1478 (setq res (concat res line)))))
1479 ;; Then process the remaining partial line.
1480 (unless (zerop start) (setq s (substring s start)))
1481 (cond ((and (string-match comint-prompt-regexp s)
1482 ;; Drop this prompt if it follows an _emacs_out...
1483 (or python-preoutput-skip-next-prompt
1484 ;; ... or if it's not gonna be inserted at BOL.
1485 ;; Maybe we could be more selective here.
1486 (if (zerop (length res))
1487 (not (bolp))
1488 (string-match ".\\'" res))))
1489 ;; The need for this seems to be system-dependent:
1490 ;; What is this all about, exactly? --Stef
1491 ;; (if (and (eq ?. (aref s 0)))
1492 ;; (accept-process-output (get-buffer-process (current-buffer)) 1))
1493 (setq python-preoutput-skip-next-prompt nil)
1494 res)
1495 ((let ((end (min (length "_emacs_out ") (length s))))
1496 (eq t (compare-strings s nil end "_emacs_out " nil end)))
1497 ;; The leftover string is a prefix of _emacs_out so we don't know
1498 ;; yet whether it's an _emacs_out or something else: wait until we
1499 ;; get more output so we can resolve this ambiguity.
1500 (set (make-local-variable 'python-preoutput-leftover) s)
1501 res)
1502 (t (concat res s)))))
1503
1504 (autoload 'comint-check-proc "comint")
1505
1506 (defvar python-version-checked nil)
1507 (defun python-check-version (cmd)
1508 "Check that CMD runs a suitable version of Python."
1509 ;; Fixme: Check on Jython.
1510 (unless (or python-version-checked
1511 (equal 0 (string-match (regexp-quote python-python-command)
1512 cmd)))
1513 (unless (shell-command-to-string cmd)
1514 (error "Can't run Python command `%s'" cmd))
1515 (let* ((res (shell-command-to-string (concat cmd " --version"))))
1516 (string-match "Python \\([0-9]\\)\\.\\([0-9]\\)" res)
1517 (unless (and (equal "2" (match-string 1 res))
1518 (match-beginning 2)
1519 (>= (string-to-number (match-string 2 res)) 2))
1520 (error "Only Python versions >= 2.2 and < 3.0 supported")))
1521 (setq python-version-checked t)))
1522
1523 ;;;###autoload
1524 (defun run-python (&optional cmd noshow new)
1525 "Run an inferior Python process, input and output via buffer *Python*.
1526 CMD is the Python command to run. NOSHOW non-nil means don't show the
1527 buffer automatically.
1528
1529 Normally, if there is a process already running in `python-buffer',
1530 switch to that buffer. Interactively, a prefix arg allows you to edit
1531 the initial command line (default is `python-command'); `-i' etc. args
1532 will be added to this as appropriate. A new process is started if:
1533 one isn't running attached to `python-buffer', or interactively the
1534 default `python-command', or argument NEW is non-nil. See also the
1535 documentation for `python-buffer'.
1536
1537 Runs the hook `inferior-python-mode-hook' \(after the
1538 `comint-mode-hook' is run). \(Type \\[describe-mode] in the process
1539 buffer for a list of commands.)"
1540 (interactive (if current-prefix-arg
1541 (list (read-string "Run Python: " python-command) nil t)
1542 (list python-command)))
1543 (unless cmd (setq cmd python-command))
1544 (python-check-version cmd)
1545 (setq python-command cmd)
1546 ;; Fixme: Consider making `python-buffer' buffer-local as a buffer
1547 ;; (not a name) in Python buffers from which `run-python' &c is
1548 ;; invoked. Would support multiple processes better.
1549 (when (or new (not (comint-check-proc python-buffer)))
1550 (with-current-buffer
1551 (let* ((cmdlist (append (python-args-to-list cmd) '("-i")))
1552 (path (getenv "PYTHONPATH"))
1553 (process-environment ; to import emacs.py
1554 (cons (concat "PYTHONPATH="
1555 (if path (concat path path-separator))
1556 data-directory)
1557 process-environment))
1558 ;; Suppress use of pager for help output:
1559 (process-connection-type nil))
1560 (apply 'make-comint-in-buffer "Python"
1561 (generate-new-buffer "*Python*")
1562 (car cmdlist) nil (cdr cmdlist)))
1563 (setq-default python-buffer (current-buffer))
1564 (setq python-buffer (current-buffer))
1565 (accept-process-output (get-buffer-process python-buffer) 5)
1566 (inferior-python-mode)
1567 ;; Load function definitions we need.
1568 ;; Before the preoutput function was used, this was done via -c in
1569 ;; cmdlist, but that loses the banner and doesn't run the startup
1570 ;; file. The code might be inline here, but there's enough that it
1571 ;; seems worth putting in a separate file, and it's probably cleaner
1572 ;; to put it in a module.
1573 ;; Ensure we're at a prompt before doing anything else.
1574 (python-send-string "import emacs")
1575 ;; The following line was meant to ensure that we're at a prompt
1576 ;; before doing anything else. However, this can cause Emacs to
1577 ;; hang waiting for a response, if that Python function fails
1578 ;; (i.e. raises an exception).
1579 ;; (python-send-receive "print '_emacs_out ()'")
1580 ))
1581 (if (derived-mode-p 'python-mode)
1582 (setq python-buffer (default-value 'python-buffer))) ; buffer-local
1583 ;; Without this, help output goes into the inferior python buffer if
1584 ;; the process isn't already running.
1585 (sit-for 1 t) ;Should we use accept-process-output instead? --Stef
1586 (unless noshow (pop-to-buffer python-buffer t)))
1587
1588 (defun python-send-command (command)
1589 "Like `python-send-string' but resets `compilation-shell-minor-mode'."
1590 (when (python-check-comint-prompt)
1591 (with-current-buffer (process-buffer (python-proc))
1592 (goto-char (point-max))
1593 (compilation-forget-errors)
1594 (python-send-string command)
1595 (setq compilation-last-buffer (current-buffer)))))
1596
1597 (defun python-send-region (start end)
1598 "Send the region to the inferior Python process."
1599 ;; The region is evaluated from a temporary file. This avoids
1600 ;; problems with blank lines, which have different semantics
1601 ;; interactively and in files. It also saves the inferior process
1602 ;; buffer filling up with interpreter prompts. We need a Python
1603 ;; function to remove the temporary file when it has been evaluated
1604 ;; (though we could probably do it in Lisp with a Comint output
1605 ;; filter). This function also catches exceptions and truncates
1606 ;; tracebacks not to mention the frame of the function itself.
1607 ;;
1608 ;; The `compilation-shell-minor-mode' parsing takes care of relating
1609 ;; the reference to the temporary file to the source.
1610 ;;
1611 ;; Fixme: Write a `coding' header to the temp file if the region is
1612 ;; non-ASCII.
1613 (interactive "r")
1614 (let* ((f (make-temp-file "py"))
1615 (command (format "emacs.eexecfile(%S)" f))
1616 (orig-start (copy-marker start)))
1617 (when (save-excursion
1618 (goto-char start)
1619 (/= 0 (current-indentation))) ; need dummy block
1620 (save-excursion
1621 (goto-char orig-start)
1622 ;; Wrong if we had indented code at buffer start.
1623 (set-marker orig-start (line-beginning-position 0)))
1624 (write-region "if True:\n" nil f nil 'nomsg))
1625 (write-region start end f t 'nomsg)
1626 (python-send-command command)
1627 (with-current-buffer (process-buffer (python-proc))
1628 ;; Tell compile.el to redirect error locations in file `f' to
1629 ;; positions past marker `orig-start'. It has to be done *after*
1630 ;; `python-send-command''s call to `compilation-forget-errors'.
1631 (compilation-fake-loc orig-start f))))
1632
1633 (defun python-send-string (string)
1634 "Evaluate STRING in inferior Python process."
1635 (interactive "sPython command: ")
1636 (comint-send-string (python-proc) string)
1637 (unless (string-match "\n\\'" string)
1638 ;; Make sure the text is properly LF-terminated.
1639 (comint-send-string (python-proc) "\n"))
1640 (when (string-match "\n[ \t].*\n?\\'" string)
1641 ;; If the string contains a final indented line, add a second newline so
1642 ;; as to make sure we terminate the multiline instruction.
1643 (comint-send-string (python-proc) "\n")))
1644
1645 (defun python-send-buffer ()
1646 "Send the current buffer to the inferior Python process."
1647 (interactive)
1648 (python-send-region (point-min) (point-max)))
1649
1650 ;; Fixme: Try to define the function or class within the relevant
1651 ;; module, not just at top level.
1652 (defun python-send-defun ()
1653 "Send the current defun (class or method) to the inferior Python process."
1654 (interactive)
1655 (save-excursion (python-send-region (progn (beginning-of-defun) (point))
1656 (progn (end-of-defun) (point)))))
1657
1658 (defun python-switch-to-python (eob-p)
1659 "Switch to the Python process buffer, maybe starting new process.
1660 With prefix arg, position cursor at end of buffer."
1661 (interactive "P")
1662 (pop-to-buffer (process-buffer (python-proc)) t) ;Runs python if needed.
1663 (when eob-p
1664 (push-mark)
1665 (goto-char (point-max))))
1666
1667 (defun python-send-region-and-go (start end)
1668 "Send the region to the inferior Python process.
1669 Then switch to the process buffer."
1670 (interactive "r")
1671 (python-send-region start end)
1672 (python-switch-to-python t))
1673
1674 (defcustom python-source-modes '(python-mode jython-mode)
1675 "Used to determine if a buffer contains Python source code.
1676 If a file is loaded into a buffer that is in one of these major modes,
1677 it is considered Python source by `python-load-file', which uses the
1678 value to determine defaults."
1679 :type '(repeat function)
1680 :group 'python)
1681
1682 (defvar python-prev-dir/file nil
1683 "Caches (directory . file) pair used in the last `python-load-file' command.
1684 Used for determining the default in the next one.")
1685
1686 (autoload 'comint-get-source "comint")
1687
1688 (defun python-load-file (file-name)
1689 "Load a Python file FILE-NAME into the inferior Python process.
1690 If the file has extension `.py' import or reload it as a module.
1691 Treating it as a module keeps the global namespace clean, provides
1692 function location information for debugging, and supports users of
1693 module-qualified names."
1694 (interactive (comint-get-source "Load Python file: " python-prev-dir/file
1695 python-source-modes
1696 t)) ; because execfile needs exact name
1697 (comint-check-source file-name) ; Check to see if buffer needs saving.
1698 (setq python-prev-dir/file (cons (file-name-directory file-name)
1699 (file-name-nondirectory file-name)))
1700 (with-current-buffer (process-buffer (python-proc)) ;Runs python if needed.
1701 ;; Fixme: I'm not convinced by this logic from python-mode.el.
1702 (python-send-command
1703 (if (string-match "\\.py\\'" file-name)
1704 (let ((module (file-name-sans-extension
1705 (file-name-nondirectory file-name))))
1706 (format "emacs.eimport(%S,%S)"
1707 module (file-name-directory file-name)))
1708 (format "execfile(%S)" file-name)))
1709 (message "%s loaded" file-name)))
1710
1711 (defun python-proc ()
1712 "Return the current Python process.
1713 See variable `python-buffer'. Starts a new process if necessary."
1714 ;; Fixme: Maybe should look for another active process if there
1715 ;; isn't one for `python-buffer'.
1716 (unless (comint-check-proc python-buffer)
1717 (run-python nil t))
1718 (get-buffer-process (if (derived-mode-p 'inferior-python-mode)
1719 (current-buffer)
1720 python-buffer)))
1721
1722 (defun python-set-proc ()
1723 "Set the default value of `python-buffer' to correspond to this buffer.
1724 If the current buffer has a local value of `python-buffer', set the
1725 default (global) value to that. The associated Python process is
1726 the one that gets input from \\[python-send-region] et al when used
1727 in a buffer that doesn't have a local value of `python-buffer'."
1728 (interactive)
1729 (if (local-variable-p 'python-buffer)
1730 (setq-default python-buffer python-buffer)
1731 (error "No local value of `python-buffer'")))
1732 \f
1733 ;;;; Context-sensitive help.
1734
1735 (defconst python-dotty-syntax-table
1736 (let ((table (make-syntax-table)))
1737 (set-char-table-parent table python-mode-syntax-table)
1738 (modify-syntax-entry ?. "_" table)
1739 table)
1740 "Syntax table giving `.' symbol syntax.
1741 Otherwise inherits from `python-mode-syntax-table'.")
1742
1743 (defvar view-return-to-alist)
1744 (eval-when-compile (autoload 'help-buffer "help-fns"))
1745
1746 (defvar python-imports) ; forward declaration
1747
1748 ;; Fixme: Should this actually be used instead of info-look, i.e. be
1749 ;; bound to C-h S? [Probably not, since info-look may work in cases
1750 ;; where this doesn't.]
1751 (defun python-describe-symbol (symbol)
1752 "Get help on SYMBOL using `help'.
1753 Interactively, prompt for symbol.
1754
1755 Symbol may be anything recognized by the interpreter's `help'
1756 command -- e.g. `CALLS' -- not just variables in scope in the
1757 interpreter. This only works for Python version 2.2 or newer
1758 since earlier interpreters don't support `help'.
1759
1760 In some cases where this doesn't find documentation, \\[info-lookup-symbol]
1761 will."
1762 ;; Note that we do this in the inferior process, not a separate one, to
1763 ;; ensure the environment is appropriate.
1764 (interactive
1765 (let ((symbol (with-syntax-table python-dotty-syntax-table
1766 (current-word)))
1767 (enable-recursive-minibuffers t))
1768 (list (read-string (if symbol
1769 (format "Describe symbol (default %s): " symbol)
1770 "Describe symbol: ")
1771 nil nil symbol))))
1772 (if (equal symbol "") (error "No symbol"))
1773 ;; Ensure we have a suitable help buffer.
1774 ;; Fixme: Maybe process `Related help topics' a la help xrefs and
1775 ;; allow C-c C-f in help buffer.
1776 (let ((temp-buffer-show-hook ; avoid xref stuff
1777 (lambda ()
1778 (toggle-read-only 1)
1779 (setq view-return-to-alist
1780 (list (cons (selected-window) help-return-method))))))
1781 (with-output-to-temp-buffer (help-buffer)
1782 (with-current-buffer standard-output
1783 ;; Fixme: Is this actually useful?
1784 (help-setup-xref (list 'python-describe-symbol symbol) (interactive-p))
1785 (set (make-local-variable 'comint-redirect-subvert-readonly) t)
1786 (print-help-return-message))))
1787 (comint-redirect-send-command-to-process (format "emacs.ehelp(%S, %s)"
1788 symbol python-imports)
1789 "*Help*" (python-proc) nil nil))
1790
1791 (add-to-list 'debug-ignored-errors "^No symbol")
1792
1793 (defun python-send-receive (string)
1794 "Send STRING to inferior Python (if any) and return result.
1795 The result is what follows `_emacs_out' in the output.
1796 This is a no-op if `python-check-comint-prompt' returns nil."
1797 (python-send-string string)
1798 (let ((proc (python-proc)))
1799 (with-current-buffer (process-buffer proc)
1800 (when (python-check-comint-prompt proc)
1801 (set (make-local-variable 'python-preoutput-result) nil)
1802 (while (progn
1803 (accept-process-output proc 5)
1804 (null python-preoutput-result)))
1805 (prog1 python-preoutput-result
1806 (kill-local-variable 'python-preoutput-result))))))
1807
1808 (defun python-check-comint-prompt (&optional proc)
1809 "Return non-nil iff there's a normal prompt in the inferior buffer.
1810 If there isn't, it's probably not appropriate to send input to return
1811 Eldoc information etc. If PROC is non-nil, check the buffer for that
1812 process."
1813 (with-current-buffer (process-buffer (or proc (python-proc)))
1814 (save-excursion
1815 (save-match-data (re-search-backward ">>> \\=" nil t)))))
1816
1817 ;; Fixme: Is there anything reasonable we can do with random methods?
1818 ;; (Currently only works with functions.)
1819 (defun python-eldoc-function ()
1820 "`eldoc-documentation-function' for Python.
1821 Only works when point is in a function name, not its arg list, for
1822 instance. Assumes an inferior Python is running."
1823 (let ((symbol (with-syntax-table python-dotty-syntax-table
1824 (current-word))))
1825 ;; This is run from timers, so inhibit-quit tends to be set.
1826 (with-local-quit
1827 ;; First try the symbol we're on.
1828 (or (and symbol
1829 (python-send-receive (format "emacs.eargs(%S, %s)"
1830 symbol python-imports)))
1831 ;; Try moving to symbol before enclosing parens.
1832 (let ((s (syntax-ppss)))
1833 (unless (zerop (car s))
1834 (when (eq ?\( (char-after (nth 1 s)))
1835 (save-excursion
1836 (goto-char (nth 1 s))
1837 (skip-syntax-backward "-")
1838 (let ((point (point)))
1839 (skip-chars-backward "a-zA-Z._")
1840 (if (< (point) point)
1841 (python-send-receive
1842 (format "emacs.eargs(%S, %s)"
1843 (buffer-substring-no-properties (point) point)
1844 python-imports))))))))))))
1845 \f
1846 ;;;; Info-look functionality.
1847
1848 (declare-function info-lookup-maybe-add-help "info-look" (&rest arg))
1849
1850 (defun python-after-info-look ()
1851 "Set up info-look for Python.
1852 Used with `eval-after-load'."
1853 (let* ((version (let ((s (shell-command-to-string (concat python-command
1854 " -V"))))
1855 (string-match "^Python \\([0-9]+\\.[0-9]+\\>\\)" s)
1856 (match-string 1 s)))
1857 ;; Whether info files have a Python version suffix, e.g. in Debian.
1858 (versioned
1859 (with-temp-buffer
1860 (with-no-warnings (Info-mode))
1861 (condition-case ()
1862 ;; Don't use `info' because it would pop-up a *info* buffer.
1863 (with-no-warnings
1864 (Info-goto-node (format "(python%s-lib)Miscellaneous Index"
1865 version))
1866 t)
1867 (error nil)))))
1868 (info-lookup-maybe-add-help
1869 :mode 'python-mode
1870 :regexp "[[:alnum:]_]+"
1871 :doc-spec
1872 ;; Fixme: Can this reasonably be made specific to indices with
1873 ;; different rules? Is the order of indices optimal?
1874 ;; (Miscellaneous in -ref first prefers lookup of keywords, for
1875 ;; instance.)
1876 (if versioned
1877 ;; The empty prefix just gets us highlighted terms.
1878 `((,(concat "(python" version "-ref)Miscellaneous Index") nil "")
1879 (,(concat "(python" version "-ref)Module Index" nil ""))
1880 (,(concat "(python" version "-ref)Function-Method-Variable Index"
1881 nil ""))
1882 (,(concat "(python" version "-ref)Class-Exception-Object Index"
1883 nil ""))
1884 (,(concat "(python" version "-lib)Module Index" nil ""))
1885 (,(concat "(python" version "-lib)Class-Exception-Object Index"
1886 nil ""))
1887 (,(concat "(python" version "-lib)Function-Method-Variable Index"
1888 nil ""))
1889 (,(concat "(python" version "-lib)Miscellaneous Index" nil "")))
1890 '(("(python-ref)Miscellaneous Index" nil "")
1891 ("(python-ref)Module Index" nil "")
1892 ("(python-ref)Function-Method-Variable Index" nil "")
1893 ("(python-ref)Class-Exception-Object Index" nil "")
1894 ("(python-lib)Module Index" nil "")
1895 ("(python-lib)Class-Exception-Object Index" nil "")
1896 ("(python-lib)Function-Method-Variable Index" nil "")
1897 ("(python-lib)Miscellaneous Index" nil ""))))))
1898 (eval-after-load "info-look" '(python-after-info-look))
1899 \f
1900 ;;;; Miscellany.
1901
1902 (defcustom python-jython-packages '("java" "javax" "org" "com")
1903 "Packages implying `jython-mode'.
1904 If these are imported near the beginning of the buffer, `python-mode'
1905 actually punts to `jython-mode'."
1906 :type '(repeat string)
1907 :group 'python)
1908
1909 ;; Called from `python-mode', this causes a recursive call of the
1910 ;; mode. See logic there to break out of the recursion.
1911 (defun python-maybe-jython ()
1912 "Invoke `jython-mode' if the buffer appears to contain Jython code.
1913 The criterion is either a match for `jython-mode' via
1914 `interpreter-mode-alist' or an import of a module from the list
1915 `python-jython-packages'."
1916 ;; The logic is taken from python-mode.el.
1917 (save-excursion
1918 (save-restriction
1919 (widen)
1920 (goto-char (point-min))
1921 (let ((interpreter (if (looking-at auto-mode-interpreter-regexp)
1922 (match-string 2))))
1923 (if (and interpreter (eq 'jython-mode
1924 (cdr (assoc (file-name-nondirectory
1925 interpreter)
1926 interpreter-mode-alist))))
1927 (jython-mode)
1928 (if (catch 'done
1929 (while (re-search-forward
1930 (rx line-start (or "import" "from") (1+ space)
1931 (group (1+ (not (any " \t\n.")))))
1932 (+ (point-min) 10000) ; Probably not worth customizing.
1933 t)
1934 (if (member (match-string 1) python-jython-packages)
1935 (throw 'done t))))
1936 (jython-mode)))))))
1937
1938 (defun python-fill-paragraph (&optional justify)
1939 "`fill-paragraph-function' handling multi-line strings and possibly comments.
1940 If any of the current line is in or at the end of a multi-line string,
1941 fill the string or the paragraph of it that point is in, preserving
1942 the string's indentation."
1943 (interactive "P")
1944 (or (fill-comment-paragraph justify)
1945 (save-excursion
1946 (end-of-line)
1947 (let* ((syntax (syntax-ppss))
1948 (orig (point))
1949 start end)
1950 (cond ((nth 4 syntax) ; comment. fixme: loses with trailing one
1951 (let (fill-paragraph-function)
1952 (fill-paragraph justify)))
1953 ;; The `paragraph-start' and `paragraph-separate'
1954 ;; variables don't allow us to delimit the last
1955 ;; paragraph in a multi-line string properly, so narrow
1956 ;; to the string and then fill around (the end of) the
1957 ;; current line.
1958 ((eq t (nth 3 syntax)) ; in fenced string
1959 (goto-char (nth 8 syntax)) ; string start
1960 (setq start (line-beginning-position))
1961 (setq end (condition-case () ; for unbalanced quotes
1962 (progn (forward-sexp)
1963 (- (point) 3))
1964 (error (point-max)))))
1965 ((re-search-backward "\\s|\\s-*\\=" nil t) ; end of fenced string
1966 (forward-char)
1967 (setq end (point))
1968 (condition-case ()
1969 (progn (backward-sexp)
1970 (setq start (line-beginning-position)))
1971 (error nil))))
1972 (when end
1973 (save-restriction
1974 (narrow-to-region start end)
1975 (goto-char orig)
1976 ;; Avoid losing leading and trailing newlines in doc
1977 ;; strings written like:
1978 ;; """
1979 ;; ...
1980 ;; """
1981 (let ((paragraph-separate
1982 ;; Note that the string could be part of an
1983 ;; expression, so it can have preceding and
1984 ;; trailing non-whitespace.
1985 (concat
1986 (rx (or
1987 ;; Opening triple quote without following text.
1988 (and (* nonl)
1989 (group (syntax string-delimiter))
1990 (repeat 2 (backref 1))
1991 ;; Fixme: Not sure about including
1992 ;; trailing whitespace.
1993 (* (any " \t"))
1994 eol)
1995 ;; Closing trailing quote without preceding text.
1996 (and (group (any ?\" ?')) (backref 2)
1997 (syntax string-delimiter))))
1998 "\\(?:" paragraph-separate "\\)"))
1999 fill-paragraph-function)
2000 (fill-paragraph justify))))))) t)
2001
2002 (defun python-shift-left (start end &optional count)
2003 "Shift lines in region COUNT (the prefix arg) columns to the left.
2004 COUNT defaults to `python-indent'. If region isn't active, just shift
2005 current line. The region shifted includes the lines in which START and
2006 END lie. It is an error if any lines in the region are indented less than
2007 COUNT columns."
2008 (interactive (if mark-active
2009 (list (region-beginning) (region-end) current-prefix-arg)
2010 (list (point) (point) current-prefix-arg)))
2011 (if count
2012 (setq count (prefix-numeric-value count))
2013 (setq count python-indent))
2014 (when (> count 0)
2015 (save-excursion
2016 (goto-char start)
2017 (while (< (point) end)
2018 (if (and (< (current-indentation) count)
2019 (not (looking-at "[ \t]*$")))
2020 (error "Can't shift all lines enough"))
2021 (forward-line))
2022 (indent-rigidly start end (- count)))))
2023
2024 (add-to-list 'debug-ignored-errors "^Can't shift all lines enough")
2025
2026 (defun python-shift-right (start end &optional count)
2027 "Shift lines in region COUNT (the prefix arg) columns to the right.
2028 COUNT defaults to `python-indent'. If region isn't active, just shift
2029 current line. The region shifted includes the lines in which START and
2030 END lie."
2031 (interactive (if mark-active
2032 (list (region-beginning) (region-end) current-prefix-arg)
2033 (list (point) (point) current-prefix-arg)))
2034 (if count
2035 (setq count (prefix-numeric-value count))
2036 (setq count python-indent))
2037 (indent-rigidly start end count))
2038
2039 (defun python-outline-level ()
2040 "`outline-level' function for Python mode.
2041 The level is the number of `python-indent' steps of indentation
2042 of current line."
2043 (1+ (/ (current-indentation) python-indent)))
2044
2045 ;; Fixme: Consider top-level assignments, imports, &c.
2046 (defun python-current-defun (&optional length-limit)
2047 "`add-log-current-defun-function' for Python."
2048 (save-excursion
2049 ;; Move up the tree of nested `class' and `def' blocks until we
2050 ;; get to zero indentation, accumulating the defined names.
2051 (let ((accum)
2052 (length -1))
2053 (catch 'done
2054 (while (or (null length-limit)
2055 (null (cdr accum))
2056 (< length length-limit))
2057 (let ((started-from (point)))
2058 (python-beginning-of-block)
2059 (end-of-line)
2060 (beginning-of-defun)
2061 (when (= (point) started-from)
2062 (throw 'done nil)))
2063 (when (looking-at (rx (0+ space) (or "def" "class") (1+ space)
2064 (group (1+ (or word (syntax symbol))))))
2065 (push (match-string 1) accum)
2066 (setq length (+ length 1 (length (car accum)))))
2067 (when (= (current-indentation) 0)
2068 (throw 'done nil))))
2069 (when accum
2070 (when (and length-limit (> length length-limit))
2071 (setcar accum ".."))
2072 (mapconcat 'identity accum ".")))))
2073
2074 (defun python-mark-block ()
2075 "Mark the block around point.
2076 Uses `python-beginning-of-block', `python-end-of-block'."
2077 (interactive)
2078 (push-mark)
2079 (python-beginning-of-block)
2080 (push-mark (point) nil t)
2081 (python-end-of-block)
2082 (exchange-point-and-mark))
2083
2084 ;; Fixme: Provide a find-function-like command to find source of a
2085 ;; definition (separate from BicycleRepairMan). Complicated by
2086 ;; finding the right qualified name.
2087 \f
2088 ;;;; Completion.
2089
2090 ;; http://lists.gnu.org/archive/html/bug-gnu-emacs/2008-01/msg00076.html
2091 (defvar python-imports "None"
2092 "String of top-level import statements updated by `python-find-imports'.")
2093 (make-variable-buffer-local 'python-imports)
2094
2095 ;; Fixme: Should font-lock try to run this when it deals with an import?
2096 ;; Maybe not a good idea if it gets run multiple times when the
2097 ;; statement is being edited, and is more likely to end up with
2098 ;; something syntactically incorrect.
2099 ;; However, what we should do is to trundle up the block tree from point
2100 ;; to extract imports that appear to be in scope, and add those.
2101 (defun python-find-imports ()
2102 "Find top-level imports, updating `python-imports'."
2103 (interactive)
2104 (save-excursion
2105 (let (lines)
2106 (goto-char (point-min))
2107 (while (re-search-forward "^import\\>\\|^from\\>" nil t)
2108 (unless (syntax-ppss-context (syntax-ppss))
2109 (let ((start (line-beginning-position)))
2110 ;; Skip over continued lines.
2111 (while (and (eq ?\\ (char-before (line-end-position)))
2112 (= 0 (forward-line 1)))
2113 t)
2114 (push (buffer-substring start (line-beginning-position 2))
2115 lines))))
2116 (setq python-imports
2117 (if lines
2118 (apply #'concat
2119 ;; This is probably best left out since you're unlikely to need the
2120 ;; doc for a function in the buffer and the import will lose if the
2121 ;; Python sub-process' working directory isn't the same as the
2122 ;; buffer's.
2123 ;; (if buffer-file-name
2124 ;; (concat
2125 ;; "import "
2126 ;; (file-name-sans-extension
2127 ;; (file-name-nondirectory buffer-file-name))))
2128 (nreverse lines))
2129 "None"))
2130 (when lines
2131 (set-text-properties 0 (length python-imports) nil python-imports)
2132 ;; The output ends up in the wrong place if the string we
2133 ;; send contains newlines (from the imports).
2134 (setq python-imports
2135 (replace-regexp-in-string "\n" "\\n"
2136 (format "%S" python-imports) t t))))))
2137
2138 ;; Fixme: This fails the first time if the sub-process isn't already
2139 ;; running. Presumably a timing issue with i/o to the process.
2140 (defun python-symbol-completions (symbol)
2141 "Return a list of completions of the string SYMBOL from Python process.
2142 The list is sorted.
2143 Uses `python-imports' to load modules against which to complete."
2144 (when symbol
2145 (let ((completions
2146 (condition-case ()
2147 (car (read-from-string
2148 (python-send-receive
2149 (format "emacs.complete(%S,%s)" symbol python-imports))))
2150 (error nil))))
2151 (sort
2152 ;; We can get duplicates from the above -- don't know why.
2153 (delete-dups completions)
2154 #'string<))))
2155
2156 (defun python-partial-symbol ()
2157 "Return the partial symbol before point (for completion)."
2158 (let ((end (point))
2159 (start (save-excursion
2160 (and (re-search-backward
2161 (rx (or buffer-start (regexp "[^[:alnum:]._]"))
2162 (group (1+ (regexp "[[:alnum:]._]"))) point)
2163 nil t)
2164 (match-beginning 1)))))
2165 (if start (buffer-substring-no-properties start end))))
2166 \f
2167 ;;;; FFAP support
2168
2169 (defun python-module-path (module)
2170 "Function for `ffap-alist' to return path to MODULE."
2171 (python-send-receive (format "emacs.modpath (%S)" module)))
2172
2173 (eval-after-load "ffap"
2174 '(push '(python-mode . python-module-path) ffap-alist))
2175 \f
2176 ;;;; Find-function support
2177
2178 ;; Fixme: key binding?
2179
2180 (defun python-find-function (name)
2181 "Find source of definition of function NAME.
2182 Interactively, prompt for name."
2183 (interactive
2184 (let ((symbol (with-syntax-table python-dotty-syntax-table
2185 (current-word)))
2186 (enable-recursive-minibuffers t))
2187 (list (read-string (if symbol
2188 (format "Find location of (default %s): " symbol)
2189 "Find location of: ")
2190 nil nil symbol))))
2191 (unless python-imports
2192 (error "Not called from buffer visiting Python file"))
2193 (let* ((loc (python-send-receive (format "emacs.location_of (%S, %s)"
2194 name python-imports)))
2195 (loc (car (read-from-string loc)))
2196 (file (car loc))
2197 (line (cdr loc)))
2198 (unless file (error "Don't know where `%s' is defined" name))
2199 (pop-to-buffer (find-file-noselect file))
2200 (when (integerp line)
2201 (goto-line line))))
2202 \f
2203 ;;;; Skeletons
2204
2205 (defcustom python-use-skeletons nil
2206 "Non-nil means template skeletons will be automagically inserted.
2207 This happens when pressing \"if<SPACE>\", for example, to prompt for
2208 the if condition."
2209 :type 'boolean
2210 :group 'python)
2211
2212 (defvar python-skeletons nil
2213 "Alist of named skeletons for Python mode.
2214 Elements are of the form (NAME . EXPANDER-FUNCTION).")
2215
2216 (define-abbrev-table 'python-mode-abbrev-table ()
2217 "Abbrev table for Python mode.
2218 The default contents correspond to the elements of `python-skeletons'."
2219 ;; Allow / in abbrevs.
2220 :regexp "\\<\\([[:word:]/]+\\)\\W*")
2221
2222 (eval-when-compile
2223 ;; Define a user-level skeleton and add it to `python-skeletons' and
2224 ;; the abbrev table.
2225 (defmacro def-python-skeleton (name &rest elements)
2226 (let* ((name (symbol-name name))
2227 (function (intern (concat "python-insert-" name))))
2228 `(progn
2229 (add-to-list 'python-skeletons ',(cons name function))
2230 ;; Usual technique for inserting a skeleton, but expand
2231 ;; to the original abbrev instead if in a comment or string.
2232 (define-abbrev python-mode-abbrev-table ,name ""
2233 ;; Quote this to give a readable abbrev table.
2234 '(lambda ()
2235 (if (python-in-string/comment)
2236 (insert ,name)
2237 (,function)))
2238 nil t) ; system abbrev
2239 (define-skeleton ,function
2240 ,(format "Insert Python \"%s\" template." name)
2241 ,@elements)))))
2242 (put 'def-python-skeleton 'lisp-indent-function 2)
2243
2244 ;; From `skeleton-further-elements' set below:
2245 ;; `<': outdent a level;
2246 ;; `^': delete indentation on current line and also previous newline.
2247 ;; Not quite like `delete-indentation'. Assumes point is at
2248 ;; beginning of indentation.
2249
2250 (def-python-skeleton if
2251 "Condition: "
2252 "if " str ":" \n
2253 > -1 ; Fixme: I don't understand the spurious space this removes.
2254 _ \n
2255 ("other condition, %s: "
2256 < ; Avoid wrong indentation after block opening.
2257 "elif " str ":" \n
2258 > _ \n nil)
2259 '(python-else) | ^)
2260
2261 (define-skeleton python-else
2262 "Auxiliary skeleton."
2263 nil
2264 (unless (eq ?y (read-char "Add `else' clause? (y for yes or RET for no) "))
2265 (signal 'quit t))
2266 < "else:" \n
2267 > _ \n)
2268
2269 (def-python-skeleton while
2270 "Condition: "
2271 "while " str ":" \n
2272 > -1 _ \n
2273 '(python-else) | ^)
2274
2275 (def-python-skeleton for
2276 "Target, %s: "
2277 "for " str " in " (skeleton-read "Expression, %s: ") ":" \n
2278 > -1 _ \n
2279 '(python-else) | ^)
2280
2281 (def-python-skeleton try/except
2282 nil
2283 "try:" \n
2284 > -1 _ \n
2285 ("Exception, %s: "
2286 < "except " str '(python-target) ":" \n
2287 > _ \n nil)
2288 < "except:" \n
2289 > _ \n
2290 '(python-else) | ^)
2291
2292 (define-skeleton python-target
2293 "Auxiliary skeleton."
2294 "Target, %s: " ", " str | -2)
2295
2296 (def-python-skeleton try/finally
2297 nil
2298 "try:" \n
2299 > -1 _ \n
2300 < "finally:" \n
2301 > _ \n)
2302
2303 (def-python-skeleton def
2304 "Name: "
2305 "def " str " (" ("Parameter, %s: " (unless (equal ?\( (char-before)) ", ")
2306 str) "):" \n
2307 "\"\"\"" - "\"\"\"" \n ; Fixme: extra space inserted -- why?).
2308 > _ \n)
2309
2310 (def-python-skeleton class
2311 "Name: "
2312 "class " str " (" ("Inheritance, %s: "
2313 (unless (equal ?\( (char-before)) ", ")
2314 str)
2315 & ")" | -2 ; close list or remove opening
2316 ":" \n
2317 "\"\"\"" - "\"\"\"" \n
2318 > _ \n)
2319
2320 (defvar python-default-template "if"
2321 "Default template to expand by `python-expand-template'.
2322 Updated on each expansion.")
2323
2324 (defun python-expand-template (name)
2325 "Expand template named NAME.
2326 Interactively, prompt for the name with completion."
2327 (interactive
2328 (list (completing-read (format "Template to expand (default %s): "
2329 python-default-template)
2330 python-skeletons nil t)))
2331 (if (equal "" name)
2332 (setq name python-default-template)
2333 (setq python-default-template name))
2334 (let ((func (cdr (assoc name python-skeletons))))
2335 (if func
2336 (funcall func)
2337 (error "Undefined template: %s" name))))
2338 \f
2339 ;;;; Bicycle Repair Man support
2340
2341 (autoload 'pymacs-load "pymacs" nil t)
2342 (autoload 'brm-init "bikemacs")
2343
2344 ;; I'm not sure how useful BRM really is, and it's certainly dangerous
2345 ;; the way it modifies files outside Emacs... Also note that the
2346 ;; current BRM loses with tabs used for indentation -- I submitted a
2347 ;; fix <URL:http://www.loveshack.ukfsn.org/emacs/bikeemacs.py.diff>.
2348 (defun python-setup-brm ()
2349 "Set up Bicycle Repair Man refactoring tool (if available).
2350
2351 Note that the `refactoring' features change files independently of
2352 Emacs and may modify and save the contents of the current buffer
2353 without confirmation."
2354 (interactive)
2355 (condition-case data
2356 (unless (fboundp 'brm-rename)
2357 (pymacs-load "bikeemacs" "brm-") ; first line of normal recipe
2358 (let ((py-mode-map (make-sparse-keymap)) ; it assumes this
2359 (features (cons 'python-mode features))) ; and requires this
2360 (brm-init) ; second line of normal recipe
2361 (remove-hook 'python-mode-hook ; undo this from `brm-init'
2362 '(lambda () (easy-menu-add brm-menu)))
2363 (easy-menu-define
2364 python-brm-menu python-mode-map
2365 "Bicycle Repair Man"
2366 '("BicycleRepairMan"
2367 :help "Interface to navigation and refactoring tool"
2368 "Queries"
2369 ["Find References" brm-find-references
2370 :help "Find references to name at point in compilation buffer"]
2371 ["Find Definition" brm-find-definition
2372 :help "Find definition of name at point"]
2373 "-"
2374 "Refactoring"
2375 ["Rename" brm-rename
2376 :help "Replace name at point with a new name everywhere"]
2377 ["Extract Method" brm-extract-method
2378 :active (and mark-active (not buffer-read-only))
2379 :help "Replace statements in region with a method"]
2380 ["Extract Local Variable" brm-extract-local-variable
2381 :active (and mark-active (not buffer-read-only))
2382 :help "Replace expression in region with an assignment"]
2383 ["Inline Local Variable" brm-inline-local-variable
2384 :help
2385 "Substitute uses of variable at point with its definition"]
2386 ;; Fixme: Should check for anything to revert.
2387 ["Undo Last Refactoring" brm-undo :help ""]))))
2388 (error (error "BicycleRepairMan setup failed: %s" data))))
2389 \f
2390 ;;;; Modes.
2391
2392 ;; pdb tracking is alert once this file is loaded, but takes no action if
2393 ;; `python-pdbtrack-do-tracking-p' is nil.
2394 (add-hook 'comint-output-filter-functions 'python-pdbtrack-track-stack-file)
2395
2396 (defvar outline-heading-end-regexp)
2397 (defvar eldoc-documentation-function)
2398 (defvar python-mode-running) ;Dynamically scoped var.
2399
2400 ;; Stuff to allow expanding abbrevs with non-word constituents.
2401 (defun python-abbrev-pc-hook ()
2402 "Reset the syntax table after possibly expanding abbrevs."
2403 (remove-hook 'post-command-hook 'python-abbrev-pc-hook t)
2404 (set-syntax-table python-mode-syntax-table))
2405
2406 (defvar python-abbrev-syntax-table
2407 (copy-syntax-table python-mode-syntax-table)
2408 "Syntax table used when expanding abbrevs.")
2409
2410 (defun python-pea-hook ()
2411 "Set the syntax table before possibly expanding abbrevs."
2412 (set-syntax-table python-abbrev-syntax-table)
2413 (add-hook 'post-command-hook 'python-abbrev-pc-hook nil t))
2414 (modify-syntax-entry ?/ "w" python-abbrev-syntax-table)
2415
2416 ;;;###autoload
2417 (define-derived-mode python-mode fundamental-mode "Python"
2418 "Major mode for editing Python files.
2419 Turns on Font Lock mode unconditionally since it is currently required
2420 for correct parsing of the source.
2421 See also `jython-mode', which is actually invoked if the buffer appears to
2422 contain Jython code. See also `run-python' and associated Python mode
2423 commands for running Python under Emacs.
2424
2425 The Emacs commands which work with `defun's, e.g. \\[beginning-of-defun], deal
2426 with nested `def' and `class' blocks. They take the innermost one as
2427 current without distinguishing method and class definitions. Used multiple
2428 times, they move over others at the same indentation level until they reach
2429 the end of definitions at that level, when they move up a level.
2430 \\<python-mode-map>
2431 Colon is electric: it outdents the line if appropriate, e.g. for
2432 an else statement. \\[python-backspace] at the beginning of an indented statement
2433 deletes a level of indentation to close the current block; otherwise it
2434 deletes a character backward. TAB indents the current line relative to
2435 the preceding code. Successive TABs, with no intervening command, cycle
2436 through the possibilities for indentation on the basis of enclosing blocks.
2437
2438 \\[fill-paragraph] fills comments and multi-line strings appropriately, but has no
2439 effect outside them.
2440
2441 Supports Eldoc mode (only for functions, using a Python process),
2442 Info-Look and Imenu. In Outline minor mode, `class' and `def'
2443 lines count as headers. Symbol completion is available in the
2444 same way as in the Python shell using the `rlcompleter' module
2445 and this is added to the Hippie Expand functions locally if
2446 Hippie Expand mode is turned on. Completion of symbols of the
2447 form x.y only works if the components are literal
2448 module/attribute names, not variables. An abbrev table is set up
2449 with skeleton expansions for compound statement templates.
2450
2451 \\{python-mode-map}"
2452 :group 'python
2453 (set (make-local-variable 'font-lock-defaults)
2454 '(python-font-lock-keywords nil nil nil nil
2455 (font-lock-syntactic-keywords
2456 . python-font-lock-syntactic-keywords)
2457 ;; This probably isn't worth it.
2458 ;; (font-lock-syntactic-face-function
2459 ;; . python-font-lock-syntactic-face-function)
2460 ))
2461 (set (make-local-variable 'parse-sexp-lookup-properties) t)
2462 (set (make-local-variable 'parse-sexp-ignore-comments) t)
2463 (set (make-local-variable 'comment-start) "# ")
2464 (set (make-local-variable 'indent-line-function) #'python-indent-line)
2465 (set (make-local-variable 'indent-region-function) #'python-indent-region)
2466 (set (make-local-variable 'paragraph-start) "\\s-*$")
2467 (set (make-local-variable 'fill-paragraph-function) 'python-fill-paragraph)
2468 (set (make-local-variable 'require-final-newline) mode-require-final-newline)
2469 (set (make-local-variable 'add-log-current-defun-function)
2470 #'python-current-defun)
2471 (set (make-local-variable 'outline-regexp)
2472 (rx (* space) (or "class" "def" "elif" "else" "except" "finally"
2473 "for" "if" "try" "while" "with")
2474 symbol-end))
2475 (set (make-local-variable 'outline-heading-end-regexp) ":\\s-*\n")
2476 (set (make-local-variable 'outline-level) #'python-outline-level)
2477 (set (make-local-variable 'open-paren-in-column-0-is-defun-start) nil)
2478 (make-local-variable 'python-saved-check-command)
2479 (set (make-local-variable 'beginning-of-defun-function)
2480 'python-beginning-of-defun)
2481 (set (make-local-variable 'end-of-defun-function) 'python-end-of-defun)
2482 (add-hook 'which-func-functions 'python-which-func nil t)
2483 (setq imenu-create-index-function #'python-imenu-create-index)
2484 (set (make-local-variable 'eldoc-documentation-function)
2485 #'python-eldoc-function)
2486 (add-hook 'eldoc-mode-hook
2487 (lambda () (run-python nil t)) ; need it running
2488 nil t)
2489 (set (make-local-variable 'symbol-completion-symbol-function)
2490 'python-partial-symbol)
2491 (set (make-local-variable 'symbol-completion-completions-function)
2492 'python-symbol-completions)
2493 ;; Fixme: should be in hideshow. This seems to be of limited use
2494 ;; since it isn't (can't be) indentation-based. Also hide-level
2495 ;; doesn't seem to work properly.
2496 (add-to-list 'hs-special-modes-alist
2497 `(python-mode "^\\s-*\\(?:def\\|class\\)\\>" nil "#"
2498 ,(lambda (arg)
2499 (python-end-of-defun)
2500 (skip-chars-backward " \t\n"))
2501 nil))
2502 (set (make-local-variable 'skeleton-further-elements)
2503 '((< '(backward-delete-char-untabify (min python-indent
2504 (current-column))))
2505 (^ '(- (1+ (current-indentation))))))
2506 (add-hook 'pre-abbrev-expand-hook 'python-pea-hook nil t)
2507 (if (featurep 'hippie-exp)
2508 (set (make-local-variable 'hippie-expand-try-functions-list)
2509 (cons 'symbol-completion-try-complete
2510 hippie-expand-try-functions-list)))
2511 ;; Python defines TABs as being 8-char wide.
2512 (set (make-local-variable 'tab-width) 8)
2513 (unless font-lock-mode (font-lock-mode 1))
2514 (when python-guess-indent (python-guess-indent))
2515 ;; Let's make it harder for the user to shoot himself in the foot.
2516 (unless (= tab-width python-indent)
2517 (setq indent-tabs-mode nil))
2518 (set (make-local-variable 'python-command) python-python-command)
2519 (python-find-imports)
2520 (unless (boundp 'python-mode-running) ; kill the recursion from jython-mode
2521 (let ((python-mode-running t))
2522 (python-maybe-jython))))
2523
2524 ;; Not done automatically in Emacs 21 or 22.
2525 (defcustom python-mode-hook nil
2526 "Hook run when entering Python mode."
2527 :group 'python
2528 :type 'hook)
2529 (custom-add-option 'python-mode-hook 'imenu-add-menubar-index)
2530 (custom-add-option 'python-mode-hook
2531 (lambda ()
2532 "Turn off Indent Tabs mode."
2533 (setq indent-tabs-mode nil)))
2534 (custom-add-option 'python-mode-hook 'turn-on-eldoc-mode)
2535 (custom-add-option 'python-mode-hook 'abbrev-mode)
2536 (custom-add-option 'python-mode-hook 'python-setup-brm)
2537
2538 ;;;###autoload
2539 (define-derived-mode jython-mode python-mode "Jython"
2540 "Major mode for editing Jython files.
2541 Like `python-mode', but sets up parameters for Jython subprocesses.
2542 Runs `jython-mode-hook' after `python-mode-hook'."
2543 :group 'python
2544 (set (make-local-variable 'python-command) python-jython-command))
2545
2546 \f
2547
2548 ;; pdbtrack features
2549
2550 (defun python-comint-output-filter-function (string)
2551 "Watch output for Python prompt and exec next file waiting in queue.
2552 This function is appropriate for `comint-output-filter-functions'."
2553 ;; TBD: this should probably use split-string
2554 (when (and (or (string-equal string ">>> ")
2555 (and (>= (length string) 5)
2556 (string-equal (substring string -5) "\n>>> ")))
2557 python-file-queue)
2558 (condition-case nil
2559 (delete-file (car python-file-queue))
2560 (error nil))
2561 (setq python-file-queue (cdr python-file-queue))
2562 (if python-file-queue
2563 (let ((pyproc (get-buffer-process (current-buffer))))
2564 (python-execute-file pyproc (car python-file-queue))))))
2565
2566 (defun python-pdbtrack-overlay-arrow (activation)
2567 "Activate or deactivate arrow at beginning-of-line in current buffer."
2568 (if activation
2569 (progn
2570 (setq overlay-arrow-position (make-marker)
2571 overlay-arrow-string "=>"
2572 python-pdbtrack-is-tracking-p t)
2573 (set-marker overlay-arrow-position
2574 (save-excursion (beginning-of-line) (point))
2575 (current-buffer)))
2576 (setq overlay-arrow-position nil
2577 python-pdbtrack-is-tracking-p nil)))
2578
2579 (defun python-pdbtrack-track-stack-file (text)
2580 "Show the file indicated by the pdb stack entry line, in a separate window.
2581
2582 Activity is disabled if the buffer-local variable
2583 `python-pdbtrack-do-tracking-p' is nil.
2584
2585 We depend on the pdb input prompt being a match for
2586 `python-pdbtrack-input-prompt'.
2587
2588 If the traceback target file path is invalid, we look for the
2589 most recently visited python-mode buffer which either has the
2590 name of the current function or class, or which defines the
2591 function or class. This is to provide for scripts not in the
2592 local filesytem (e.g., Zope's 'Script \(Python)', but it's not
2593 Zope specific). If you put a copy of the script in a buffer
2594 named for the script and activate python-mode, then pdbtrack will
2595 find it."
2596 ;; Instead of trying to piece things together from partial text
2597 ;; (which can be almost useless depending on Emacs version), we
2598 ;; monitor to the point where we have the next pdb prompt, and then
2599 ;; check all text from comint-last-input-end to process-mark.
2600 ;;
2601 ;; Also, we're very conservative about clearing the overlay arrow,
2602 ;; to minimize residue. This means, for instance, that executing
2603 ;; other pdb commands wipe out the highlight. You can always do a
2604 ;; 'where' (aka 'w') PDB command to reveal the overlay arrow.
2605
2606 (let* ((origbuf (current-buffer))
2607 (currproc (get-buffer-process origbuf)))
2608
2609 (if (not (and currproc python-pdbtrack-do-tracking-p))
2610 (python-pdbtrack-overlay-arrow nil)
2611
2612 (let* ((procmark (process-mark currproc))
2613 (block (buffer-substring (max comint-last-input-end
2614 (- procmark
2615 python-pdbtrack-track-range))
2616 procmark))
2617 target target_fname target_lineno target_buffer)
2618
2619 (if (not (string-match (concat python-pdbtrack-input-prompt "$") block))
2620 (python-pdbtrack-overlay-arrow nil)
2621
2622 (setq target (python-pdbtrack-get-source-buffer block))
2623
2624 (if (stringp target)
2625 (progn
2626 (python-pdbtrack-overlay-arrow nil)
2627 (message "pdbtrack: %s" target))
2628
2629 (setq target_lineno (car target)
2630 target_buffer (cadr target)
2631 target_fname (buffer-file-name target_buffer))
2632 (switch-to-buffer-other-window target_buffer)
2633 (goto-line target_lineno)
2634 (message "pdbtrack: line %s, file %s" target_lineno target_fname)
2635 (python-pdbtrack-overlay-arrow t)
2636 (pop-to-buffer origbuf t)
2637 ;; in large shell buffers, above stuff may cause point to lag output
2638 (goto-char procmark)
2639 )))))
2640 )
2641
2642 (defun python-pdbtrack-get-source-buffer (block)
2643 "Return line number and buffer of code indicated by block's traceback text.
2644
2645 We look first to visit the file indicated in the trace.
2646
2647 Failing that, we look for the most recently visited python-mode buffer
2648 with the same name or having the named function.
2649
2650 If we're unable find the source code we return a string describing the
2651 problem."
2652
2653 (if (not (string-match python-pdbtrack-stack-entry-regexp block))
2654
2655 "Traceback cue not found"
2656
2657 (let* ((filename (match-string 1 block))
2658 (lineno (string-to-number (match-string 2 block)))
2659 (funcname (match-string 3 block))
2660 funcbuffer)
2661
2662 (cond ((file-exists-p filename)
2663 (list lineno (find-file-noselect filename)))
2664
2665 ((setq funcbuffer (python-pdbtrack-grub-for-buffer funcname lineno))
2666 (if (string-match "/Script (Python)$" filename)
2667 ;; Add in number of lines for leading '##' comments:
2668 (setq lineno
2669 (+ lineno
2670 (save-excursion
2671 (set-buffer funcbuffer)
2672 (if (equal (point-min)(point-max))
2673 0
2674 (count-lines
2675 (point-min)
2676 (max (point-min)
2677 (string-match "^\\([^#]\\|#[^#]\\|#$\\)"
2678 (buffer-substring
2679 (point-min) (point-max)))
2680 )))))))
2681 (list lineno funcbuffer))
2682
2683 ((= (elt filename 0) ?\<)
2684 (format "(Non-file source: '%s')" filename))
2685
2686 (t (format "Not found: %s(), %s" funcname filename)))
2687 )
2688 )
2689 )
2690
2691 (defun python-pdbtrack-grub-for-buffer (funcname lineno)
2692 "Find recent python-mode buffer named, or having function named funcname."
2693 (let ((buffers (buffer-list))
2694 buf
2695 got)
2696 (while (and buffers (not got))
2697 (setq buf (car buffers)
2698 buffers (cdr buffers))
2699 (if (and (save-excursion (set-buffer buf)
2700 (string= major-mode "python-mode"))
2701 (or (string-match funcname (buffer-name buf))
2702 (string-match (concat "^\\s-*\\(def\\|class\\)\\s-+"
2703 funcname "\\s-*(")
2704 (save-excursion
2705 (set-buffer buf)
2706 (buffer-substring (point-min)
2707 (point-max))))))
2708 (setq got buf)))
2709 got))
2710
2711 (defun python-toggle-shells (arg)
2712 "Toggles between the CPython and JPython shells.
2713
2714 With positive argument ARG (interactively \\[universal-argument]),
2715 uses the CPython shell, with negative ARG uses the JPython shell, and
2716 with a zero argument, toggles the shell.
2717
2718 Programmatically, ARG can also be one of the symbols `cpython' or
2719 `jpython', equivalent to positive arg and negative arg respectively."
2720 (interactive "P")
2721 ;; default is to toggle
2722 (if (null arg)
2723 (setq arg 0))
2724 ;; preprocess arg
2725 (cond
2726 ((equal arg 0)
2727 ;; toggle
2728 (if (string-equal python-which-bufname "Python")
2729 (setq arg -1)
2730 (setq arg 1)))
2731 ((equal arg 'cpython) (setq arg 1))
2732 ((equal arg 'jpython) (setq arg -1)))
2733 (let (msg)
2734 (cond
2735 ((< 0 arg)
2736 ;; set to CPython
2737 (setq python-which-shell python-python-command
2738 python-which-args python-python-command-args
2739 python-which-bufname "Python"
2740 msg "CPython"
2741 mode-name "Python"))
2742 ((> 0 arg)
2743 (setq python-which-shell python-jython-command
2744 python-which-args python-jython-command-args
2745 python-which-bufname "JPython"
2746 msg "JPython"
2747 mode-name "JPython")))
2748 (message "Using the %s shell" msg)))
2749
2750 ;; Python subprocess utilities and filters
2751 (defun python-execute-file (proc filename)
2752 "Send to Python interpreter process PROC \"execfile('FILENAME')\".
2753 Make that process's buffer visible and force display. Also make
2754 comint believe the user typed this string so that
2755 `kill-output-from-shell' does The Right Thing."
2756 (let ((curbuf (current-buffer))
2757 (procbuf (process-buffer proc))
2758 ; (comint-scroll-to-bottom-on-output t)
2759 (msg (format "## working on region in file %s...\n" filename))
2760 ;; add some comment, so that we can filter it out of history
2761 (cmd (format "execfile(r'%s') # PYTHON-MODE\n" filename)))
2762 (unwind-protect
2763 (save-excursion
2764 (set-buffer procbuf)
2765 (goto-char (point-max))
2766 (move-marker (process-mark proc) (point))
2767 (funcall (process-filter proc) proc msg))
2768 (set-buffer curbuf))
2769 (process-send-string proc cmd)))
2770 ;;;###autoload
2771 (defun python-shell (&optional argprompt)
2772 "Start an interactive Python interpreter in another window.
2773 This is like Shell mode, except that Python is running in the window
2774 instead of a shell. See the `Interactive Shell' and `Shell Mode'
2775 sections of the Emacs manual for details, especially for the key
2776 bindings active in the `*Python*' buffer.
2777
2778 With optional \\[universal-argument], the user is prompted for the
2779 flags to pass to the Python interpreter. This has no effect when this
2780 command is used to switch to an existing process, only when a new
2781 process is started. If you use this, you will probably want to ensure
2782 that the current arguments are retained (they will be included in the
2783 prompt). This argument is ignored when this function is called
2784 programmatically, or when running in Emacs 19.34 or older.
2785
2786 Note: You can toggle between using the CPython interpreter and the
2787 JPython interpreter by hitting \\[python-toggle-shells]. This toggles
2788 buffer local variables which control whether all your subshell
2789 interactions happen to the `*JPython*' or `*Python*' buffers (the
2790 latter is the name used for the CPython buffer).
2791
2792 Warning: Don't use an interactive Python if you change sys.ps1 or
2793 sys.ps2 from their default values, or if you're running code that
2794 prints `>>> ' or `... ' at the start of a line. `python-mode' can't
2795 distinguish your output from Python's output, and assumes that `>>> '
2796 at the start of a line is a prompt from Python. Similarly, the Emacs
2797 Shell mode code assumes that both `>>> ' and `... ' at the start of a
2798 line are Python prompts. Bad things can happen if you fool either
2799 mode.
2800
2801 Warning: If you do any editing *in* the process buffer *while* the
2802 buffer is accepting output from Python, do NOT attempt to `undo' the
2803 changes. Some of the output (nowhere near the parts you changed!) may
2804 be lost if you do. This appears to be an Emacs bug, an unfortunate
2805 interaction between undo and process filters; the same problem exists in
2806 non-Python process buffers using the default (Emacs-supplied) process
2807 filter."
2808 (interactive "P")
2809 ;; Set the default shell if not already set
2810 (when (null python-which-shell)
2811 (python-toggle-shells python-default-interpreter))
2812 (let ((args python-which-args))
2813 (when (and argprompt
2814 (interactive-p)
2815 (fboundp 'split-string))
2816 ;; TBD: Perhaps force "-i" in the final list?
2817 (setq args (split-string
2818 (read-string (concat python-which-bufname
2819 " arguments: ")
2820 (concat
2821 (mapconcat 'identity python-which-args " ") " ")
2822 ))))
2823 (switch-to-buffer-other-window
2824 (apply 'make-comint python-which-bufname python-which-shell nil args))
2825 (make-local-variable 'comint-prompt-regexp)
2826 (set-process-sentinel (get-buffer-process (current-buffer))
2827 'python-sentinel)
2828 (setq comint-prompt-regexp "^>>> \\|^[.][.][.] \\|^(pdb) ")
2829 (add-hook 'comint-output-filter-functions
2830 'python-comint-output-filter-function nil t)
2831 ;; pdbtrack
2832 (set-syntax-table python-mode-syntax-table)
2833 (use-local-map python-shell-map)))
2834
2835 (defun python-pdbtrack-toggle-stack-tracking (arg)
2836 (interactive "P")
2837 (if (not (get-buffer-process (current-buffer)))
2838 (error "No process associated with buffer '%s'" (current-buffer)))
2839 ;; missing or 0 is toggle, >0 turn on, <0 turn off
2840 (if (or (not arg)
2841 (zerop (setq arg (prefix-numeric-value arg))))
2842 (setq python-pdbtrack-do-tracking-p (not python-pdbtrack-do-tracking-p))
2843 (setq python-pdbtrack-do-tracking-p (> arg 0)))
2844 (message "%sabled Python's pdbtrack"
2845 (if python-pdbtrack-do-tracking-p "En" "Dis")))
2846
2847 (defun turn-on-pdbtrack ()
2848 (interactive)
2849 (python-pdbtrack-toggle-stack-tracking 1))
2850
2851 (defun turn-off-pdbtrack ()
2852 (interactive)
2853 (python-pdbtrack-toggle-stack-tracking 0))
2854
2855 (defun python-sentinel (proc msg)
2856 (setq overlay-arrow-position nil))
2857
2858 (provide 'python)
2859 (provide 'python-21)
2860
2861 ;; arch-tag: 6fce1d99-a704-4de9-ba19-c6e4912b0554
2862 ;;; python.el ends here