def test(self): with open(get_real_path('jjdecoder_tests.txt'), 'r') as tests: while True: req = tests.readline() if not req: break exp = tests.readline().strip() result = JJDecoder.decode(req) self.assertEqual(exp, result)
def get(self): page = UZ_BASE headers = dict() headers['User-Agent'] = self.user_agent resp = requests.get(page, headers=headers, cookies=self.cookies) self.cookies = resp.cookies raw_token = resp.content.decode('utf-8') marker_start = "_gaq.push(['_trackPageview']);" marker_stop = "(function ()" t_start = raw_token.find(marker_start) t_stop = raw_token.find(marker_stop, t_start) if t_start != -1 and t_stop != -1: raw_token = raw_token[t_start + len(marker_start):t_stop] decoded_token = JJDecoder.decode(raw_token) self.token = decoded_token[decoded_token.find('", "') + 4:-3] for c in self.cookies: if c.name == 'HTTPSERVERID': self.server = c.value elif c.name == '_gv_sessid': self.session_id = c.value return resp