# -*- coding: utf-8 -*- import datetime import hashlib import time from module.common.json_layer import json_loads from module.plugins.internal.Account import Account class RapideoPl(Account): __name__ = "RapideoPl" __type__ = "account" __version__ = "0.03" __status__ = "testing" __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 load_account_info(self, name, req): self._req = req try: result = json_loads(self.run_auth_query()) 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 = json_loads(self.run_auth_query()) except Exception: self.wrong_password() if "errno" in response.keys(): self.wrong_password() data['usr'] = self._usr data['pwd'] = self._pwd def create_auth_query(self): query = self.API_QUERY query['username'] = self._usr query['password'] = self._pwd return query def run_auth_query(self): return self.load(self.API_URL, post=self.create_auth_query())