Initial poivy import

modules/poivy/pages.py

@@ -0,0 +1,91 @@
+# -*- 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
+from datetime import datetime, date, time
+
+import re
+
+__all__ = ['LoginPage', 'HomePage', 'HistoryPage', 'BillsPage']
+
+
+class LoginPage(BasePage):
+    def on_loaded(self):
+        pass
+
+    def login(self, login, password):
+        # Form without name
+        self.browser.select_form(nr=1)
+        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.submit(nologin=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)
+            detail.price = Decimal(price)
+            detail.currency = Currency.CUR_EUR
+
+            yield detail
+
+
+class BillsPage(BasePage):
+    def on_loaded(self):
+        pass