CrudBuilderRegistry

Complexity

Total Complexity

13

Size/Duplication

Total Lines	68
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	1

from django.db import models
class CrudBuilderRegistry(dict):
    """Dictionary like object representing a collection of objects."""

    @classmethod
    def extract_args(cls, *args):
        """
        Takes any arguments like a model and crud, or just one of
        those, in any order, and return a model and crud.
        """
        model = None
        crudbuilder = None

        for arg in args:
            if issubclass(arg, models.Model):
                model = arg
            else:
                crudbuilder = arg

        return [model, crudbuilder]

    def register(self, *args, **kwargs):
        """
        Register a crud.

        Two unordered arguments are accepted, at least one should be passed:
        - a model,
        - a crudbuilder class
        """

        assert len(args) <= 2, 'register takes at most 2 args'
        assert len(args) > 0, 'register takes at least 1 arg'

        model, crudbuilder = self.__class__.extract_args(*args)

        if not issubclass(model, models.Model):
            msg = "First argument should be Django Model"
            raise NotModelException(msg)

        key = self._model_key(model)

        if key in self:
            msg = "Key '{key}' has already been registered.".format(
                key=key
                )
            raise AlreadyRegistered(msg)

        self.__setitem__(key, crudbuilder)
        return crudbuilder

    def _model_key(self, model):
        app_label = model._meta.app_label
        model_name = model.__name__.lower()
        return '{}-{}'.format(app_label, model_name)

    def unregister(self, model):
        key = self._model_key(model)
        if key in self:
            self.__delitem__(key)

    def __getitem__(self, name):
        """
        Return the CrudBuilder class registered for this name. If none is
        registered, raise NotRegistered.
        """
        try:
            return super(CrudBuilderRegistry, self).__getitem__(name)
        except KeyError:
            raise NotRegistered(name, self)