Save format changed to be executable code

bobf · bobf · commit d486b95b9a42 · 2009-02-23T19:38:44.000Z
Save to file now outputs executable code (i.e. without the &gt;&gt;&gt; and ... and with
"# OUT: " prepended to all output lines). I never used this feature much but
someone asked for this behaviour.

committer: Bob Farrell &lt;robertanthonyfarrell@gmail.com&gt;

--HG--
extra : convert_revision : 6f48a2ca66db9e7c30b146ea0812812e93da5f7d
diff --git a/CHANGELOG b/CHANGELOG
@@ -19,6 +19,10 @@ encoding.
 Bohdan Vlasyuk sent me a patch that fixes a problem with the above patch from
 Mark if sys.__stdout__.encoding didn't exist.
 
+Save to file now outputs executable code (i.e. without the >>> and ... and with
+"# OUT: " prepended to all output lines). I never used this feature much but
+someone asked for this behaviour.
+
 v0.7.1
 ======
 Added support for a history file, defaults to ~/.pythonhist and 100 lines but
diff --git a/bpython/cli.py b/bpython/cli.py
@@ -688,6 +688,19 @@ def getstdout(self):
 
         return self.stdout_hist + '\n'
 
+    def formatforfile(self, s):
+        """Format the stdout buffer to something suitable for writing to disk,
+        i.e. without >>> and ... at input lines and with "# OUT: " prepended to
+        output lines."""
+
+        def process():
+            for line in s.split('\n'):
+                if line.startswith('>>>') or line.startswith('...'):
+                    yield line[4:]
+                elif line.rstrip():
+                    yield "# OUT: %s" % (line,)
+        return "\n".join(process())
+
     def write2file(self):
         """Prompt for a filename and write the current contents of the stdout
         buffer to disk."""
@@ -697,7 +710,7 @@ def write2file(self):
         if fn.startswith('~'):
             fn = os.path.expanduser(fn)
 
-        s = self.getstdout()
+        s = self.formatforfile(self.getstdout())
 
         try:
             f = open(fn, 'w')
