From 919a8bb5388d6e7060f65ebbea72bdf84c0496c3 Mon Sep 17 00:00:00 2001 From: Sylvie Ye Date: Fri, 19 Apr 2019 17:53:23 +0200 Subject: [PATCH] [cragr] filter able to do transfer accounts In transfer initiation, get only able to do transfer accounts --- modules/cragr/api/browser.py | 4 +++- modules/cragr/api/transfer_pages.py | 5 +++++ 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/modules/cragr/api/browser.py b/modules/cragr/api/browser.py index cd76d41361..8fd48e72df 100644 --- a/modules/cragr/api/browser.py +++ b/modules/cragr/api/browser.py @@ -592,7 +592,9 @@ def iter_debit_accounts(self): assert self.recipients.is_here() for index, debit_accounts in enumerate(self.page.iter_debit_accounts()): debit_accounts._index = index - yield debit_accounts + if self.page.is_sender_account(debit_accounts.id): + # only yield able to do transfer accounts + yield debit_accounts @need_login def iter_transfer_recipients(self, account, transfer_space_info=None): diff --git a/modules/cragr/api/transfer_pages.py b/modules/cragr/api/transfer_pages.py index f34da373fb..565324dff4 100644 --- a/modules/cragr/api/transfer_pages.py +++ b/modules/cragr/api/transfer_pages.py @@ -40,6 +40,11 @@ def is_sender_account(self, account_id): @method class iter_debit_accounts(DictElement): + def store(self, obj): + # can have accounts with same ID + # filter it on `browser.py` to have 'index' (needed to do transfer) + return obj + class item(ItemElement): def condition(self): return Dict('accountNumber', default=None)(self) -- GitLab