116 lines
3.8 KiB
Python
116 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright(C) 2013 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.tools.browser import BasePage
|
|
from weboob.capabilities.base import Currency
|
|
from weboob.capabilities.bill import Subscription, Detail
|
|
from decimal import Decimal, InvalidOperation
|
|
from datetime import datetime, date, time
|
|
|
|
import re
|
|
|
|
__all__ = ['LoginPage', 'HomePage', 'HistoryPage', 'BillsPage', 'ErrorPage']
|
|
|
|
|
|
class ErrorPage(BasePage):
|
|
def on_loaded(self):
|
|
pass
|
|
|
|
|
|
class LoginPage(BasePage):
|
|
def on_loaded(self):
|
|
pass
|
|
|
|
def _predicate_form(self, form):
|
|
try:
|
|
return form.attrs['class'] == "form-detail"
|
|
except:
|
|
return False
|
|
|
|
def login(self, login, password):
|
|
captcha = self.document.xpath('//label[@class="label_captcha_input"]')
|
|
if len(captcha) > 0:
|
|
return False
|
|
|
|
form_newsletter = self.document.xpath('//form[@id="newsletter_form"]')[0]
|
|
hidden_input = form_newsletter.xpath('./input[@type="hidden"]')[0]
|
|
hidden_id = hidden_input.attrib["value"]
|
|
hidden_name = hidden_input.attrib["name"]
|
|
|
|
# Form without name
|
|
self.browser.select_form(predicate=self._predicate_form)
|
|
self.browser.set_all_readonly(False)
|
|
self.browser['login[username]'] = login.encode('iso-8859-1')
|
|
self.browser['login[password]'] = password.encode('iso-8859-1')
|
|
self.browser[hidden_name] = hidden_id
|
|
self.browser.submit(nologin=True)
|
|
return True
|
|
|
|
|
|
class HomePage(BasePage):
|
|
def on_loaded(self):
|
|
pass
|
|
|
|
def get_list(self):
|
|
spanabo = self.document.xpath('//span[@class="welcome-text"]/b')[0]
|
|
owner = spanabo.text_content()
|
|
credit = self.document.xpath('//span[@class="balance"]')[0].text_content()
|
|
|
|
subscription = Subscription(owner)
|
|
subscription.label = u"Poivy - %s - %s" % (owner, credit)
|
|
subscription._balance = Decimal(re.sub(u'[^\d\-\.]', '', credit))
|
|
|
|
return [subscription]
|
|
|
|
|
|
class HistoryPage(BasePage):
|
|
months = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
|
|
|
|
def on_loaded(self):
|
|
pass
|
|
|
|
def get_calls(self):
|
|
table = self.document.xpath('//table/tbody')[0]
|
|
for tr in table.xpath('tr'):
|
|
tds = tr.xpath('td')
|
|
|
|
rawdate = tds[0].text_content()
|
|
splitdate = rawdate.split('-')
|
|
month_no = self.months.index(splitdate[1]) + 1
|
|
mydate = date(int(splitdate[2]), month_no, int(splitdate[0]))
|
|
|
|
rawtime = tds[1].text_content()
|
|
mytime = time(*[int(x) for x in rawtime.split(":")])
|
|
|
|
price = re.sub(u'[^\d\-\.]', '', tds[6].text)
|
|
detail = Detail()
|
|
detail.datetime = datetime.combine(mydate, mytime)
|
|
detail.label = u"%s from %s to %s - %s" % (tds[2].text, tds[3].text, tds[4].text, tds[5].text)
|
|
try:
|
|
detail.price = Decimal(price)
|
|
except InvalidOperation:
|
|
detail.price = Decimal(0) # free calls
|
|
detail.currency = Currency.CUR_EUR
|
|
|
|
yield detail
|
|
|
|
|
|
class BillsPage(BasePage):
|
|
def on_loaded(self):
|
|
pass
|