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

# Copyright(C) 2010-2011  Romain Bignon, Pierre Mazière
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3 of the License.
#
# This program 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.


from datetime import date
from weboob.capabilities.bank import Operation
from weboob.capabilities.bank import Account
from weboob.tools.browser import BasePage, BrowserUnavailable

class LoginPage(BasePage):
    def login(self, agency, login, passwd):
        self.browser.select_form(nr=0)
        self.browser['agenceId'] = agency
        self.browser['compteId'] = login
        self.browser['CodeId'] = passwd
        try:
            self.browser.submit()
        except BrowserUnavailable:
            # Login is not valid
            return False
        return True

class LoginResultPage(BasePage):
    def is_error(self):
        for text in self.document.find('body').itertext():
            text=text.strip()
            # Login seems valid, but password does not
            needle='Les données saisies sont incorrectes'
            if text.startswith(needle.decode('utf-8')):
                return True

        return False

class FramePage(BasePage):
    pass


class AccountsPage(BasePage):
    def get_list(self):
        l = []
        for div in self.document.getiterator('div'):
            if div.attrib.get('class')=="unCompte-CC" :
                account = Account()
                account.id = div.attrib.get('id').replace('-','')
                for td in div.getiterator('td'):
                    if td.find("div") is not None and td.find("div").attrib.get('class') == 'libelleCompte':
                        account.label = td.find("div").text
                    elif td.find('a') is not None and td.find('a').attrib.get('class') is None:
                        balance = td.find('a').text.replace(u"\u00A0",'').replace('.','').replace('+','').replace(',','.')
                        account.balance = float(balance)
                        account.link_id = td.find('a').attrib.get('href')

                l.append(account)

        return l

class AccountHistoryPage(BasePage):
    def on_loaded(self):
        self.operations = []
        done=False
        for table in self.document.getiterator('table'):
            title_tr=table.find('tr')
            if title_tr is None:
                continue
            for text in title_tr.itertext():
                prefix='Opérations effectuées'
                if text.startswith(prefix.decode('utf-8')):
                    for tr in table.iter('tr'):
                        tr_class=tr.attrib.get('class')
                        if tr_class == 'tbl1' or tr_class=='tbl2':
                            tds=tr.findall('td')
                            d=date(*reversed([int(x) for x in tds[0].text.split('/')]))
                            label=u''+tds[1].find('a').text.strip()
                            if tds[3].text.strip() != u"":
                                amount = - float(tds[3].text.strip().replace('.','').replace(',','.').replace(u"\u00A0",'').replace(' ',''))
                            else:
                                amount= float(tds[4].text.strip().replace('.','').replace(',','.').replace(u"\u00A0",'').replace(' ',''))
                            operation=Operation(len(self.operations))
                            operation.date=d
                            operation.label=label
                            operation.amount=amount
                            self.operations.append(operation)
                    done=True
                    break
            if done:
                break

    def get_operations(self):
        return self.operations