Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Use atomic_write() for duktape writefile implementation

Browse Source
This commit is contained in:
Kovid Goyal 2015-11-17 23:51:25 +05:30
parent 66b0234c8b
commit 9175ef0925
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/duktape/__init__.py
View File

@ -9,7 +9,7 @@ __docformat__ = 'restructuredtext en'
__all__ = ['dukpy', 'Context', 'undefined', 'JSError', 'to_python'] __all__ = ['dukpy', 'Context', 'undefined', 'JSError', 'to_python']
import errno, os, sys, numbers, hashlib, json import errno, os, sys, numbers, hashlib, json, tempfile, stat
from functools import partial from functools import partial
from calibre.constants import plugins from calibre.constants import plugins
@ -119,14 +119,19 @@ def readfile(path, enc='utf-8'):
except EnvironmentError as e: except EnvironmentError as e:
return [None, errno.errorcode[e.errno], 'Failed to read from file: %s with error: %s' % (path, e.message or e)] return [None, errno.errorcode[e.errno], 'Failed to read from file: %s with error: %s' % (path, e.message or e)]
def atomic_write(name, raw):
with tempfile.NamedTemporaryFile(dir=os.getcwdu(), delete=False) as f:
f.write(raw)
os.fchmod(f.fileno(), stat.S_IREAD|stat.S_IWRITE|stat.S_IRGRP|stat.S_IROTH)
os.rename(f.name, name)
def writefile(path, data, enc='utf-8'): def writefile(path, data, enc='utf-8'):
if enc == undefined: if enc == undefined:
enc = 'utf-8' enc = 'utf-8'
try: try:
if isinstance(data, type('')): if isinstance(data, type('')):
data = data.encode(enc or 'utf-8') data = data.encode(enc or 'utf-8')
with open(path, 'wb') as f: atomic_write(path, data)
f.write(data)
except UnicodeEncodeError as e: except UnicodeEncodeError as e:
return ['', 'Failed to encode the data for file: %s with specified encoding: %s' % (path, enc)] return ['', 'Failed to encode the data for file: %s with specified encoding: %s' % (path, enc)]
except EnvironmentError as e: except EnvironmentError as e: