Merge pull request #829 from python-cmd2/line_buffering

Enabled line buffering when redirecting output to a file

1 files changed, 9 insertions, 6 deletions

diff --git a/cmd2/cmd2.py b/cmd2/cmd2.py
index 753fb5ab..bd581919 100644
--- a/cmd2/cmd2.py
+++ b/cmd2/cmd2.py
@@ -1874,22 +1874,25 @@ class Cmd(cmd.Cmd):
                 self.perror("Cannot redirect to paste buffer; missing 'pyperclip' and/or pyperclip dependencies")
                 redir_error = True
 
+            # Redirecting to a file
             elif statement.output_to:
-                # going to a file
-                mode = 'w'
-                # statement.output can only contain
-                # REDIRECTION_APPEND or REDIRECTION_OUTPUT
+                # statement.output can only contain REDIRECTION_APPEND or REDIRECTION_OUTPUT
                 if statement.output == constants.REDIRECTION_APPEND:
                     mode = 'a'
+                else:
+                    mode = 'w'
+
                 try:
-                    new_stdout = open(utils.strip_quotes(statement.output_to), mode)
+                    # Use line buffering
+                    new_stdout = open(utils.strip_quotes(statement.output_to), mode=mode, buffering=1)
                     saved_state.redirecting = True
                     sys.stdout = self.stdout = new_stdout
                 except OSError as ex:
                     self.pexcept('Failed to redirect because - {}'.format(ex))
                     redir_error = True
+
+            # Redirecting to a paste buffer
             else:
-                # going to a paste buffer
                 new_stdout = tempfile.TemporaryFile(mode="w+")
                 saved_state.redirecting = True
                 sys.stdout = self.stdout = new_stdout