Selenium模拟登录魅族
- 魅族的验证码采用了第三方「极验」验证码平台，破解比较复杂，故采用打码平台超级鹰进行识别。原理是将验证码图片转换成字节流发送给超级鹰，超级鹰会返回验证码所在的坐标。
- 超级鹰注册后绑定微信会获取1000题分供测试使用。
- 超级鹰python API
#!/usr/bin/env python
# coding:utf-8
import requests
from hashlib import md5
# 超级鹰打码平台
class Chaojiying(object):
def __init__(self, username, password, soft_id):
self.username = username
password = password.encode('utf8')
self.password = md5(password).hexdigest()
self.soft_id = soft_id
self.base_params = {
'user': self.username,
'pass2': self.password,
'softid': self.soft_id,
}
self.headers = {
'Connection': 'Keep-Alive',
'User-Agent': 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
}
def PostPic(self, im, codetype):
"""
im: 图片字节
codetype: 题目类型 参考 http://www.chaojiying.com/price.html
"""
params = {
'codetype': codetype,
}
params.update(self.base_params)
files = {'userfile': ('ccc.jpg', im)}
r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files,
headers=self.headers)
return r.json()
def ReportError(self, im_id):
"""
im_id:报错题目的图片ID
"""
params = {
'id': im_id,
}
params.update(self.base_params)
r = requests.post('http://upload.chaojiying.net/Upload/ReportError.php', data=params, headers=self.headers)
return r.json()
if __name__ == '__main__':
chaojiying = Chaojiying('xxxx', 'xxxx', 'xxxx')
im = open('captch.png', 'rb').read()
print(chaojiying.PostPic(im, 9004))
-
meizu_login.py
==难点是验证码图片的获取要完整，在这里我选择了通过URL去获取，获得的图片类似于下图：==
captcha.png
import time
import urllib.request
from io import BytesIO
from PIL import Image
from selenium import webdriver
from selenium.webdriver import ActionChains
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from robot.media_admin.meizu_spider.chaojiying import Chaojiying
from utils.LogUtil import LogHandler
logger = LogHandler('meizu_login')
CHAOJIYING_USERNAME = 'xxx'
CHAOJIYING_PASSWORD = 'xxx'
CHAOJIYING_SOFT_ID = '901142' #软件ID官网生成
CHAOJIYING_KIND = '9004'
class MeizuLogin():
"""
主类
通过调用 chaojiying 类中的 多个方法进行图片上传，结果获取
"""
def __init__(self, username, password):
self.url = "https://login.flyme.cn/sso?appuri=http%3A%2F%2Fmzdsp.meizu.com%2Flogin&useruri=&sid=&service=app&" \
"autodirct=true"
path = '自己设备上chromedriver的路径'
self.browser = webdriver.Chrome(executable_path=path)
self.browser.maximize_window()
self.wait = WebDriverWait(self.browser, 20)
self.email = username
self.password = password
self.chaojiying = Chaojiying(CHAOJIYING_USERNAME, CHAOJIYING_PASSWORD, CHAOJIYING_SOFT_ID)
def __del__(self):
self.browser.close()
def open(self):
"""
打开网页输入用户名密码
:return: None
"""
self.browser.get(self.url)
email = self.wait.until(EC.presence_of_element_located((By.ID, 'account')))
password = self.wait.until(EC.presence_of_element_located((By.ID, 'password')))
email.send_keys(self.email)
password.send_keys(self.password)
def get_touclick_element(self):
"""
获取验证图片对象，就是验证码的点击区域
:return: 图片对象
"""
element = self.wait.until(EC.presence_of_element_located(
(By.CLASS_NAME, 'geetest_item_img')))
return element
def get_position(self):
"""
从节点对象，获取验证码位置
:return: 验证码位置元组
"""
element = self.get_touclick_element()
time.sleep(2)
location = element.location
size = element.size
top, bottom, left, right = location['y'], location['y'] + size['height'], location['x'], location['x'] + size[
'width']
return (top * 2, bottom * 2, left * 2, right * 2)
def get_screenshot(self):
"""
根据验证码位置，获取网页截图
:return: 截图对象
"""
screenshot = self.browser.get_screenshot_as_png()
screenshot = Image.open(BytesIO(screenshot))
return screenshot
def get_touclick_image(self, name='captcha.png'):
"""
保存验证码图片到本地
:return: 图片对象
"""
top, bottom, left, right = self.get_position()
print('验证码位置', top, bottom, left, right)
screenshot = self.get_screenshot()
captcha = screenshot.crop((left, top, right, bottom))
captcha.save(name)
return captcha
def get_points(self, captcha_result):
"""
{'err_no': 0, 'err_str': 'OK', 'pic_id': '9069220022448500002', 'pic_str': '208,75|269,156', 'md5': 'a2175369130a2ca56a77573cbf5cc7dc'}
解析识别结果，就是解析超级鹰返回的数据
:param captcha_result: 识别结果
:return: 转化后的结果
"""
groups = captcha_result.get('pic_str').split('|')
locations = [[int(number) for number in group.split(',')] for group in groups]
return locations
def touch_click_words(self, locations):
"""
点击验证图片
:param locations: 点击位置
:return: None
"""
for location in locations:
print(location)
ActionChains(self.browser).move_to_element_with_offset(self.get_touclick_element(), location[0],
location[1]).click().perform()
time.sleep(1)
def crack(self):
"""
破解入口
:return: None
"""
try:
self.open()
self.browser.find_element_by_class_name('geetest_radar_tip_content').click()
time.sleep(2)
# 获取验证码图片
image_src = str(self.browser.find_element_by_class_name('geetest_item_img').get_attribute('src'))
print(image_src)
image_src = image_src + 'png'
image = urllib.request.urlretrieve(image_src, filename='captcha.png')
im = open('captcha.png', 'rb').read()
# 识别验证码
result = self.chaojiying.PostPic(im, CHAOJIYING_KIND)
logger.info(result)
locations = self.get_points(result)
print(locations)
# 进行点击图片
self.touch_click_words(locations)
time.sleep(2)
# 点击登陆
self.browser.find_element_by_class_name('geetest_commit_tip').click()
time.sleep(1)
submit = self.wait.until(EC.element_to_be_clickable((By.ID, 'login')))
submit.click()
time.sleep(2)
login_id = self.browser.find_element_by_xpath('//*[@id="app"]/header/div[2]/div/p[2]/small').text
print(login_id)
return login_id
except Exception as e:
# 如果失败了超级鹰会返回分值
logger.error('登录失败:%s', e)
self.chaojiying.ReportError(result['pic_id'])
return None
finally:
# 退出chrome-headless 浏览器
self.browser.quit()
if __name__ == '__main__':
username = 'xxx'
password = 'xxx'
for i in range(3):
crack = MeizuLogin(username, password)
uid = crack.crack()
if uid is None:
continue
else:
print(uid)
break
