# -*- coding: utf-8 -*-

# Copyright(C) 2009-2011  Romain Bignon, Florent Fourcot
#
# This file is part of weboob.
#
# weboob is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# weboob is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with weboob. If not, see <http://www.gnu.org/licenses/>.


from weboob.capabilities.bank import Recipient, AccountNotFound
from weboob.tools.browser import BasePage
from weboob.tools.mech import ClientForm


__all__ = ['TransferPage']


class TransferPage(BasePage):
    def on_loaded(self):
        pass

    def get_recipients(self):
        # First, internals recipients
        table = self.document.xpath('//table[@id="transfer_form:receiptAccount"]')
        for tr in table[0].xpath('tbody/tr'):
            tds = tr.xpath('td')
            id = tds[0].xpath('input')[0].attrib['value']
            name = tds[0].xpath('label')[0].text
            name += u" " + tds[1].xpath('span')[0].text
            recipient = Recipient()
            recipient.id = id
            recipient.label = name
            yield recipient

        # Second, externals recipients
        select = self.document.xpath('//select[@id="transfer_form:externalAccounts"]')
        recipients = select[0].xpath('option')
        recipients.pop(0)
        for option in recipients:
            recipient = Recipient()
            recipient.id = option.attrib['value']
            recipient.label = option.text
            yield recipient

    def ischecked(self, account):
        id = account.id
        # remove prefix (CC-, LA-, ...)
        id = id[3:]
        search = '//input[@value="%s"]' % id
        option = self.document.xpath('//input[@value="%s"]' % id)
        if len(option) < 0:
            raise AccountNotFound()
        else:
            option = option[0]
        try:
            if option.attrib["checked"] == "checked":
                return True
            else:
                return False
        except:
            return False

    def transfer(self, recipient, amount, reason):
        self.browser.select_form("transfer_form")
        self.browser.set_all_readonly(False)
        for a in self.browser.controls[:]:
            #for label in a.get_labels():
            if "transfer_form:_link_hidden_" in str(a) or "transfer_form:j_idcl" in str(a):
                self.browser.controls.remove(a)
        self.browser.controls.append(ClientForm.TextControl('text', 'AJAXREQUEST', {'value': "transfer_form:transfer_region"}))
        self.browser['transfer_form:transferMotive'] = reason
        self.browser.controls.append(ClientForm.TextControl('text', 'transfer_form:valide', {'value': "transfer_form:valide"}))
        self.browser['transfer_form:validateDoTransfer'] = "needed"
        self.browser['transfer_form:transferAmount'] = str(amount)
        self.browser['transfer_recipient_radio'] = [recipient]
        self.browser.submit()


class TransferConfirmPage(BasePage):
    def on_loaded(self):
        pass