From f65ee537a4754cb7b4518482db05cac7f40a3f7b Mon Sep 17 00:00:00 2001
From: Maxime Pommier <maxime.pommier@budget-insight.com>
Date: Thu, 9 May 2019 18:02:13 +0200
Subject: [PATCH] [societegenerale] Add account usage

---
 modules/societegenerale/browser.py      | 6 +++++-
 modules/societegenerale/sgpe/browser.py | 3 ++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/modules/societegenerale/browser.py b/modules/societegenerale/browser.py
index a906def3b8..abbe51efb2 100644
--- a/modules/societegenerale/browser.py
+++ b/modules/societegenerale/browser.py
@@ -25,7 +25,7 @@
 
 from weboob.browser import LoginBrowser, URL, need_login, StatesMixin
 from weboob.exceptions import BrowserIncorrectPassword, ActionNeeded, BrowserUnavailable
-from weboob.capabilities.bank import Account, TransferBankError, AddRecipientStep, TransactionType
+from weboob.capabilities.bank import Account, TransferBankError, AddRecipientStep, TransactionType, AccountOwnerType
 from weboob.capabilities.base import find_object, NotAvailable
 from weboob.browser.exceptions import BrowserHTTPNotFound, ClientError
 from weboob.capabilities.profile import ProfileMissing
@@ -164,6 +164,7 @@ def iter_cards(self, account):
                 card.currency = account.currency
                 card._internal_id = el['idTechnique']
                 card._prestation_id = el['id']
+                card.owner_type = AccountOwnerType.PRIVATE
                 yield card
 
     @need_login
@@ -202,8 +203,11 @@ def get_accounts_list(self):
             for card in self.iter_cards(account):
                 card.parent = account
                 card.ownership = account.ownership
+                card.owner_type = AccountOwnerType.PRIVATE
                 yield card
 
+            account.owner_type = AccountOwnerType.PRIVATE
+
             if account._prestation_id in account_ibans:
                 account.iban = account_ibans[account._prestation_id]
 
diff --git a/modules/societegenerale/sgpe/browser.py b/modules/societegenerale/sgpe/browser.py
index 9ab2a40b95..ab1fec866d 100644
--- a/modules/societegenerale/sgpe/browser.py
+++ b/modules/societegenerale/sgpe/browser.py
@@ -29,7 +29,7 @@
 from weboob.capabilities.base import find_object
 from weboob.capabilities.bank import (
     AccountNotFound, RecipientNotFound, AddRecipientStep, AddRecipientBankError,
-    Recipient, TransferBankError,
+    Recipient, TransferBankError, AccountOwnerType,
 )
 from weboob.tools.value import Value
 
@@ -174,6 +174,7 @@ def get_accounts_list(self):
             self.intraday_balances.go()
 
         for acc in self.page.populate_balances(accounts):
+            acc.owner_type = AccountOwnerType.ORGANIZATION
             yield acc
 
     @need_login
-- 
GitLab