| Index: third_party/gsutil/gslib/plurality_checkable_iterator.py
|
| diff --git a/third_party/gsutil/gslib/plurality_checkable_iterator.py b/third_party/gsutil/gslib/plurality_checkable_iterator.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..a24b93b1efb0ce2756450e2b8764131123e7f955
|
| --- /dev/null
|
| +++ b/third_party/gsutil/gslib/plurality_checkable_iterator.py
|
| @@ -0,0 +1,56 @@
|
| +# Copyright 2012 Google Inc. All Rights Reserved.
|
| +#
|
| +# Licensed under the Apache License, Version 2.0 (the "License");
|
| +# you may not use this file except in compliance with the License.
|
| +# You may obtain a copy of the License at
|
| +#
|
| +# http://www.apache.org/licenses/LICENSE-2.0
|
| +#
|
| +# Unless required by applicable law or agreed to in writing, software
|
| +# distributed under the License is distributed on an "AS IS" BASIS,
|
| +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
| +# See the License for the specific language governing permissions and
|
| +# limitations under the License.
|
| +
|
| +"""
|
| +Iterator wrapper that allows you to check whether the wrapped iterator
|
| +is empty and whether it has more than 1 element.
|
| +"""
|
| +
|
| +class PluralityCheckableIterator(object):
|
| +
|
| + def __init__(self, it):
|
| + self.it = it.__iter__()
|
| + self.head = []
|
| + # Populate first 2 elems into head so we can check whether iterator has
|
| + # more than 1 item.
|
| + for i in range(0, 2):
|
| + self.__populate_head__()
|
| +
|
| + def __populate_head__(self):
|
| + try:
|
| + e = self.it.next()
|
| + self.underlying_iter_empty = False
|
| + self.head.append(e)
|
| + except StopIteration:
|
| + # Indicates we can no longer call next() on underlying iterator, but
|
| + # there could still be elements left to iterate in head.
|
| + self.underlying_iter_empty = True
|
| +
|
| + def __iter__(self):
|
| + while len(self.head) > 0:
|
| + yield self.next()
|
| + else:
|
| + raise StopIteration()
|
| +
|
| + def next(self):
|
| + # Backfill into head each time we pop an element so we can always check
|
| + # for emptiness and for has_plurality().
|
| + self.__populate_head__()
|
| + return self.head.pop(0)
|
| +
|
| + def is_empty(self):
|
| + return len(self.head) == 0
|
| +
|
| + def has_plurality(self):
|
| + return len(self.head) > 1
|
|
|