"""设备传输模块"""

from kobo_manga.config import TransferConfig
from kobo_manga.transfer.calibre import CalibreTransfer
from kobo_manga.transfer.usb import USBTransfer

__all__ = ["USBTransfer", "CalibreTransfer", "get_transfer"]


def get_transfer(config: TransferConfig) -> USBTransfer | CalibreTransfer:
    """根据配置返回对应的传输器。"""
    if config.method == "calibre":
        return CalibreTransfer()
    return USBTransfer()