src.registration.supplements.get_supplement_class()

Complexity

Conditions

6

Size

Total Lines

34

Duplication

Lines	0
Ratio	0 %

# -*- coding: utf-8 -*-
def get_supplement_class(path=None):
    """
    Return an instance of a registration supplement, given the dotted
    Python import path (as a string) to the supplement class.

    If the addition cannot be located (e.g., because no such module
    exists, or because the module does not contain a class of the
    appropriate name), ``django.core.exceptions.ImproperlyConfigured``
    is raised.
   
    """
    from registration.conf import settings
    path = path or settings.REGISTRATION_SUPPLEMENT_CLASS
    if not path:
        return None
    i = path.rfind('.')
    module, attr = path[:i], path[i+1:]
    try:
        mod = import_module(module)
    except ImportError as e:
        raise ImproperlyConfigured(
                'Error loading registration addition %s: "%s"' % (module, e))
    try:
        cls = getattr(mod, attr)
    except AttributeError:
        raise ImproperlyConfigured((
                'Module "%s" does not define a registration supplement named '
                '"%s"') % (module, attr))
    from registration.supplements.base import RegistrationSupplementBase
    if cls and not issubclass(cls, RegistrationSupplementBase):
        raise ImproperlyConfigured((
            'Registration supplement class "%s" must be a subclass of '
            '``registration.supplements.RegistrationSupplementBase``') % path)
    return cls