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

import storage.device
from storage.device import set_flags
from trezor import wire
from trezor.messages.Success import Success


async def apply_flags(ctx, msg):
    if not storage.device.is_initialized():
        raise wire.NotInitialized("Device is not initialized")
    set_flags(msg.flags)
    return Success(message="Flags applied")