Source code for faust.fixups

"""Transport registry."""

from typing import Iterator, Type

from mode.utils.imports import FactoryMapping

from faust.types import AppT, FixupT

__all__ = ["by_name", "by_url", "fixups"]

FIXUPS: FactoryMapping[Type[FixupT]] = FactoryMapping(
    django="faust.fixups.django:Fixup",
)
FIXUPS.include_setuptools_namespace("faust.fixups")
by_name = FIXUPS.by_name
by_url = FIXUPS.by_url


def fixups(app: AppT) -> Iterator[FixupT]:
    """Iterate over enabled fixups.

    Fixups are installed by setuptools, using the 'faust.fixups'
    namespace.

    Fixups modify the Faust library to work with frameworks
    such as Django.
    """
    for Fixup in FIXUPS.iterate():
        fixup = Fixup(app)
        if fixup.enabled():
            yield fixup