#
# Copyright (c) 2008-2015 Thierry Florac <tflorac AT ulthar.net>
# All Rights Reserved.
#
# This software is subject to the provisions of the Zope Public License,
# Version 2.1 (ZPL). A copy of the ZPL should accompany this distribution.
# THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
# WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
# FOR A PARTICULAR PURPOSE.
#
__docformat__ = 'restructuredtext'
from hypatia.query import Query
from zope.intid.interfaces import IIntIds
from pyams_utils.registry import query_utility
[docs]class CatalogResultSet(object):
"""Catalog query result set wrapper"""
def __init__(self, query):
self.query = query
self.intids = query_utility(IIntIds)
self.first = []
self.last = []
def __iter__(self):
for item in self.first:
yield item
intids = self.intids
if intids is not None:
query = self.query
if isinstance(query, Query):
query = query.execute()
if isinstance(query, tuple):
query = query[1]
for oid in query:
if isinstance(oid, int):
target = intids.queryObject(oid)
if target is not None:
yield target
else:
yield oid
for item in self.last:
yield item
def __len__(self):
return len(self.first) + len(self.query) + len(self.last)
[docs] def prepend(self, items):
insert = self.first.insert
for index, item in enumerate(items):
insert(index, item)
[docs] def append(self, items):
append = self.last.append
for item in items:
append(item)
[docs]def or_(source, added):
"""Combine two queries with 'or'"""
if source is None:
source = added
else:
source |= added
return source
[docs]def and_(source, added):
"""Combine two queries with 'and'"""
if source is None:
source = added
else:
source &= added
return source