From 5ab67632443d81568afdc1f401c1db72046eed23 Mon Sep 17 00:00:00 2001 From: Daira Hopwood Date: Tue, 9 Sep 2014 18:35:53 +0100 Subject: [PATCH] WIP Signed-off-by: Daira Hopwood --- src/allmydata/util/fileutil.py | 22 +++++++++++++++++----- 1 file changed, 17 insertions(+), 5 deletions(-) diff --git a/src/allmydata/util/fileutil.py b/src/allmydata/util/fileutil.py index e71ddb4d..582139e1 100644 --- a/src/allmydata/util/fileutil.py +++ b/src/allmydata/util/fileutil.py @@ -218,15 +218,18 @@ def rm_dir(dirname): raise OSError, excs -def remove_if_possible(f): +def remove_if_possible(path): + #precondition_abspath(path) + try: - remove(f) + remove(path) except: pass def du(basedir): - size = 0 + #precondition_abspath(basedir) + size = 0 for root, dirs, files in os.walk(basedir): for f in files: fn = os.path.join(root, f) @@ -237,19 +240,26 @@ def du(basedir): def move_into_place(source, dest): """Atomically replace a file, or as near to it as the platform allows. The dest file may or may not exist.""" + #precondition_abspath(source) + #precondition_abspath(dest) + if "win32" in sys.platform.lower(): remove_if_possible(dest) os.rename(source, dest) def write_atomically(target, contents, mode="b"): - f = open(target+".tmp", "w"+mode) + #precondition_abspath(target) + + f = open(target + u".tmp", "w"+mode) try: f.write(contents) finally: f.close() - move_into_place(target+".tmp", target) + move_into_place(target + u".tmp", target) def write(path, data, mode="wb"): + #precondition_abspath(path) + wf = open(path, mode) try: wf.write(data) @@ -257,6 +267,8 @@ def write(path, data, mode="wb"): wf.close() def read(path): + #precondition_abspath(path) + rf = open(path, "rb") try: return rf.read() -- 2.45.2