common/registerers.py
author Laure Bourgois <Laure.Bourgois@logilab.fr>
Mon, 01 Dec 2008 16:21:26 +0100
changeset 162 5012aba69e62
parent 0 b97547f5f1fa
child 631 99f5852f8604
child 1252 e782f333408b
permissions -rw-r--r--
repairing error from preceding commit

"""This file contains some basic registerers required by application objects
registry to handle registration at startup time.

A registerer is responsible to tell if an object should be registered according
to the application's schema or to already registered object

:organization: Logilab
:copyright: 2006-2008 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
:contact: http://www.logilab.fr/ -- mailto:contact@logilab.fr
"""
__docformat__ = "restructuredtext en"

from cubicweb.vregistry import registerer


def _accepts_interfaces(obj):
    return sorted(getattr(obj, 'accepts_interfaces', ()))


class yes_registerer(registerer):
    """register without any other action"""
    def do_it_yourself(self, registered):
        return self.vobject

class priority_registerer(registerer):
    """systematically kick previous registered class and register the
    wrapped class (based on the fact that directory containing vobjects
    are loaded from the most generic to the most specific).

    This is usually for templates or startup views where we want to
    keep only the latest in the load path
    """
    def do_it_yourself(self, registered):
        if registered:
            if len(registered) > 1:
                self.warning('priority_registerer found more than one registered objects '
                             '(registerer monkey patch ?)')
            for regobj in registered[:]:
                self.kick(registered, regobj)
        return self.vobject
    
    def remove_equivalents(self, registered):
        for _obj in registered[:]:
            if self.equivalent(_obj):
                self.kick(registered, _obj)
                break
            
    def remove_all_equivalents(self, registered):
        for _obj in registered[:]:
            if _obj is self.vobject:
                continue
            if self.equivalent(_obj):
                self.kick(registered, _obj)

    def equivalent(self, other):
        raise NotImplementedError(self, self.vobject)


class kick_registerer(registerer):
    """systematically kick previous registered class and don't register the
    wrapped class. This is temporarily used to discard library object registrable
    but that we don't want to use
    """
    def do_it_yourself(self, registered):
        if registered:
            self.kick(registered, registered[-1])
        return 
    

class accepts_registerer(priority_registerer):
    """register according to the .accepts attribute of the wrapped
    class, which should be a tuple refering some entity's types

    * if no type is defined the application'schema, skip the wrapped
      class
    * if the class defines a requires attribute, each entity type defined
      in the requires list must be in the schema
    * if an object previously registered has equivalent .accepts
      attribute, kick it out
    * register
    """
    def do_it_yourself(self, registered):
        # if object is accepting interface, we have register it now and
        # remove it latter if no object is implementing accepted interfaces
        if _accepts_interfaces(self.vobject):
            return self.vobject
        if not 'Any' in self.vobject.accepts:
            for ertype in self.vobject.accepts:
                if ertype in self.schema:
                    break
            else:
                self.skip()
                return None
        for required in getattr(self.vobject, 'requires', ()):
            if required not in self.schema:
                self.skip()
                return
        self.remove_equivalents(registered)
        return self.vobject
    
    def equivalent(self, other):
        if _accepts_interfaces(self.vobject) != _accepts_interfaces(other):
            return False
        try:
            newaccepts = list(other.accepts)
            for etype in self.vobject.accepts:
                try:
                    newaccepts.remove(etype)
                except ValueError:
                    continue
            if newaccepts:
                other.accepts = tuple(newaccepts)
                return False
            return True
        except AttributeError:
            return False


class id_registerer(priority_registerer):
    """register according to the "id" attribute of the wrapped class,
    refering to an entity type.
    
    * if the type is not Any and is not defined the application'schema,
      skip the wrapped class
    * if an object previously registered has the same .id attribute,
      kick it out
    * register
    """
    def do_it_yourself(self, registered):
        etype = self.vobject.id
        if etype != 'Any' and not self.schema.has_entity(etype):
            self.skip()
            return
        self.remove_equivalents(registered)
        return self.vobject
    
    def equivalent(self, other):
        return other.id == self.vobject.id


class etype_rtype_registerer(registerer):
    """registerer handling optional .etype and .rtype attributes.:
    
    * if .etype is set and is not an entity type defined in the
      application schema, skip the wrapped class
    * if .rtype or .relname is set and is not a relation type defined in
      the application schema, skip the wrapped class
    * register
    """
    def do_it_yourself(self, registered):
        cls = self.vobject
        if hasattr(cls, 'etype'):
            if not self.schema.has_entity(cls.etype):
                return
        rtype = getattr(cls, 'rtype', None)
        if rtype and not self.schema.has_relation(rtype):
            return
        return cls

class etype_rtype_priority_registerer(etype_rtype_registerer):
    """add priority behaviour to the etype_rtype_registerer
    """
    def do_it_yourself(self, registered):
        cls = super(etype_rtype_priority_registerer, self).do_it_yourself(registered)
        if cls:
            registerer = priority_registerer(self.registry, cls)
            cls = registerer.do_it_yourself(registered)
        return cls

class action_registerer(etype_rtype_registerer):
    """'all in one' actions registerer, handling optional .accepts,
    .etype and .rtype attributes:
    
    * if .etype is set and is not an entity type defined in the
      application schema, skip the wrapped class
    * if .rtype or .relname is set and is not a relation type defined in
      the application schema, skip the wrapped class
    * if .accepts is set, delegate to the accepts_registerer
    * register
    """
    def do_it_yourself(self, registered):
        cls = super(action_registerer, self).do_it_yourself(registered)
        if hasattr(cls, 'accepts'):
            registerer = accepts_registerer(self.registry, cls)
            cls = registerer.do_it_yourself(registered)
        return cls


class extresources_registerer(priority_registerer):
    """'registerer according to a .need_resources attributes which
    should list necessary resource identifiers for the wrapped object.
    If one of its resources is missing, don't register
    """
    def do_it_yourself(self, registered):
        if not hasattr(self.config, 'has_resource'):
            return
        for resourceid in self.vobject.need_resources:
            if not self.config.has_resource(resourceid):
                return
        return super(extresources_registerer, self).do_it_yourself(registered)
    

__all__ = [cls.__name__ for cls in globals().values()
           if isinstance(cls, type) and issubclass(cls, registerer)
           and not cls is registerer]