"""
Public transport routines
"""

import sys
from Bcfg2.Reporting.Transport.base import TransportError, \
    TransportImportError


def load_transport(transport_name, setup):
    """
    Try to load the transport.  Raise TransportImportError on failure
    """
    try:
        mod_name = "%s.%s" % (__name__, transport_name)
        mod = getattr(__import__(mod_name).Reporting.Transport, transport_name)
    except ImportError:
        try:
            mod = __import__(transport_name)
        except:
            raise TransportImportError("Error importing transport %s: %s" %
                                       (transport_name, sys.exc_info()[1]))
    try:
        return getattr(mod, transport_name)(setup)
    except:
        raise TransportImportError("Error instantiating transport %s: %s" %
                                   (transport_name, sys.exc_info()[1]))


def load_transport_from_config(setup):
    """Load the transport in the config... eventually"""
    try:
        return load_transport(setup['reporting_transport'], setup)
    except KeyError:
        raise TransportImportError('Transport missing in config')