mirror of
https://github.com/onionshare/onionshare.git
synced 2025-06-27 07:50:39 -04:00
Catch more exceptions in ReceiveModeFile that trigger on a full disk
This commit is contained in:
parent
09c2696c27
commit
e625d3b009
1 changed files with 10 additions and 7 deletions
|
@ -198,22 +198,25 @@ class ReceiveModeFile(object):
|
|||
|
||||
try:
|
||||
bytes_written = self.f.write(b)
|
||||
except:
|
||||
# If we can't write the file, close early
|
||||
self.upload_error = True
|
||||
return
|
||||
self.onionshare_write_func(self.onionshare_filename, bytes_written)
|
||||
|
||||
except:
|
||||
self.upload_error = True
|
||||
|
||||
def close(self):
|
||||
"""
|
||||
Custom close method that calls out to onionshare_close_func
|
||||
"""
|
||||
try:
|
||||
self.f.close()
|
||||
|
||||
if not self.upload_error:
|
||||
# Rename the in progress file to the final filename
|
||||
os.rename(self.filename_in_progress, self.filename)
|
||||
|
||||
except:
|
||||
self.upload_error = True
|
||||
|
||||
self.onionshare_close_func(self.onionshare_filename)
|
||||
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue