Add option to append '.py' to filesnames while saving

sebastinas · sebastinas · commit d5f4c2a0a038 · 2013-03-13T01:09:47.000+01:00
diff --git a/bpython/config.py b/bpython/config.py
@@ -72,6 +72,7 @@ def loadini(struct, configfile):
             'pastebin_private': True,
             'pastebin_show_url': 'http://bpaste.net/show/$paste_id/',
             'pastebin_helper': '',
+            'save_append_py': False
         },
         'keyboard': {
             'clear_line': 'C-u',
@@ -152,6 +153,7 @@ def loadini(struct, configfile):
     methods = config.get('general', 'magic_methods')
     struct.magic_methods = [meth.strip() for meth in methods.split(",")]
     struct.autocomplete_mode = config.get('general', 'autocomplete_mode')
+    struct.save_append_py = config.getboolean('general', 'save_append_py')
 
     struct.gtk_font = config.get('gtk', 'font')
 
diff --git a/bpython/repl.py b/bpython/repl.py
@@ -704,6 +704,8 @@ def write2file(self):
 
         if fn.startswith('~'):
             fn = os.path.expanduser(fn)
+        if not fn.endswith('.py') and self.config.save_append_py:
+            fn = fn + '.py'
 
         s = self.formatforfile(self.getstdout())
 
diff --git a/doc/sphinx/source/configuration-options.rst b/doc/sphinx/source/configuration-options.rst
@@ -130,6 +130,12 @@ flush_output
 ^^^^^^^^^^^^
 Whether to flush all output to stdout on exit (default: True).
 
+save_append_py
+^^^^^^^^^^^^^^
+Whether to append ``.py`` to the filename while saving the input to a file.
+
+.. versionadded:: 0.13
+
 Keyboard
 --------
 This section refers to the ``[keyboard]`` section in your ``$XDG_CONFIG_HOME/bpython/config``.
