summaryrefslogtreecommitdiffstats
path: root/Lib
diff options
context:
space:
mode:
authorSerhiy Storchaka <storchaka@gmail.com>2013-05-09 11:36:58 (GMT)
committerSerhiy Storchaka <storchaka@gmail.com>2013-05-09 11:36:58 (GMT)
commitb3c5d8568030842c6fbd30a6f197c8b684e0ccc6 (patch)
tree517e83deb06792e056b50606483824d66db46f8f /Lib
parentf28dfdd07b57fa16f7d489e7ac99d9a15f1bef80 (diff)
parent263fab94ee1f32b53b709c25c76044719d0eb88e (diff)
downloadcpython-b3c5d8568030842c6fbd30a6f197c8b684e0ccc6.zip
cpython-b3c5d8568030842c6fbd30a6f197c8b684e0ccc6.tar.gz
cpython-b3c5d8568030842c6fbd30a6f197c8b684e0ccc6.tar.bz2
Issue #16601: Restarting iteration over tarfile no more continues from where
it left off. Patch by Michael Birtwell.
Diffstat (limited to 'Lib')
-rw-r--r--Lib/tarfile.py12
-rw-r--r--Lib/test/test_tarfile.py8
2 files changed, 15 insertions, 5 deletions
diff --git a/Lib/tarfile.py b/Lib/tarfile.py
index ffa7327..06df7df 100644
--- a/Lib/tarfile.py
+++ b/Lib/tarfile.py
@@ -2397,16 +2397,18 @@ class TarIter:
# Fix for SF #1100429: Under rare circumstances it can
# happen that getmembers() is called during iteration,
# which will cause TarIter to stop prematurely.
- if not self.tarfile._loaded:
+
+ if self.index == 0 and self.tarfile.firstmember is not None:
+ tarinfo = self.tarfile.next()
+ elif self.index < len(self.tarfile.members):
+ tarinfo = self.tarfile.members[self.index]
+ elif not self.tarfile._loaded:
tarinfo = self.tarfile.next()
if not tarinfo:
self.tarfile._loaded = True
raise StopIteration
else:
- try:
- tarinfo = self.tarfile.members[self.index]
- except IndexError:
- raise StopIteration
+ raise StopIteration
self.index += 1
return tarinfo
diff --git a/Lib/test/test_tarfile.py b/Lib/test/test_tarfile.py
index 5bbd48e..a69437c 100644
--- a/Lib/test/test_tarfile.py
+++ b/Lib/test/test_tarfile.py
@@ -415,6 +415,14 @@ class MiscReadTest(CommonReadTest):
finally:
support.unlink(empty)
+ def test_parallel_iteration(self):
+ # Issue #16601: Restarting iteration over tarfile continued
+ # from where it left off.
+ with tarfile.open(self.tarname) as tar:
+ for m1, m2 in zip(tar, tar):
+ self.assertEqual(m1.offset, m2.offset)
+ self.assertEqual(m1.get_info(), m2.get_info())
+
class StreamReadTest(CommonReadTest):