atomic csv write

gracchus163 · Jul 24, 2020 · 899ff5a · 899ff5a
1 parent 9bbfc07
commit 899ff5a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/bot_commands.py b/bot_commands.py
@@ -2,6 +2,7 @@
 
 from asyncio import Lock
 import logging
+from os import fsync, rename
 
 from bot_actions import community_invite, is_admin, valid_token
 from chat_functions import send_text_to_room
@@ -114,9 +115,14 @@ async def _process_request(self, ticket_type):
 
                 if tokens[h] == "unused":
                     tokens[h] = self.event.sender
-                    with open(filename, "w") as f:
+                    filename_temp = filename + ".atomic"
+                    with open(filename_temp, "w") as f:
                         for key in tokens.keys():
                             f.write("%s,%s\n" % (key, tokens[key]))
+                        f.flush()
+                        fsync(f.fileno())
+                    rename(filename_temp, filename)
+
                 return
             else:
                 logging.info(