【ツール配布】安全確認回避、ポップアップ回避のseleniumライブラリ

13000 ワード
Python Selenium Python テキストリンク

コード

github:pyAutoTest

目的

やりたいこと

  • seleniumをシナリオで動かしたい
    • seleniumはコード書く必要あるが、毎回コードを書きたくない
      • 他部署の非エンジニアでも使えるようにしたい
    • ブラウザでコード出力できるが、毎回手作業したくない
  • システム毎に改修する部分は一ヶ所にまとめたい
  • デバッグ中にオレオレ証明書というだけで止めたくない
  • クライアント証明書提出で止めたくない

後回しでいいこと

  • 処理速度
    • 夜間に走らせるため
  • 結果確認の自動化
    • 証拠残す必要もあるのでスクショの確認でOK
    • 今後の課題

実装

オレオレ証明書回避

コード

def avoid_security_check(self):
    ok_flg = True

    if self.__driver.title:
        # ie
        if self.__driver.title == "このサイトは安全ではありません":
            ok_flg  = self.click("moreInfoContainer")

            ok_flg &= self.click("overridelink")

            time.sleep(0.5)

        # edge
        elif self.__driver.title == "証明書エラー: ナビゲーションはブロックされました":
            ok_flg  = self.click("moreInformationDropdownSpan")

            ok_flg &= self.click("invalidcert_continue")

            time.sleep(0.5)

    return ok_flg

説明

  • ページタイトルを確認して、忠告していればOK
    • 遷移に時間かかるため、sleep必須

クライアント証明書提出

コード

def cert(self, cmd):
    self.__pki_flg = False

    ok_flg = True
    with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:
        futures = []
        futures.append(executor.submit(self.post_cert_chrome, cmd))
        futures.append(executor.submit(self.submit, cmd))
        for future in concurrent.futures.as_completed(futures):
            ok_flg &= future.result()

    return ok_flg

def submit(self):
    ok_flg  = self.click("submit")

    return ok_flg

def post_cert_chrome(self, cmd):
    # cmd check is practiced
    if cmd == "ok":
        ok_flg = self.move_click("img/chrome/ok.png")

    elif cmd == "cancel":
        ok_flg = self.move_click("img/chrome/cancel.png")

    elif cmd.isdecimal():
        diff = 80 + (int(cmd) * 40)
        ok_flg  = self.move_click("img/chrome/next.png", 0, diff)

        ok_flg &= self.move_click("img/chrome/ok.png")

    return ok_flg

def move_click(self, img_path, x=0, y=0, seconds=0.2):
    # search from image
    pos, ok_flg = self.search_from_image(img_path)
    if ok_flg:
        # move mouse
        pyautogui.moveTo(pos.x + x, pos.y + y, seconds)
        # click
        pyautogui.click()

    return ok_flg

def search_from_image(self, img_path):
    # check image
    if not os.path.isfile(img_path):
        self.__log.printLog(LOG_LEVEL.EEROR, "failed to setting")
        return None, False

    i = 30
    while i > 0:
        i -= 1

        # get position
        pos = pyautogui.locateCenterOnScreen(img_path)

        if pos:
            break

        time.sleep(0.2)

    if not pos:
        return None, False

    return pos, True

説明

  • ポップアップ中はseleniumが止まるため、並列処理
    • selenium
      • ポップアップ出現のトリガを行う
    • pyautogui
      • ポップアップ画像を検索し、押下
        • ポップアップ出現までラグのある場合も考慮し、指定時間内はループ

まとめ

  • メリット
    • サンプルとしての有用性はある
      • 安全確認の回避
      • ポップアップの回避
  • 今後の課題
    • 結果確認が目視のため、自動化したい
    • ポップアップ対応のため、不都合がある
      • 画像検索するため非表示実行できない
      • ツールでマウス操作する場面があるため、テスト中にPC操作ができない
Python爬虫学习笔记(例:猫眼映画ランキング上位100)
大きい整数クラスの4則演算と論理演算