From a060c8c3c09b03c95968be0c1724857242c49077 Mon Sep 17 00:00:00 2001
From: Vincent Ardisson <vardisson@budget-insight.com>
Date: Fri, 6 Mar 2020 10:47:31 +0100
Subject: [PATCH] [nalo] fill Investment.asset_category

---
 modules/nalo/browser.py | 1 +
 modules/nalo/pages.py   | 9 +++++++++
 2 files changed, 10 insertions(+)

diff --git a/modules/nalo/browser.py b/modules/nalo/browser.py
index 96212c8869..7b417bee6f 100644
--- a/modules/nalo/browser.py
+++ b/modules/nalo/browser.py
@@ -74,4 +74,5 @@ def iter_investment(self, account):
             inv.label = item['name']
             inv.portfolio_share = item['share']
             inv.valuation = account.balance * inv.portfolio_share
+            inv.asset_category = item['asset_type']
             yield inv
diff --git a/modules/nalo/pages.py b/modules/nalo/pages.py
index 77ee206a87..000f8951ae 100644
--- a/modules/nalo/pages.py
+++ b/modules/nalo/pages.py
@@ -70,6 +70,14 @@ class HistoryPage(LoggedPage, JsonPage):
     pass
 
 
+# using site labels
+ASSET_TYPE = {
+    'risky': 'Actions',
+    'risk_free': 'Obligations',
+    'guaranteed': 'Fonds à capital garanti',
+}
+
+
 class InvestPage(LoggedPage, JsonPage):
     ENCODING = 'utf-8'
 
@@ -91,4 +99,5 @@ def get_invest(self, kind, risk):
                 'isin': line['isin'],
                 'name': line['name'],
                 'share': float_to_decimal(line['weight']) / 100,
+                'asset_type': ASSET_TYPE[line['risk_type']],
             }
-- 
GitLab