trezor-firmware/core/src/apps/management/backup_types.py

from trezor.messages import BackupType

if False:
    from trezor.messages.ResetDevice import EnumTypeBackupType


_BIP39_WORD_COUNTS = (12, 18, 24)
_SLIP39_WORD_COUNTS = (20, 33)


def is_slip39_word_count(word_count: int) -> bool:
    """
    Returns True for SLIP-39 and False for BIP-39.
    Raise RuntimeError otherwise.
    """
    if word_count in _SLIP39_WORD_COUNTS:
        return True
    elif word_count in _BIP39_WORD_COUNTS:
        return False
    # Unknown word count.
    raise RuntimeError


def is_slip39_backup_type(backup_type: EnumTypeBackupType):
    return backup_type in (BackupType.Slip39_Basic, BackupType.Slip39_Advanced)