diff --git a/modules/cmso/par/browser.py b/modules/cmso/par/browser.py index 2c987210738cb0411cd04f8391bfe36e39a3d2f6..7e5583c22201618b86c1aabd7853ef96102f1dee 100644 --- a/modules/cmso/par/browser.py +++ b/modules/cmso/par/browser.py @@ -175,8 +175,7 @@ def iter_accounts(self): self.page.check_response() for key in self.page.get_keys(): for a in self.page.iter_accounts(key=key): - if a.id in accounts_eligibilite_debit: - a._eligible_debit = accounts_eligibilite_debit[a.id] + a._eligible_debit = accounts_eligibilite_debit.get(a.id, False) # Can have duplicate account, avoid them if a._index not in seen: self.accounts_list.append(a) @@ -187,8 +186,7 @@ def iter_accounts(self): page = self.accounts.go(data=json.dumps({}), type='epargne', headers=self.json_headers) for key in page.get_keys(): for a in page.iter_savings(key=key, numbers=numbers): - if a.id in accounts_eligibilite_debit: - a._eligible_debit = accounts_eligibilite_debit[a.id] + a._eligible_debit = accounts_eligibilite_debit.get(a.id, False) if a._index in seen: acc = seen[a._index] self.accounts_list.remove(acc) diff --git a/modules/cmso/par/transfer_pages.py b/modules/cmso/par/transfer_pages.py index fdfd9fcfbee0b856f86a9f5bb09389fd1449d0ea..d9f567636ab45a02c09bce8c6b6ff89dbd0b3388 100644 --- a/modules/cmso/par/transfer_pages.py +++ b/modules/cmso/par/transfer_pages.py @@ -72,23 +72,26 @@ def get_transfer_info(self, info): ret = {} ret.update({ - d[key]: d[value] + d[key]: d.get(value) for d in self.doc['listCompteTitulaireCotitulaire'] }) ret.update({ - d[key]: d[value] + d[key]: d.get(value) for p in self.doc['listCompteMandataire'].values() for d in p }) ret.update({ - d[key]: d[value] + d[key]: d.get(value) for p in self.doc['listCompteLegalRep'].values() for d in p }) return ret def get_numbers(self): - return self.get_transfer_info('numbers') + transfer_numbers = self.get_transfer_info('numbers') + for key, value in transfer_numbers.items(): + assert value, "The 'numeroContratSouscrit' associated with the account index: %s is empty" % key + return transfer_numbers def get_eligibilite_debit(self): return self.get_transfer_info('eligibilite_debit')