Support of downloading monthly pdf report

modules/ing/pages/bills.py

@@ -0,0 +1,80 @@
+# -*- coding: utf-8 -*-
+
+# Copyright(C) 2009-2012  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.tools.mech import ClientForm
+from weboob.capabilities.bill import Bill, Subscription
+from weboob.tools.browser import BasePage
+
+
+__all__ = ['BillsPage']
+
+
+class BillsPage(BasePage):
+    def on_loaded(self):
+        pass
+
+    def iter_account(self):
+        ul = self.document.xpath('//ul[@id="accountsel_form:accountsel"]')
+        #subscriber = unicode(self.document.find('//h5').text)
+        for li in ul[0].xpath('li/a'):
+            label = li.text
+            id = label.split(' ')[-1]
+            subscription = Subscription(id)
+            subscription.label = label
+#            subscription.subscriber = subscriber
+            subscription._localid = li.attrib['id']
+            yield subscription
+
+    def selectyear(self, id):
+        self.browser.select_form("accountsel_form")
+        self.browser.set_all_readonly(False)
+        self.browser.controls.append(ClientForm.TextControl('text', 'AJAXREQUEST', {'value': 'accountsel_form:accountsel_region'}))
+        self.browser.controls.append(ClientForm.TextControl('text', id, {'value': id}))
+        self.browser.submit(nologin=True)
+
+    def postpredown(self, id):
+        self.browser.select_form("statements_form")
+        self.browser.set_all_readonly(False)
+        self.browser.controls.append(ClientForm.TextControl('text', 'AJAXREQUEST', {'value': 'statements_form:stat_region'}))
+        self.browser.controls.append(ClientForm.TextControl('text', id, {'value': id}))
+        self.browser.submit(nologin=True)
+
+    def islast(self):
+        return True
+
+    def next_page(self):
+        pass
+
+    def iter_bills(self, subscriptionid):
+        ul = self.document.xpath('//ul[@id="statements_form:statementsel"]')
+        lis = ul[0].xpath('li')
+        lis.pop(0)  # Select alls
+        for li in lis:
+            acheck = li.xpath('a')[0]
+            adirect = li.xpath('a')[1]
+            label = unicode(acheck.text_content())
+            id = subscriptionid + '-' + label.replace(' ', '-')
+            bill = Bill()
+            bill.id = id
+            bill.label = label
+            bill.format = u"pdf"
+            onmouse = adirect.attrib['onmouseover']
+            bill._localid = onmouse.split("'")[5]
+            bill._url = adirect.attrib['href']
+            yield bill