seleniumで、chromedriverが開いていない状態であればログイン処理、開いている状態であればログイン後のページに直接飛ぶという処理をしたい。

Question

前提・実現したいこと

・前提
プログラミング初心者
pythonでseleniumとchromedriverを使用して自動ログインするプログラムを組んでいます。

・実現したいこと
chromedriverが開いていない状態であればログイン処理、開いている状態(chromedriverのウインドウが開いている)であればログイン後のページに直接飛ぶという処理をしたいです。

・補足
実現したいことの大元は、プログラムを動かす際に毎回二段階認証を行うのをやめたいということです。
そのためにユーザープロファイルを使ってログイン状態を維持しようと考えています。

###教えていただきたいこと
・chromedriverのウインドウが開いているかを判断する方法

実現したいことを叶える、もっと効率的な方法などがありましたらそれも教えていただけると幸いです。

想定イメージ

if ドライバーが開いている=false:
    ログイン処理

elif ドライバーが開いている=true:
    driver.get(url)

それ以降の処理

補足情報（FW/ツールのバージョンなど）

Python 3.8.3 64-bit(conda)
windows10

Answer 1

import time
from selenium import webdriver
import chromedriver_binary
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities

def main():
  user_profile = 'UserProfile'
  options = webdriver.ChromeOptions()
  options.add_argument('--user-data-dir=' + user_profile)

  driver = webdriver.Chrome(options=options)
  url = "https://teratail.local/wp-admin/edit.php"
  driver.get(url)

  time.sleep(5)

  if "ログイン" in driver.title:
    login(driver)

  time.sleep(5)
  driver.close()

def login(driver):
  username = "admin"
  password = "password"
  driver.find_element_by_css_selector("#user_login").send_keys(username)
  driver.find_element_by_xpath("//*[@id='user_pass']").send_keys(password)
  driver.find_element_by_css_selector("#rememberme").click()
  driver.find_element_by_css_selector("#wp-submit").click()

if __name__ == "__main__":
  main()

ローカル環境(WordPress)で試しています。

1回目はログイン情報が保存されていないので管理ページを開こうとすると
ログインページに転送されます。
転送されたことを確認（ページタイトルがログイン）して、
ユーザー名、パスワードを入力して submit します。

2回目は UserProfile に認証情報が保存されているので、
ログインページに転送されることなく管理ページが開きます。

目的とは外れているかもしれませんが、今ひとつ問題点が把握できないので参考までにお知らせします。

Answer 2

私もそれほど詳しいわけではないので外れていたらすみません。

Open multiple tabs in selenium using python

こちらを参考に、

url = 'https://www.yahoo.com'
control_string = "window.open('{0}')".format(url)
driver.execute_script(control_string)

とすれば別のタブでページを開くことができました。

タブの移動は、
How to Switch Tabs in Selenium For Python

windows = driver.window_handles
for w in windows:
  driver.switch_to.window(w)

これで確認しました。次の記事も参考になるかもしれません。

（参考）Seleniumで新規タブを開く

Answer 3

Python+Selenium+ChromeDriverでログオン認証状態を維持する方法
こちらの記事は参考になりますでしょうか。

Comments

[Rujuu]

回答ありがとうございます！

教えていただいた記事を見たのですが、記事に書いてある「まず、スクリプトを実行すると最初は認証が必要なため、ログオン画面が開きます。手動でログオン情報を入力してみてください。」というのも自動化したいのです。

１つのプログラムファイルで組みたいので、初回起動時はログイン処理、2回目以降はログイン処理を行わないというようなプログラムにしたいと考えています。

ログイン情報を入力するタグ(?)がなかったらログイン処理を行わないというプログラムはかけそうですが、初回起動時にchromedriverを起動する処理が必要となるので、chromedriverのウインドウが複数出てきてしまいます。

なにか良い方法はありますでしょうか、、。

[退会済みユーザー]

ソースコードで確認しないと行き違いがあるかもしれませんが、理解している範囲で。

