diff options
author | Serhiy Storchaka <storchaka@gmail.com> | 2013-05-09 11:36:58 (GMT) |
---|---|---|
committer | Serhiy Storchaka <storchaka@gmail.com> | 2013-05-09 11:36:58 (GMT) |
commit | b3c5d8568030842c6fbd30a6f197c8b684e0ccc6 (patch) | |
tree | 517e83deb06792e056b50606483824d66db46f8f | |
parent | f28dfdd07b57fa16f7d489e7ac99d9a15f1bef80 (diff) | |
parent | 263fab94ee1f32b53b709c25c76044719d0eb88e (diff) | |
download | cpython-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.
-rw-r--r-- | Lib/tarfile.py | 12 | ||||
-rw-r--r-- | Lib/test/test_tarfile.py | 8 | ||||
-rw-r--r-- | Misc/ACKS | 1 | ||||
-rw-r--r-- | Misc/NEWS | 3 |
4 files changed, 19 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): @@ -118,6 +118,7 @@ Adrian von Bidder David Binger Dominic Binks Philippe Biondi +Michael Birtwell Stuart Bishop Roy Bixler Daniel Black @@ -86,6 +86,9 @@ Core and Builtins Library ------- +- Issue #16601: Restarting iteration over tarfile no more continues from where + it left off. Patch by Michael Birtwell. + - Issue #17289: The readline module now plays nicer with external modules or applications changing the rl_completer_word_break_characters global variable. Initial patch by Bradley Froehle. |