reivilibre/scone
1
0
Fork 0
Code Issues 21 Pull Requests Releases Wiki Activity

Write files atomically

Browse Source 
Fixes #20.
This commit is contained in:
Olivier 'reivilibre' 2021-07-03 10:53:38 +01:00
parent aecf39f23c
commit 9ab4044582
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scone/default/utensils/basic_utensils.py
View File

@ -35,10 +35,13 @@ from scone.sous.utensils import Utensil, Worktop
class WriteFile(Utensil):
path: str
mode: int
atomic: bool = attr.ib(default=True)
async def execute(self, channel: Channel, worktop):
oldumask = os.umask(0)
fdnum = os.open(self.path, os.O_WRONLY | os.O_CREAT | os.O_TRUNC, self.mode)
temp_path = self.path + "._scone-part"
write_path = temp_path if self.atomic else self.path
fdnum = os.open(write_path, os.O_WRONLY | os.O_CREAT | os.O_TRUNC, self.mode)
os.umask(oldumask)
with open(fdnum, "wb") as file:
@ -49,6 +52,9 @@ class WriteFile(Utensil):
assert isinstance(next_chunk, bytes)
file.write(next_chunk)
if self.atomic:
shutil.move(temp_path, self.path)
await channel.send("OK")