1. 程式人生 > >使用selenium模擬登陸oschina

使用selenium模擬登陸oschina

Selenium把元素定位介面封裝得更簡單易用了,支援Xpath、CSS選擇器、以及標籤名、標籤屬性和標籤文字查詢。

from selenium.webdriver import PhantomJS
from random import randint
import time
from selenium.webdriver.common.keys import Keys
from requests.cookies import RequestsCookieJar
import requests

def savepic():
    filename = '{}-{}.png'.format(int(time.time()), randint(100, 999))
    driver.save_screenshot(filename=filename)

with PhantomJS() as driver:
    driver.set_window_size(width=1280, height=1024)
    url = 'https://www.oschina.net/home/login'
    driver.get(url=url)
    # savepic()

    username = driver.find_element_by_id(id_='userMail')
    password = driver.find_element_by_id(id_='userPassword')
    username.send_keys('
[email protected]
') # 輸入使用者名稱 password.send_keys('password') # 輸入密碼 # savepic() password.send_keys(Keys.ENTER) # 輸入回車,提交表單 time.sleep(10) print(driver.current_url) # 登陸後跳轉到首頁 # userinfo = driver.find_element_by_class_name(name='user-info') while not driver.find_element_by_class_name(name='user-info').is_displayed(): time.sleep(1) savepic() cookies = driver.get_cookies() # 獲取cookie print(cookies, type(cookies)) for cookie in cookies: print(cookie) jar = RequestsCookieJar() for cookie in cookies: jar.set(name=cookie.get('name'), value=cookie.get('value')) print(jar) ua = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.102 Safari/537.36" headers = {'user-agent': ua} with requests.get(url=url, headers=headers) as resp: print(resp.url) # 不帶cookie會停留在登陸頁 with requests.get(url=url, headers=headers, cookies=jar) as resp: print(resp.url) # 帶上cookie會自動登陸跳轉到首頁 with open('osc.html', 'wb') as f: f.write(resp.content)

參考:
https://selenium-python.readthedocs.io/locating-elements.html
https://selenium-python.readthedocs.io/api.html#module-selenium.webdriver.common.keys