oci/zeronet
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

atomicWrite function now preserve file permissions

Browse source
This commit is contained in:
Thibaut Broggi 2016-05-26 15:59:40 +02:00
parent 81f196647b
commit a10e80318f
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/util/helper.py
View file

@ -1,4 +1,5 @@
import os import os
import stat
import socket import socket
import struct import struct
import re import re
@ -10,6 +11,7 @@ import base64
def atomicWrite(dest, content, mode="w"): def atomicWrite(dest, content, mode="w"):
try: try:
permissions = stat.S_IMODE(os.lstat(dest).st_mode)
with open(dest + "-new", mode) as f: with open(dest + "-new", mode) as f:
f.write(content) f.write(content)
f.flush() f.flush()
@ -18,6 +20,7 @@ def atomicWrite(dest, content, mode="w"):
os.rename(dest + "-old", dest + "-old-%s" % time.time()) os.rename(dest + "-old", dest + "-old-%s" % time.time())
os.rename(dest, dest + "-old") os.rename(dest, dest + "-old")
os.rename(dest + "-new", dest) os.rename(dest + "-new", dest)
os.chmod(dest, permissions)
os.unlink(dest + "-old") os.unlink(dest + "-old")
return True return True
except Exception, err: except Exception, err: