[Python-checkins] cpython (3.2): Issue #16477: Close tarfile internal handlers in case of exception.
andrew.svetlov
python-checkins at python.org
Thu Nov 29 13:22:43 CET 2012
http://hg.python.org/cpython/rev/b7bdc0b3c2fe
changeset: 80648:b7bdc0b3c2fe
branch: 3.2
parent: 80645:4ec9f0b0f8f1
user: Andrew Svetlov <andrew.svetlov at gmail.com>
date: Thu Nov 29 14:20:47 2012 +0200
summary:
Issue #16477: Close tarfile internal handlers in case of exception.
Patch by Serhiy Storchaka.
files:
Lib/tarfile.py | 24 +++++++++++-------------
1 files changed, 11 insertions(+), 13 deletions(-)
diff --git a/Lib/tarfile.py b/Lib/tarfile.py
--- a/Lib/tarfile.py
+++ b/Lib/tarfile.py
@@ -2077,9 +2077,8 @@
# Append the tar header and data to the archive.
if tarinfo.isreg():
- f = bltn_open(name, "rb")
- self.addfile(tarinfo, f)
- f.close()
+ with bltn_open(name, "rb") as f:
+ self.addfile(tarinfo, f)
elif tarinfo.isdir():
self.addfile(tarinfo)
@@ -2292,16 +2291,15 @@
"""
source = self.fileobj
source.seek(tarinfo.offset_data)
- target = bltn_open(targetpath, "wb")
- if tarinfo.sparse is not None:
- for offset, size in tarinfo.sparse:
- target.seek(offset)
- copyfileobj(source, target, size)
- else:
- copyfileobj(source, target, tarinfo.size)
- target.seek(tarinfo.size)
- target.truncate()
- target.close()
+ with bltn_open(targetpath, "wb") as target:
+ if tarinfo.sparse is not None:
+ for offset, size in tarinfo.sparse:
+ target.seek(offset)
+ copyfileobj(source, target, size)
+ else:
+ copyfileobj(source, target, tarinfo.size)
+ target.seek(tarinfo.size)
+ target.truncate()
def makeunknown(self, tarinfo, targetpath):
"""Make a file from a TarInfo object with an unknown type
--
Repository URL: http://hg.python.org/cpython
More information about the Python-checkins
mailing list