# -*- coding: utf-8 -*-

import re
from time import mktime, strptime

from module.plugins.Account import Account


class MegasharesCom(Account):
    __name__    = "MegasharesCom"
    __type__    = "account"
    __version__ = "0.03"

    __description__ = """Megashares.com account plugin"""
    __license__     = "GPLv3"
    __authors__     = [("zoidberg", "zoidberg@mujmail.cz")]


    VALID_UNTIL_PATTERN = r'<p class="premium_info_box">Period Ends: (\w{3} \d{1,2}, \d{4})</p>'


    def loadAccountInfo(self, user, req):
        #self.relogin(user)
        html = req.load("http://d01.megashares.com/myms.php", decode=True)

        premium = False if '>Premium Upgrade<' in html else True

        validuntil = trafficleft = -1
        try:
            timestr = re.search(self.VALID_UNTIL_PATTERN, html).group(1)
            self.logDebug(timestr)
            validuntil = mktime(strptime(timestr, "%b %d, %Y"))
        except Exception, e:
            self.logError(e)

        return {"validuntil": validuntil, "trafficleft": -1, "premium": premium}


    def login(self, user, data, req):
        html = req.load('http://d01.megashares.com/myms_login.php',
                        post={"httpref"       : "",
                              "myms_login"    : "Login",
                              "mymslogin_name": user,
                              "mymspassword"  : data['password']},
                        decode=True)

        if not '<span class="b ml">%s</span>' % user in html:
            self.wrongPassword()