From d7be2ca16daad164c18dd4461863c3667ca413b0 Mon Sep 17 00:00:00 2001 From: Kevin Froman Date: Fri, 3 Jan 2020 04:14:04 -0600 Subject: [PATCH] delete upload memory file file on shutdown if upload queue is empty --- src/communicator/uploadqueue/__init__.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/src/communicator/uploadqueue/__init__.py b/src/communicator/uploadqueue/__init__.py index 8763a4d4..02e4df8b 100644 --- a/src/communicator/uploadqueue/__init__.py +++ b/src/communicator/uploadqueue/__init__.py @@ -59,5 +59,11 @@ class UploadQueue: def save(self): """Save to disk on shutdown or if called manually.""" bl: deadsimplekv.DeadSimpleKV = self.communicator.blocksToUpload - self.store_obj.put('uploads', bl) - self.store_obj.flush() + if len(bl) == 0: + try: + os.remove(UPLOAD_MEMORY_FILE) + except FileNotFoundError: + pass + else: + self.store_obj.put('uploads', bl) + self.store_obj.flush()