From bc4624d7e39379f641ebd631384b1d39b7402b65 Mon Sep 17 00:00:00 2001
From: Romain Bignon <romain@symlink.me>
Date: Sun, 19 Oct 2014 13:20:38 +0200
Subject: [PATCH] implement get_account_status()

---
 modules/playme/browser.py |  3 +++
 modules/playme/module.py  | 11 ++++++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/modules/playme/browser.py b/modules/playme/browser.py
index 471b2abf..de9cc0d1 100644
--- a/modules/playme/browser.py
+++ b/modules/playme/browser.py
@@ -91,6 +91,7 @@ class PlayMeBrowser(DomainBrowser):
 
         self.my_id = me['id']
         self.my_name = me['name']
+        self.credits = me['credits']['count']
 
     def get_threads(self):
         return reversed(self.request('/users/%s/contacts' % self.my_id))
@@ -127,9 +128,11 @@ class PlayMeBrowser(DomainBrowser):
             r = self.request('/users/%s/challenge/%s' % (self.my_id, user_id))
         except ClientError as e:
             r = json.loads(e.response.content)
+            self.credits = r['credits']['count']
             raise NoCredits(r['credits']['next_restore_in_seconds'])
 
         t = self.get_theme()
+        self.credits = r['credits']['count']
 
         data = {}
         data['theme'] = {'id': t['theme']['id'], 'is_vip': 0}
diff --git a/modules/playme/module.py b/modules/playme/module.py
index d999ede1..dfce6ad6 100644
--- a/modules/playme/module.py
+++ b/modules/playme/module.py
@@ -22,6 +22,7 @@ import datetime
 
 from weboob.capabilities.messages import CapMessages, CapMessagesPost, Thread, Message
 from weboob.capabilities.dating import CapDating, Optimization
+from weboob.capabilities.account import CapAccount, StatusField
 from weboob.tools.backend import Module, BackendConfig
 from weboob.tools.value import Value, ValueBackendPassword
 from weboob.tools.date import local2utc
@@ -81,7 +82,7 @@ class ProfilesWalker(Optimization):
                 self._view_cron = self._sched.schedule(delay, self.view_profile)
 
 
-class PlayMeModule(Module, CapMessages, CapMessagesPost, CapDating):
+class PlayMeModule(Module, CapMessages, CapMessagesPost, CapDating, CapAccount):
     NAME = 'playme'
     DESCRIPTION = u'PlayMe dating mobile application'
     MAINTAINER = u'Roger Philibert'
@@ -191,5 +192,13 @@ class PlayMeModule(Module, CapMessages, CapMessagesPost, CapDating):
     def post_message(self, message):
         self.browser.post_message(message.thread.id, message.content)
 
+    # ---- CapAccount methods ---------------------
+
+    def get_account_status(self):
+        return (
+                StatusField(u'myname', u'My name', unicode(self.browser.my_name)),
+                StatusField(u'credits', u'Credits', unicode(self.browser.credits)),
+               )
+
     OBJECTS = {Thread: fill_thread,
               }