From d740895bd68f0ded97e382e5944feb26fa759875 Mon Sep 17 00:00:00 2001
From: Craig Raw <craig@quirk.biz>
Date: Wed, 29 Apr 2020 15:04:09 +0200
Subject: [PATCH] ability to display type of wallet model

---
 .../java/com/sparrowwallet/drongo/wallet/WalletModel.java | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/main/java/com/sparrowwallet/drongo/wallet/WalletModel.java b/src/main/java/com/sparrowwallet/drongo/wallet/WalletModel.java
index 0517149..3ecfb83 100644
--- a/src/main/java/com/sparrowwallet/drongo/wallet/WalletModel.java
+++ b/src/main/java/com/sparrowwallet/drongo/wallet/WalletModel.java
@@ -7,6 +7,14 @@ public enum WalletModel {
         return valueOf(model.toUpperCase());
     }
 
+    public String getType() {
+        if(this == TREZOR_1 || this == TREZOR_T) {
+            return "trezor";
+        }
+
+        return this.toString();
+    }
+
     public String toDisplayString() {
         String line = this.toString().toLowerCase();
         String[] words = line.split("_");