* progmodes/python.el:

(python-shell-send-buffer): Better handling of "if __name__ ==
'__main__':" conditionals when sending the buffer.

diff --git a/lisp/ChangeLog b/lisp/ChangeLog
index 270d548b71854daaed86fcce21139ea5b70b7812..5539ee41b26c75e822886765b60d38a68aeac8f9 100644 (file)
--- a/lisp/ChangeLog
+++ b/lisp/ChangeLog
@@ -1,3 +1,9 @@
+2012-09-24  Fabián Ezequiel Gallina  <fgallina@cuca>
+
+       * progmodes/python.el:
+       (python-shell-send-buffer): Better handling of "if __name__ ==
+       '__main__':" conditionals when sending the buffer.
+
 2012-09-24  Glenn Morris  <rgm@gnu.org>
 
        * eshell/esh-cmd.el (eshell-find-alias-function):

diff --git a/lisp/progmodes/python.el b/lisp/progmodes/python.el
index cc835ca79ac35c865c1d860a20a588b904c70202..e99e6bda4b830b1fe447322a01df68ca311a09f1 100644 (file)
--- a/lisp/progmodes/python.el
+++ b/lisp/progmodes/python.el
@@ -1919,19 +1919,18 @@ Returns the output.  See `python-shell-send-string-no-output'."
 
 (defun python-shell-send-buffer (&optional arg)
   "Send the entire buffer to inferior Python process.
-
-With prefix ARG include lines surrounded by \"if __name__ == '__main__':\""
+With prefix ARG allow execution of code inside blocks delimited
+by \"if __name__== '__main__':\""
   (interactive "P")
   (save-restriction
     (widen)
-    (python-shell-send-region
-     (point-min)
-     (or (and
-          (not arg)
-          (save-excursion
-            (re-search-forward (python-rx if-name-main) nil t))
-          (match-beginning 0))
-         (point-max)))))
+    (let ((str (buffer-substring (point-min) (point-max))))
+      (and
+       (not arg)
+       (setq str (replace-regexp-in-string
+                  (python-rx if-name-main)
+                  "if __name__ == '__main__ ':" str)))
+      (python-shell-send-string str))))
 
 (defun python-shell-send-defun (arg)
   "Send the current defun to inferior Python process.