# -*- coding: utf-8 -*-
import re
import time
from module.plugins.internal.Account import Account
class CatShareNet(Account):
    __name__    = "CatShareNet"
    __type__    = "account"
    __version__ = "0.08"
    __status__  = "testing"
    __description__ = """Catshare.net account plugin"""
    __license__     = "GPLv3"
    __authors__     = [("prOq", None)]
    PREMIUM_PATTERN      = r'Konto:[\s\n]*Premium'
    VALID_UNTIL_PATTERN  = r'>Konto premium.*?(.*?)'
    TRAFFIC_LEFT_PATTERN = r'([0-9.]+ [kMG]B)'
    def parse_info(self, user, password, data, req):
        premium     = False
        validuntil  = -1
        trafficleft = -1
        html = self.load("http://catshare.net/")
        if re.search(self.PREMIUM_PATTERN, html):
            premium = True
        try:
            expiredate = re.search(self.VALID_UNTIL_PATTERN, html).group(1)
            self.log_debug("Expire date: " + expiredate)
            validuntil = time.mktime(time.strptime(expiredate, "%Y-%m-%d %H:%M:%S"))
        except Exception:
            pass
        try:
            trafficleft = self.parse_traffic(re.search(self.TRAFFIC_LEFT_PATTERN, html).group(1))
        except Exception:
            pass
        return {'premium': premium, 'trafficleft': trafficleft, 'validuntil': validuntil}
    def login(self, user, password, data, req):
        html = self.load("http://catshare.net/login",  #@TODO: Revert to `https` in 0.4.10
                         post={'user_email'    : user,
                               'user_password' : password,
                               'remindPassword': 0,
                               'user[submit]'  : "Login"})
        if not 'Wyloguj' in html:
            self.login_fail()