上の記事でUserProfileの保存ができると思うので、
そのまま目的のページを driver.get(url) してみて、
その結果ログインページが表示されれば (タイトルがログインとか、現在のURLがログインを確認）
ログインスクリプトを実行でいかがでしょうか。

初回はログインページに転送されるのでログインスクリプトを実行。
2回目以降はUserProfileにより目的のURLが開くと思います。

[Rujuu]

公開できる範囲のソースコードはこのような感じです。(公開できる範囲が狭くて申し訳ないです)
"""
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.keys import Keys
import re
import time
#url
url1 = 楽天RMSのURL
# ユーザプロファイルのフォルダ名(実行フォルダに作成されます)
user_profile = 'UserProfile'
#chromedriverのoption設定
options = webdriver.ChromeOptions()
options.add_argument('--user-data-dir=' + user_profile)
options.add_argument('--no-sandbox')
options.add_argument('--disable-dev-shm-usage')
driver = webdriver.Chrome(options=options)
#要素のロード待ち時間
driver.implicitly_wait(10)

driver.get(url1)
"""

教えていただいたようにプログラムを組んでみたのですが、楽天RMSは一度ブラウザを閉じてしまうと再度ログインが必要なサイトのようで上手くいきませんでした。

・試したこと
試しにブラウザを閉じずにプログラムを動かしたのですが、下記サイトの最後に書かれているように、プロファイルが既に使用されているためエラーが発生してしまいました。
https://rabbitfoot.xyz/selenium-chrome-profile/

・改善案
ログイン後のブラウザを開いたまま、新規タブで目的のページのURLに飛ぶとログイン処理を飛ばすことができるので、「ブラウザが開いたままであれば新規タブでURLを開く」といった処理ができれば解決できるのではないかと考えています。
このような処理は可能なのでしょうか、、。

[退会済みユーザー]

> 一度ブラウザを閉じてしまうと再度ログインが必要

ブラウザを閉じる必要があるのでしょうか。
すみません。ちょっとこのあたりの状況がわかりません。

[Rujuu]

申し訳ないです、、、状況が説明不足でした。
現在の作業が以下の流れで、

①(プログラム実行↓
・webdriverが開かれる
・ログイン処理
・自動操作
プログラム終了)

②(別の作業)

③(プログラム実行↓
・webdriverが開かれる
・ログイン処理
・自動操作
プログラム終了)

実現したいことは、③のログイン処理を省略することです。

>ブラウザを閉じる必要があるのでしょうか。
プログラムを終了した後、再度同じユーザープロファイルでwebdriverを開くと
「Message: invalid argument: user data directory is already in use, please specify a unique value for --user-data-dir argument, or don't use --user-data-dir」
というエラーが出てきてしまうため、①で開いたブラウザ(ウインドウ)を閉じる必要があります。

Amazonなどであれば一度ブラウザを閉じたとしても、同じユーザープロファイルで開きなおせばログイン状態を維持できるとは思いますが、楽天RMSはウインドウを閉じると再度ログインが必要となるため行き詰っています。

下記サイトの方法も試したのですが、上記と同じエラーが出てしまいました。
URL：https://www.it-swarm-ja.tech/ja/selenium/selenium%E3%81%AF%E6%97%A2%E5%AD%98%E3%81%AE%E3%83%96%E3%83%A9%E3%82%A6%E3%82%B6%E3%83%BC%E3%82%BB%E3%83%83%E3%82%B7%E3%83%A7%E3%83%B3%E3%81%A8%E5%AF%BE%E8%A9%B1%E3%81%A7%E3%81%8D%E3%81%BE%E3%81%99%E3%81%8B%EF%BC%9F/941471844/

Answer 4

色々調べた結果RMSのシステム上難しそうだったので、一度プログラムを終了させるという方法をやめて、for文で自動操作の処理をすべて終わらせてから手で作業しなければならない工程を行うように変更しました。