File: CatalogTool.py

package info (click to toggle)
zope-cmfmember 1%3A1.1b2-2
  • links: PTS
  • area: main
  • in suites: etch, etch-m68k
  • size: 1,316 kB
  • ctags: 1,126
  • sloc: python: 6,558; makefile: 109; sh: 8
file content (51 lines) | stat: -rw-r--r-- 1,614 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
from Products.CMFCore.interfaces.portal_catalog \
        import portal_catalog as ICatalogTool
from Products.CMFCore.ActionProviderBase import ActionProviderBase
from Products.CMFCore.ActionProviderBase import ActionProviderBase

from Products.CMFPlone.CatalogTool import CatalogTool as BaseTool
from Products.CMFCore.CatalogTool import IndexableObjectWrapper
from Products.ZCatalog.ZCatalog import ZCatalog


class CatalogTool(BaseTool):
    """Override CatalogTool so we can add some extra methods to all objects
    before cataloging them."""
    __implements__ = (ICatalogTool, ActionProviderBase.__implements__)

    meta_type = 'Portal CMFMember Catalog Tool'
    security = ClassSecurityInfo()

    def catalog_object(self, object, uid, idxs=[],**kwargs):
        try:
            usersWithLocalRoles = dict(object.get_local_roles()).keys()
        except ConflictError:
            raise
        except:
            usersWithLocalRoles = []

        try:
            owner_info = object.owner_info()
            owner = '/' + owner_info['path'] + '/' + owner_info['id']
        except ConflictError:
            raise
        except:
            owner = None

        wf = getattr(self, 'portal_workflow', None)
        if wf is not None:
            vars = wf.getCatalogVariablesFor(object)
        else:
            vars = {}

        vars['indexedOwner'] = owner
        vars['indexedUsersWithLocalRoles'] = usersWithLocalRoles

        w = IndexableObjectWrapper(vars, object)

        ZCatalog.catalog_object(self, w, uid, idxs)


CatalogTool.__doc__ = BaseTool.__doc__

InitializeClass(CatalogTool)