# -*- coding: utf-8 -*- import datetime import hashlib import time from module.plugins.Account import Account 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 Exception: # 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 = time.mktime(datetime.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 Exception: 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