diff options
Diffstat (limited to 'snaps/file_utils.py')
-rw-r--r-- | snaps/file_utils.py | 15 |
1 files changed, 10 insertions, 5 deletions
diff --git a/snaps/file_utils.py b/snaps/file_utils.py index 7b93a6d..819c707 100644 --- a/snaps/file_utils.py +++ b/snaps/file_utils.py @@ -55,11 +55,16 @@ def download(url, dest_path, name=None): dest = dest_path + '/' + name logger.debug('Downloading file from - ' + url) # Override proxy settings to use localhost to download file - with open(dest, 'wb') as f: - logger.debug('Saving file to - ' + dest) - response = __get_url_response(url) - f.write(response.read()) - return f + f = None + try: + with open(dest, 'wb') as f: + logger.debug('Saving file to - ' + dest) + response = __get_url_response(url) + f.write(response.read()) + return f + finally: + if f: + f.close() def get_content_length(url): |