mirror of
https://github.com/Mayuri-Chan/pyrofork.git
synced 2026-01-01 05:04:51 +00:00
Check whether get_file failed or not #37
If it failed, also delete any eventual temporary file
This commit is contained in:
parent
440654a63f
commit
2b33f23990
1 changed files with 9 additions and 1 deletions
|
|
@ -567,6 +567,9 @@ class Client:
|
|||
progress=progress
|
||||
)
|
||||
|
||||
if tmp_file_name is None:
|
||||
return None
|
||||
|
||||
if file_name is not None:
|
||||
path[0] = "downloads/{}".format(file_name)
|
||||
|
||||
|
|
@ -2334,11 +2337,16 @@ class Client:
|
|||
if len(chunk) < limit:
|
||||
break
|
||||
except Exception as e:
|
||||
log.error(e)
|
||||
raise e
|
||||
finally:
|
||||
cdn_session.stop()
|
||||
except Exception as e:
|
||||
log.error(e)
|
||||
|
||||
try:
|
||||
os.remove(file_name)
|
||||
except OSError:
|
||||
pass
|
||||
else:
|
||||
return file_name
|
||||
finally:
|
||||
|
|
|
|||
Loading…
Reference in a new issue