From b20ffc3f6e6b34db2a22b6db6f751ee4a4d58a44 Mon Sep 17 00:00:00 2001 From: Walter Purcaro Date: Sun, 26 Oct 2014 19:39:20 +0100 Subject: Assign in-line regex to PATTERN variable in a couple of accounts --- module/plugins/accounts/MegaRapidCz.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) (limited to 'module/plugins/accounts/MegaRapidCz.py') diff --git a/module/plugins/accounts/MegaRapidCz.py b/module/plugins/accounts/MegaRapidCz.py index 3ef98560a..04abf8c8a 100644 --- a/module/plugins/accounts/MegaRapidCz.py +++ b/module/plugins/accounts/MegaRapidCz.py @@ -18,22 +18,26 @@ class MegaRapidCz(Account): login_timeout = 60 + + LIMITDL_PATTERN = ur'Max. počet paralelních stahování: (\d+)' + VALID_UNTIL_PATTERN = ur'Paušální stahování aktivní. Vyprší (.*?)' + TRAFFIC_LEFT_PATTERN = r'Kredit(.*?) GiB' def loadAccountInfo(self, user, req): src = req.load("http://megarapid.cz/mujucet/", decode=True) - m = re.search(ur'Max. počet paralelních stahování: (\d+)', src) + m = re.search(self.LIMITDL_PATTERN, src) if m: data = self.getAccountData(user) data['options']['limitDL'] = [int(m.group(1))] - m = re.search(ur'Paušální stahování aktivní. Vyprší (.*?)', src) + m = re.search(self.VALID_UNTIL_PATTERN, src) if m: validuntil = mktime(strptime(m.group(1), "%d.%m.%Y - %H:%M")) return {"premium": True, "trafficleft": -1, "validuntil": validuntil} - m = re.search(r'Kredit(.*?) GiB', src) + m = re.search(self.TRAFFIC_LEFT_PATTERN, src) if m: trafficleft = float(m.group(1)) * (1 << 20) return {"premium": True, "trafficleft": trafficleft, "validuntil": -1} -- cgit v1.2.3