From 4b598ca6f91236c94da250282ac8c89c7d947ee2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Isma=C3=ABl=20Bouya?= <ismael.bouya@normalesup.org>
Date: Wed, 25 Jul 2018 20:29:08 +0200
Subject: Some fixes

- DB cursor expects an enumerable
- None should be returned when repartition is not available
---
 store.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'store.py')

diff --git a/store.py b/store.py
index 1a1ed76..5533759 100644
--- a/store.py
+++ b/store.py
@@ -564,7 +564,8 @@ class Portfolio:
             cls.retrieve_cryptoportfolio()
         cls.get_cryptoportfolio()
         liquidities = cls.liquidities.get(liquidity)
-        return liquidities[cls.last_date.get()].copy()
+        if liquidities is not None and cls.last_date.get() in liquidities:
+            return liquidities[cls.last_date.get()].copy()
 
     @classmethod
     def get_cryptoportfolio(cls, refetch=False):
-- 
cgit v1.2.3