python ログインサイトの画面キャプチャ

Posted 2021-05-09

from selenium import webdriver

# 1. ログインサイトにアクセス
url_login="http://uta.pw/sakusibbs/users.php?action=login&m=try"
driver = webdriver.PhantomJS()
driver.get(url_login)
print("ログインページへアクセス")


# 2. フォームにIDとパスワードをセットして送信
USER = "ユーザーネーム"
PASS = "パスワード"

textbox = driver.find_element_by_id("user")
textbox.clear()
textbox.send_keys(USER)

textbox = driver.find_element_by_id("pass")
textbox.clear()
textbox.send_keys(PASS)
form = driver.find_element_by_css_selector("#loginForm form")
form.submit()

# 3. ログイン後HTMLからマイページURLを取得
a = driver.find_element_by_css_selector(".islogin a")
url_mypage = a.get_attribute('href')
print("マイページURL",url_mypage)

# 4. ログイン後のURLの画面をキャプチャ
driver.get(url_mypage)
driver.save_screenshot("result.png")
driver.quit()