| Index: third_party/gsutil/boto/sdb/queryresultset.py
|
| diff --git a/third_party/gsutil/boto/sdb/queryresultset.py b/third_party/gsutil/boto/sdb/queryresultset.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..10bafd1c92c9fe4319e9f6e9be69b3832bf1e84a
|
| --- /dev/null
|
| +++ b/third_party/gsutil/boto/sdb/queryresultset.py
|
| @@ -0,0 +1,92 @@
|
| +# Copyright (c) 2006,2007 Mitch Garnaat http://garnaat.org/
|
| +#
|
| +# Permission is hereby granted, free of charge, to any person obtaining a
|
| +# copy of this software and associated documentation files (the
|
| +# "Software"), to deal in the Software without restriction, including
|
| +# without limitation the rights to use, copy, modify, merge, publish, dis-
|
| +# tribute, sublicense, and/or sell copies of the Software, and to permit
|
| +# persons to whom the Software is furnished to do so, subject to the fol-
|
| +# lowing conditions:
|
| +#
|
| +# The above copyright notice and this permission notice shall be included
|
| +# in all copies or substantial portions of the Software.
|
| +#
|
| +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
| +# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
|
| +# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
|
| +# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
|
| +# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
| +# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
| +# IN THE SOFTWARE.
|
| +
|
| +def query_lister(domain, query='', max_items=None, attr_names=None):
|
| + more_results = True
|
| + num_results = 0
|
| + next_token = None
|
| + while more_results:
|
| + rs = domain.connection.query_with_attributes(domain, query, attr_names,
|
| + next_token=next_token)
|
| + for item in rs:
|
| + if max_items:
|
| + if num_results == max_items:
|
| + raise StopIteration
|
| + yield item
|
| + num_results += 1
|
| + next_token = rs.next_token
|
| + more_results = next_token != None
|
| +
|
| +class QueryResultSet:
|
| +
|
| + def __init__(self, domain=None, query='', max_items=None, attr_names=None):
|
| + self.max_items = max_items
|
| + self.domain = domain
|
| + self.query = query
|
| + self.attr_names = attr_names
|
| +
|
| + def __iter__(self):
|
| + return query_lister(self.domain, self.query, self.max_items, self.attr_names)
|
| +
|
| +def select_lister(domain, query='', max_items=None):
|
| + more_results = True
|
| + num_results = 0
|
| + next_token = None
|
| + while more_results:
|
| + rs = domain.connection.select(domain, query, next_token=next_token)
|
| + for item in rs:
|
| + if max_items:
|
| + if num_results == max_items:
|
| + raise StopIteration
|
| + yield item
|
| + num_results += 1
|
| + next_token = rs.next_token
|
| + more_results = next_token != None
|
| +
|
| +class SelectResultSet(object):
|
| +
|
| + def __init__(self, domain=None, query='', max_items=None,
|
| + next_token=None, consistent_read=False):
|
| + self.domain = domain
|
| + self.query = query
|
| + self.consistent_read = consistent_read
|
| + self.max_items = max_items
|
| + self.next_token = next_token
|
| +
|
| + def __iter__(self):
|
| + more_results = True
|
| + num_results = 0
|
| + while more_results:
|
| + rs = self.domain.connection.select(self.domain, self.query,
|
| + next_token=self.next_token,
|
| + consistent_read=self.consistent_read)
|
| + for item in rs:
|
| + if self.max_items and num_results >= self.max_items:
|
| + raise StopIteration
|
| + yield item
|
| + num_results += 1
|
| + self.next_token = rs.next_token
|
| + if self.max_items and num_results >= self.max_items:
|
| + raise StopIteration
|
| + more_results = self.next_token != None
|
| +
|
| + def next(self):
|
| + return self.__iter__().next()
|
|
|