From 88f61581d924138b0926a5459a634459c517cf68 Mon Sep 17 00:00:00 2001
From: misagh <misagh.shakeri@gmail.com>
Date: Wed, 21 Nov 2018 19:47:51 +0100
Subject: [PATCH] 1) NamedTuple refactored 2) Missing data handled

---
 freqtrade/wallets.py | 26 ++++++++++++++++----------
 1 file changed, 16 insertions(+), 10 deletions(-)

diff --git a/freqtrade/wallets.py b/freqtrade/wallets.py
index b4b89bc59..5ee33f41e 100644
--- a/freqtrade/wallets.py
+++ b/freqtrade/wallets.py
@@ -1,13 +1,21 @@
 # pragma pylint: disable=W0603
 """ Wallet """
 import logging
-from typing import Dict, Any
+from typing import Dict, Any, NamedTuple
 from collections import namedtuple
 from freqtrade.exchange import Exchange
 
 logger = logging.getLogger(__name__)
 
 
+class Wallet(NamedTuple):
+    exchange: str = None
+    currency: str = None
+    free: float = 0
+    used: float = 0
+    total: float = 0
+
+
 class Wallets(object):
 
     # wallet data structure
@@ -25,14 +33,12 @@ class Wallets(object):
         balances = self.exchange.get_balances()
 
         for currency in balances:
-            info = {
-                'exchange': self.exchange.id,
-                'currency': currency,
-                'free': balances[currency]['free'],
-                'used': balances[currency]['used'],
-                'total': balances[currency]['total']
-            }
-
-            self.wallets[currency] = self.wallet(**info)
+            self.wallets[currency] = Wallet(
+                self.exchange.id,
+                currency,
+                balances[currency].get('free', None),
+                balances[currency].get('used', None),
+                balances[currency].get('total', None)
+            )
 
         logger.info('Wallets synced ...')