kev/Onionr
1
0
Fork 0
Code Issues 24 Pull Requests Projects Releases 2 Wiki Activity

delete upload memory file file on shutdown if upload queue is empty

Browse Source
This commit is contained in:
Kevin Froman 2020-01-03 04:14:04 -06:00
parent ee7b68c844
commit d7be2ca16d
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/communicator/uploadqueue/__init__.py
View File

@ -59,5 +59,11 @@ class UploadQueue:
def save(self): def save(self):
"""Save to disk on shutdown or if called manually.""" """Save to disk on shutdown or if called manually."""
bl: deadsimplekv.DeadSimpleKV = self.communicator.blocksToUpload bl: deadsimplekv.DeadSimpleKV = self.communicator.blocksToUpload
self.store_obj.put('uploads', bl) if len(bl) == 0:
self.store_obj.flush() try:
os.remove(UPLOAD_MEMORY_FILE)
except FileNotFoundError:
pass
else:
self.store_obj.put('uploads', bl)
self.store_obj.flush()