safer handling of result archives

ReCodEx · Dec 13, 2017 · 0c82728 · 0c82728
1 parent 40f9c75
commit 0c82728
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/fileserver/views.py b/fileserver/views.py
@@ -107,9 +107,12 @@ def store_result(id, ext, dirs: DirectoryStructure):
     """
 
     path = os.path.join(dirs.result_dir, "{0}.{1}".format(id, ext))
-    with open(path, 'wb') as f:
+    path_tmp = path + ".part"
+    with open(path_tmp, 'wb') as f:
         f.write(request.data)
 
+    os.replace(path_tmp, path)
+
     return json.dumps({
         "result": "OK"
     })