[OboomCom] Fix account login

1 files changed, 28 insertions, 19 deletions

diff --git a/module/plugins/accounts/OboomCom.py b/module/plugins/accounts/OboomCom.py
index a37759f9a..84e0791e4 100644
--- a/module/plugins/accounts/OboomCom.py
+++ b/module/plugins/accounts/OboomCom.py
@@ -2,7 +2,7 @@
 
 import time
 
-from module.lib.beaker.crypto.pbkdf2 import PBKDF2
+from beaker.crypto.pbkdf2 import PBKDF2
 
 from module.common.json_layer import json_loads
 from module.plugins.Account import Account
@@ -11,7 +11,7 @@ from module.plugins.Account import Account
 class OboomCom(Account):
     __name__ = "OboomCom"
     __type__ = "account"
-    __version__ = "0.1"
+    __version__ = "0.2"
 
     __description__ = """Oboom.com account plugin"""
     __author_name__ = "stanley"
@@ -22,32 +22,41 @@ class OboomCom(Account):
         passwd = self.getAccountData(user)['password']
         salt = passwd[::-1]
         pbkdf2 = PBKDF2(passwd, salt, 1000).hexread(16)
-        result = json_loads(req.load("https://www.oboom.com/1.0/login", get={"auth": user, "pass": pbkdf2}))
+        result = json_loads(req.load("https://www.oboom.com/1/login", get={"auth": user, "pass": pbkdf2}))
         if not result[0] == 200:
             self.logWarning("Failed to log in: %s" % result[1])
             self.wrongPassword()
         return result[1]
 
+
     def loadAccountInfo(self, name, req):
         accountData = self.loadAccountData(name, req)
+
         userData = accountData['user']
 
-        if "premium_unix" in userData:
-            validUntilUtc = int(userData['premium_unix'])
-            if validUntilUtc > int(time.time()):
-                premium = True
-                validUntil = validUntilUtc
-                traffic = userData['traffic']
-                trafficLeft = traffic['current']
-                maxTraffic = traffic['max']
-                session = accountData['session']
-                return {"premium": premium,
-                        "validuntil": validUntil,
-                        "trafficleft": trafficLeft / 1024,
-                        "maxtraffic": maxTraffic / 1024,
-                        "session": session
-                }
-        return {"premium": False, "validuntil": -1}
+        if userData['premium'] == "null":
+            premium = False
+        else:
+            premium = True
+
+        if userData['premium_unix'] == "null":
+            validUntil = -1
+        else:
+            validUntil = int(userData['premium_unix'])
+
+        traffic = userData['traffic']
+
+        trafficLeft = traffic['current']
+        maxTraffic = traffic['max']
+
+        session = accountData['session']
+
+        return {'premium': premium,
+                'validuntil': validUntil,
+                'trafficleft': trafficLeft / 1024,
+                'maxtraffic': maxTraffic / 1024,
+                'session': session}
+
 
     def login(self, user, data, req):
         self.loadAccountData(user, req)