common/registerers.py
author sylvain.thenault@logilab.fr
Tue, 17 Feb 2009 16:41:46 +0100
branchtls-sprint
changeset 664 04154a2b238d
parent 631 99f5852f8604
child 670 6c332f5c969c
permissions -rw-r--r--
drop unused support of .required, no more necessary .accepts check, and extresources_registerer

"""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-2009 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
        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
    

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