Set mod of history file to 0600 (fixes bpython#489)

sebastinas · sebastinas · commit 28510c9a2a77 · 2015-02-23T17:47:28.000+01:00
Signed-off-by: Sebastian Ramacher <sebastian+dev@ramacher.at> (cherry picked from commit d4d9616)
diff --git a/bpython/history.py b/bpython/history.py
@@ -26,6 +26,7 @@
 from __future__ import unicode_literals
 import io
 import os
+import stat
 from itertools import islice
 from six.moves import range
 
@@ -187,6 +188,8 @@ def load_from(self, fd):
         return entries
 
     def save(self, filename, encoding, lines=0):
+        fd = os.open(filename, os.O_WRONLY | os.O_CREAT | os.TRUNC,
+                     stat.S_IRUSR | stat.S_IWUSR)
         with io.open(filename, 'w', encoding=encoding,
                      errors='ignore') as hfile:
             with FileLock(hfile):
@@ -204,7 +207,9 @@ def append_reload_and_write(self, s, filename, encoding):
             return self.append(s)
 
         try:
-            with io.open(filename, 'a+', encoding=encoding,
+            fd = os.open(filename, os.O_APPEND | os.O_RDWR | os.O_CREAT,
+                         stat.S_IRUSR | stat.S_IWUSR)
+            with io.open(fd, 'a+', encoding=encoding,
                          errors='ignore') as hfile:
                 with FileLock(hfile):
                     # read entries
