mercurial-scm/hg-stable: mercurial/util.py comparison

add an atomic argument to util.opener when atomic is used, the file while be renamed on close

author	Benoit Boissinot <benoit.boissinot@ens-lyon.org>
date	Fri, 11 Nov 2005 15:33:59 -0800
parents	11a58d2cdffb
children	bf4e7ef08741

comparison

equal deleted inserted replaced

-:c13fce7167c2
+:c9f33196805b
 this function is used to hide the details of COW semantics and
 remote file access from higher level code.
 """
 p = base
-def o(path, mode="r", text=False):
+def mktempcopy(name):
+d, fn = os.path.split(name)
+fd, temp = tempfile.mkstemp(prefix=fn, dir=d)
+fp = os.fdopen(fd, "wb")
+try:
+fp.write(file(name, "rb").read())
+except:
+try: os.unlink(temp)
+except: pass
+raise
+fp.close()
+st = os.lstat(name)
+os.chmod(temp, st.st_mode)
+return temp
+class atomicfile(file):
+"""the file will only be copied on close"""
+def __init__(self, name, mode, atomic=False):
+self.__name = name
+self.temp = mktempcopy(name)
+file.__init__(self, self.temp, mode)
+def close(self):
+if not self.closed:
+rename(self.temp, self.__name)
+file.close(self)
+def __del__(self):
+self.close()
+def o(path, mode="r", text=False, atomic=False):
 f = os.path.join(p, path)
 if not text:
 mode += "b" # for that other OS
 except OSError:
 d = os.path.dirname(f)
 if not os.path.isdir(d):
 os.makedirs(d)
 else:
+if atomic:
+return atomicfile(f, mode)
 if nlink > 1:
-d, fn = os.path.split(f)
+rename(mktempcopy(f), f)
-fd, temp = tempfile.mkstemp(prefix=fn, dir=d)
-fp = os.fdopen(fd, "wb")
-try:
-fp.write(file(f, "rb").read())
-except:
-try: os.unlink(temp)
-except: pass
-raise
-fp.close()
-st = os.lstat(f)
-os.chmod(temp, st.st_mode)
-rename(temp, f)
 return file(f, mode)
 return o
 def _makelock_file(info, pathname):

Mercurial > public > mercurial-scm > hg-stable