fix cmdline trezor

plugins/trezor/cmdline.py

@@ -0,0 +1,36 @@
+from trezor import TrezorPlugin
+from electrum.util import print_msg
+from electrum.plugins import hook
+
+class TrezorCmdLineHandler:
+
+    def get_passphrase(self, msg):
+        import getpass
+        print_msg(msg)
+        return getpass.getpass('')
+
+    def get_pin(self, msg):
+        t = { 'a':'7', 'b':'8', 'c':'9', 'd':'4', 'e':'5', 'f':'6', 'g':'1', 'h':'2', 'i':'3'}
+        print_msg(msg)
+        print_msg("a b c\nd e f\ng h i\n-----")
+        o = raw_input()
+        return ''.join(map(lambda x: t[x], o))
+
+    def stop(self):
+        pass
+
+    def show_message(self, msg):
+        print_msg(msg)
+
+
+class Plugin(TrezorPlugin):
+
+    @hook
+    def cmdline_load_wallet(self, wallet):
+        print "cmdline load wallet"
+        self.wallet = wallet
+        self.wallet.plugin = self
+        if self.handler is None:
+            self.handler = TrezorCmdLineHandler()
+        print self.wallet
+

plugins/trezor/trezor.py

@@ -273,13 +273,6 @@ class TrezorPlugin(BasePlugin):
             self.client = None
         self.wallet = None
 
-    @hook
-    def cmdline_load_wallet(self, wallet):
-        self.wallet = wallet
-        self.wallet.plugin = self
-        if self.handler is None:
-            self.handler = TrezorCmdLineHandler()
-
     def sign_transaction(self, tx, prev_tx, xpub_path):
         self.prev_tx = prev_tx
         self.xpub_path = xpub_path
@@ -451,27 +444,6 @@ class TrezorGuiMixin(object):
         return proto.WordAck(word=word)
 
 
-class TrezorCmdLineHandler:
-
-    def get_passphrase(self, msg):
-        import getpass
-        print_msg(msg)
-        return getpass.getpass('')
-
-    def get_pin(self, msg):
-        t = { 'a':'7', 'b':'8', 'c':'9', 'd':'4', 'e':'5', 'f':'6', 'g':'1', 'h':'2', 'i':'3'}
-        print_msg(msg)
-        print_msg("a b c\nd e f\ng h i\n-----")
-        o = raw_input()
-        return ''.join(map(lambda x: t[x], o))
-
-    def stop(self):
-        pass
-
-    def show_message(self, msg):
-        print_msg(msg)
-
-
 
 
 if TREZOR: