[PremiumTo] Fixup

1 files changed, 9 insertions, 7 deletions

diff --git a/module/plugins/accounts/PremiumTo.py b/module/plugins/accounts/PremiumTo.py
index ef3d0cc19..2c486a1dd 100644
--- a/module/plugins/accounts/PremiumTo.py
+++ b/module/plugins/accounts/PremiumTo.py
@@ -6,7 +6,7 @@ from module.plugins.Account import Account
 class PremiumTo(Account):
     __name__    = "PremiumTo"
     __type__    = "account"
-    __version__ = "0.05"
+    __version__ = "0.06"
 
     __description__ = """Premium.to account plugin"""
     __license__     = "GPLv3"
@@ -16,12 +16,14 @@ class PremiumTo(Account):
 
 
     def loadAccountInfo(self, user, req):
-        api_r = req.load("http://premium.to/api/straffic.php",
-                         get={'username': self.username, 'password': self.password})
-
-        trafficleft = sum(map(float, api_r.split(';'))) / 1024  #@TODO: Remove `/ 1024` in 0.4.10
-
-        return {'premium': True, 'trafficleft': trafficleft, 'validuntil': -1}
+        traffic = req.load("http://premium.to/api/traffic.php",
+                           get={'username': self.username, 'password': self.password})
+
+        if "wrong username" not in traffic:
+            trafficleft = float(traffic.strip()) / 1024  #@TODO: Remove `/ 1024` in 0.4.10
+            return {'premium': True, 'trafficleft': trafficleft, 'validuntil': -1}
+        else:
+            return {'premium': False, 'trafficleft': None, 'validuntil': None}
 
 
     def login(self, user, data, req):