# -*- 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 NoPremiumPl(Account):
    __name__ = "NoPremiumPl"
    __version__ = "0.01"
    __type__ = "account"
    __description__ = "NoPremium.pl account plugin"
    __author_name__ = ("goddie")
    __author_mail__ = ("dev@nopremium.pl")

    _api_url = "http://crypt.nopremium.pl"

    _api_query = {
        "site": "nopremium",
        "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"] is not None:
            premium = True
            valid_untill = mktime(datetime.fromtimestamp(int(result["expire"])).timetuple())
        traffic_left = result["balance"] * 1024

        return ({
                    "validuntil": valid_untill,
                    "trafficleft": traffic_left,
                    "premium": premium
                })

    def login(self, user, data, req):
        self._usr = user
        self._pwd = hashlib.sha1(hashlib.md5(data["password"]).hexdigest()).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
# -*- 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 NoPremiumPl(Account):
    __name__ = "NoPremiumPl"
    __version__ = "0.01"
    __type__ = "account"
    __description__ = "NoPremium.pl account plugin"
    __author_name__ = ("goddie")
    __author_mail__ = ("dev@nopremium.pl")

    _api_url = "http://crypt.nopremium.pl"

    _api_query = {
        "site": "nopremium",
        "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"] * 1024

        return ({
                    "validuntil": valid_untill,
                    "trafficleft": traffic_left,
                    "premium": premium
                })

    def login(self, user, data, req):
        self._usr = user
        self._pwd = hashlib.sha1(hashlib.md5(data["password"]).hexdigest()).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
remove explicit "is not None"