0
0
Fork 1
mirror of https://mau.dev/maunium/synapse.git synced 2024-12-15 14:23:50 +01:00

Correctly reraise exception

This commit is contained in:
Erik Johnston 2018-01-12 14:44:02 +00:00
parent 85a4d78213
commit e21370ba54

View file

@ -24,6 +24,7 @@ import contextlib
import os import os
import logging import logging
import shutil import shutil
import sys
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -114,12 +115,13 @@ class MediaStorage(object):
try: try:
with open(fname, "wb") as f: with open(fname, "wb") as f:
yield f, fname, finish yield f, fname, finish
except Exception as e: except Exception:
t, v, tb = sys.exc_info()
try: try:
os.remove(fname) os.remove(fname)
except Exception: except Exception:
pass pass
raise e raise t, v, tb
if not finished_called: if not finished_called:
raise Exception("Finished callback not called") raise Exception("Finished callback not called")