diff --git a/lib/galaxy/tools/imp_exp/unpack_tar_gz_archive.py b/lib/galaxy/tools/imp_exp/unpack_tar_gz_archive.py index 5611fe6d2be..f3fad868169 100644 --- a/lib/galaxy/tools/imp_exp/unpack_tar_gz_archive.py +++ b/lib/galaxy/tools/imp_exp/unpack_tar_gz_archive.py @@ -29,14 +29,13 @@ def url_to_file(url, dest_file): assert url_reader.ok, "History import failed, server returned '%s'" % url_reader.reason CHUNK = 10 * 1024 # 10k total = 0 - fp = open(dest_file, 'wb') - for chunk in url_reader.iter_content(chunk_size=CHUNK): - if chunk: - fp.write(chunk) - total += CHUNK - if total > MAX_SIZE: - break - fp.close() + with open(dest_file, 'wb') as fp: + for chunk in url_reader.iter_content(chunk_size=CHUNK): + if chunk: + fp.write(chunk) + total += CHUNK + if total > MAX_SIZE: + break return dest_file