From d309ca0e96d9f190a80edc6d5f1af7bdb033426e Mon Sep 17 00:00:00 2001 From: Florent Date: Fri, 1 Mar 2013 21:58:02 +0100 Subject: [PATCH] Add an id for balance objects --- modules/leclercmobile/backend.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/modules/leclercmobile/backend.py b/modules/leclercmobile/backend.py index e32ceba5..f668467b 100644 --- a/modules/leclercmobile/backend.py +++ b/modules/leclercmobile/backend.py @@ -99,5 +99,9 @@ class LeclercMobileBackend(BaseBackend, ICapBill): return self.browser.readurl(bill._url) def get_balance(self, subscription): + if not isinstance(subscription, Subscription): + subscription = self.get_subscription(subscription) with self.browser: - return self.browser.get_balance() + balance = self.browser.get_balance() + balance.id = "%s-balance" % subscription.id + return balance