Add balance command to leclercmobile

2012-08-29 16:41:36 +02:00 · 2012-08-29 16:41:36 +02:00 · 681b83e4b3
commit 681b83e4b3
parent 757167a6eb
4 changed files with 20 additions and 1 deletions
--- a/modules/leclercmobile/browser.py
+++ b/modules/leclercmobile/browser.py
@ -129,3 +129,10 @@ class Leclercmobile(BaseBrowser):
        for a in l:
            if a.id == id:
                return a
+
+    def get_balance(self):
+        if not self.is_on_page(HistoryPage):
+            self.location(self.conso)
+        response = self.openurl('/EspaceClient/pgeWERL015_RecupReleveConso.aspx?m=-0')
+        pdf = PdfPage(StringIO.StringIO(response.read()))
+        return pdf.get_balance()