from electrum.plugins import hook
from .trezor import TrezorPlugin
from ..hw_wallet import CmdLineHandler

class Plugin(TrezorPlugin):
    handler = CmdLineHandler()
    @hook
    def init_keystore(self, keystore):
        if not isinstance(keystore, self.keystore_class):
            return
        keystore.handler = self.handler

    def create_handler(self, window):
        return self.handler