Merge pull request #682 from synweap15/rapideo

MultihosterRapideo.pl

1 files changed, 80 insertions, 0 deletions

diff --git a/module/plugins/accounts/RapideoPl.py b/module/plugins/accounts/RapideoPl.py
new file mode 100644
index 000000000..438ce7ad3
--- /dev/null
+++ b/module/plugins/accounts/RapideoPl.py
@@ -0,0 +1,80 @@
+# -*- coding: utf-8 -*-
+
+from datetime import datetime
+import hashlib
+
+from module.plugins.Account import Account
+from time import mktime
+from module.common.json_layer import json_loads as loads
+
+
+class RapideoPl(Account):
+    __name__ = "RapideoPl"
+    __version__ = "0.01"
+    __type__ = "account"
+    __description__ = "Rapideo.pl account plugin"
+    __license__ = "GPLv3"
+    __authors__ = [("goddie", "dev@rapideo.pl")]
+
+    _api_url = "http://enc.rapideo.pl"
+
+    _api_query = {
+        "site": "newrd",
+        "username": "",
+        "password": "",
+        "output": "json",
+        "loc": "1",
+        "info": "1"
+    }
+
+    _req = None
+    _usr = None
+    _pwd = None
+
+    def loadAccountInfo(self, name, req):
+        self._req = req
+        try:
+            result = loads(self.runAuthQuery())
+        except:
+            # todo: return or let it be thrown?
+            return
+
+        premium = False
+        valid_untill = -1
+        if "expire" in result.keys() and result["expire"]:
+            premium = True
+            valid_untill = mktime(datetime.fromtimestamp(int(result["expire"])).timetuple())
+
+        traffic_left = result["balance"]
+
+        return ({
+                    "validuntil": valid_untill,
+                    "trafficleft": traffic_left,
+                    "premium": premium
+                })
+
+    def login(self, user, data, req):
+        self._usr = user
+        self._pwd = hashlib.md5(data["password"]).hexdigest()
+        self._req = req
+        try:
+            response = loads(self.runAuthQuery())
+        except:
+            self.wrongPassword()
+
+        if "errno" in response.keys():
+            self.wrongPassword()
+        data['usr'] = self._usr
+        data['pwd'] = self._pwd
+
+    def createAuthQuery(self):
+        query = self._api_query
+        query["username"] = self._usr
+        query["password"] = self._pwd
+
+        return query
+
+    def runAuthQuery(self):
+        data = self._req.load(self._api_url, post=self.createAuthQuery())
+
+        return data
\ No newline at end of file