return original text on non-zero exit in external editor

thomasballinger · thomasballinger · commit c5edb5dbc7ab · 2014-03-04T12:36:22.000-05:00
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -995,6 +995,7 @@ def clear_current_line(self):
         It prevents autoindentation from occuring after a traceback."""
 
     def send_to_external_editor(self, text, filename=None):
+        """Returns modified text from an editor, or the oriignal text if editor exited with non-zero"""
         editor = (self.config.editor or
                   os.environ.get('VISUAL', os.environ.get('EDITOR', 'vim')))
         editor_args = shlex.split(editor)
@@ -1004,6 +1005,8 @@ def send_to_external_editor(self, text, filename=None):
             if subprocess.call(editor_args + [temp.name]) == 0:
                 with open(temp.name) as f:
                     return f.read()
+            else:
+                return text
 
 
 def next_indentation(line, tab_length):
