Source code for faust.transport.drivers

"""Transport registry."""

from yarl import URL

from .aiokafka import Transport as AIOKafkaTransport

__all__ = ["by_name", "by_url"]


DRIVERS = {
    "aiokafka": AIOKafkaTransport,
    "kafka": AIOKafkaTransport,
}



[docs]def by_name(driver_name: str):
    return DRIVERS[driver_name]




[docs]def by_url(url: URL):
    scheme = url.scheme
    return DRIVERS[scheme]