|
|
|
"""Collects all file transmitters.""" |
|
|
|
from .local_file_transmitter import LocalFileTransmitter |
|
from .dummy_file_transmitter import DummyFileTransmitter |
|
|
|
__all__ = ['build_file_transmitter'] |
|
|
|
_TRANSMITTERS = { |
|
'local': LocalFileTransmitter, |
|
'dummy': DummyFileTransmitter, |
|
} |
|
|
|
|
|
def build_file_transmitter(transmitter_type='local', **kwargs): |
|
"""Builds a file transmitter. |
|
|
|
Args: |
|
transmitter_type: Type of the file transmitter_type, which is case |
|
insensitive. (default: `normal`) |
|
**kwargs: Additional arguments to build the file transmitter. |
|
|
|
Raises: |
|
ValueError: If the `transmitter_type` is not supported. |
|
""" |
|
transmitter_type = transmitter_type.lower() |
|
if transmitter_type not in _TRANSMITTERS: |
|
raise ValueError(f'Invalid transmitter type: `{transmitter_type}`!\n' |
|
f'Types allowed: {list(_TRANSMITTERS)}.') |
|
return _TRANSMITTERS[transmitter_type](**kwargs) |
|
|
