# -*- 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 decimal import Decimal
import re

from weboob.capabilities.bank import Account, Currency
from weboob.capabilities.base import NotAvailable
from weboob.tools.browser import BasePage
from weboob.tools.capabilities.bank.transactions import FrenchTransaction


__all__ = ['AccountsList']


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

    def get_list(self):
        ids = set()
        for td in self.document.xpath('.//td[@nowrap="nowrap"]'):
            account = Account()
            account.currency = Currency.CUR_EUR
            link = td.xpath('.//a')[0]
            account._index = int(re.search('\d', link.attrib['href']).group(0))
            if not account._index in ids:
                ids.add(account._index)
                account.id = unicode(link.text.strip())
                account.label = account.id
                urltofind = './/a[@href="' + link.attrib['href'] + '"]'
                linkbis = self.document.xpath(urltofind).pop()
                if linkbis.text == link.text:
                    linkbis = self.document.xpath(urltofind)[1]
                account.balance = Decimal(FrenchTransaction.clean_amount(linkbis.text))
                account.coming = NotAvailable
                yield account