Mercurial > repos > guerler > springsuite
comparison planemo/lib/python3.7/site-packages/boto/sdb/queryresultset.py @ 0:d30785e31577 draft
"planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"
| author | guerler |
|---|---|
| date | Fri, 31 Jul 2020 00:18:57 -0400 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| -1:000000000000 | 0:d30785e31577 |
|---|---|
| 1 from boto.compat import six | |
| 2 # Copyright (c) 2006,2007 Mitch Garnaat http://garnaat.org/ | |
| 3 # | |
| 4 # Permission is hereby granted, free of charge, to any person obtaining a | |
| 5 # copy of this software and associated documentation files (the | |
| 6 # "Software"), to deal in the Software without restriction, including | |
| 7 # without limitation the rights to use, copy, modify, merge, publish, dis- | |
| 8 # tribute, sublicense, and/or sell copies of the Software, and to permit | |
| 9 # persons to whom the Software is furnished to do so, subject to the fol- | |
| 10 # lowing conditions: | |
| 11 # | |
| 12 # The above copyright notice and this permission notice shall be included | |
| 13 # in all copies or substantial portions of the Software. | |
| 14 # | |
| 15 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS | |
| 16 # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL- | |
| 17 # ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT | |
| 18 # SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, | |
| 19 # WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
| 20 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS | |
| 21 # IN THE SOFTWARE. | |
| 22 | |
| 23 def query_lister(domain, query='', max_items=None, attr_names=None): | |
| 24 more_results = True | |
| 25 num_results = 0 | |
| 26 next_token = None | |
| 27 while more_results: | |
| 28 rs = domain.connection.query_with_attributes(domain, query, attr_names, | |
| 29 next_token=next_token) | |
| 30 for item in rs: | |
| 31 if max_items: | |
| 32 if num_results == max_items: | |
| 33 raise StopIteration | |
| 34 yield item | |
| 35 num_results += 1 | |
| 36 next_token = rs.next_token | |
| 37 more_results = next_token is not None | |
| 38 | |
| 39 class QueryResultSet(object): | |
| 40 | |
| 41 def __init__(self, domain=None, query='', max_items=None, attr_names=None): | |
| 42 self.max_items = max_items | |
| 43 self.domain = domain | |
| 44 self.query = query | |
| 45 self.attr_names = attr_names | |
| 46 | |
| 47 def __iter__(self): | |
| 48 return query_lister(self.domain, self.query, self.max_items, self.attr_names) | |
| 49 | |
| 50 def select_lister(domain, query='', max_items=None): | |
| 51 more_results = True | |
| 52 num_results = 0 | |
| 53 next_token = None | |
| 54 while more_results: | |
| 55 rs = domain.connection.select(domain, query, next_token=next_token) | |
| 56 for item in rs: | |
| 57 if max_items: | |
| 58 if num_results == max_items: | |
| 59 raise StopIteration | |
| 60 yield item | |
| 61 num_results += 1 | |
| 62 next_token = rs.next_token | |
| 63 more_results = next_token is not None | |
| 64 | |
| 65 class SelectResultSet(object): | |
| 66 | |
| 67 def __init__(self, domain=None, query='', max_items=None, | |
| 68 next_token=None, consistent_read=False): | |
| 69 self.domain = domain | |
| 70 self.query = query | |
| 71 self.consistent_read = consistent_read | |
| 72 self.max_items = max_items | |
| 73 self.next_token = next_token | |
| 74 | |
| 75 def __iter__(self): | |
| 76 more_results = True | |
| 77 num_results = 0 | |
| 78 while more_results: | |
| 79 rs = self.domain.connection.select(self.domain, self.query, | |
| 80 next_token=self.next_token, | |
| 81 consistent_read=self.consistent_read) | |
| 82 for item in rs: | |
| 83 if self.max_items and num_results >= self.max_items: | |
| 84 raise StopIteration | |
| 85 yield item | |
| 86 num_results += 1 | |
| 87 self.next_token = rs.next_token | |
| 88 if self.max_items and num_results >= self.max_items: | |
| 89 raise StopIteration | |
| 90 more_results = self.next_token is not None | |
| 91 | |
| 92 def next(self): | |
| 93 return next(self.__iter__()) |
