Enhancements on `python-beginning-of-defun-function' for `mark-defun' behavior.

`mark-defun' was failing when point was right at the start of a defun.

1 files changed, 13 insertions, 10 deletions

diff --git a/lisp/progmodes/python.el b/lisp/progmodes/python.el
index 4654143a160..1279c38b68b 100644
--- a/lisp/progmodes/python.el
+++ b/lisp/progmodes/python.el
@@ -1052,16 +1052,19 @@ negative do the same but backwards.  When NODECORATORS is non-nil
 decorators are not included.  Return non-nil if point is moved to the
 `beginning-of-defun'."
   (when (or (null arg) (= arg 0)) (setq arg 1))
-  (if (> arg 0)
-      (dotimes (i arg (python-nav-beginning-of-defun nodecorators)))
-    (let ((found))
-      (dotimes (i (- arg) found)
-        (python-end-of-defun-function)
-        (python-util-forward-comment)
-        (goto-char (line-end-position))
-        (when (not (eobp))
-          (setq found
-                (python-nav-beginning-of-defun nodecorators)))))))
+  (cond ((and (eq this-command 'mark-defun)
+              (looking-at python-nav-beginning-of-defun-regexp)))
+        ((> arg 0)
+         (dotimes (i arg (python-nav-beginning-of-defun nodecorators))))
+        (t
+         (let ((found))
+           (dotimes (i (- arg) found)
+             (python-end-of-defun-function)
+             (python-util-forward-comment)
+             (goto-char (line-end-position))
+             (when (not (eobp))
+               (setq found
+                     (python-nav-beginning-of-defun nodecorators))))))))
 
 (defun python-end-of-defun-function ()
   "Move point to the end of def or class.