electsys.py 文件源码

def recognize(self):
        def format_captcha(captcha):
            temp = ''
            for i in captcha:
                if (ord(i)>=48 and ord(i)<=57) or (ord(i)>=65 and ord(i)<=90) or (ord(i)>=97 and ord(i)<=122):
                    temp = temp + i
            if temp=='':
                temp = 'aaaa'
            return temp
        cookie = self.browser.cookies.all()
        opener = urllib2.build_opener()
        opener.addheaders.append(('User-Agent', 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/56.0.2924.87 Safari/537.36'))
        opener.addheaders.append(('Host','jaccount.sjtu.edu.cn'))
        opener.addheaders.append(('Referer',self.browser.url))
        opener.addheaders.append(('Cookie',  "; ".join('%s=%s' % (k,v) for k,v in cookie.items())))
        f = opener.open("https://jaccount.sjtu.edu.cn/jaccount/captcha?1488154642719")
        data = f.read()
        with file('captcha.png','wb') as f:
            f.write(data)
        img = Image.open("captcha.png").convert('L')
        result = format_captcha(pytesseract.image_to_string(img,lang="eng"))
        return result