selenium 引数でセレクタなどを選択したい！！

Question

前提・実現したいこと

ここに質問の内容を詳しく書いてください。
プログラミングを学び始めて間もないので見苦しい点が多々あると思いますがどうかお許しください。
現在seleniumを使ってアプリケーションのテストを行うことになりました。
関数をまとめておくtest.pyを作り複数のファイルから関数を呼び出したいと思っています。
ただ、セレクタの数だけ関数を作成するのは面倒だと思い、下記のような関数を書きましたがうまくいきません。
どこか間違ってますでしょうか？
そもそも無理なことなのでしょうか？

下記以外にも色々な書き方を試してみましたが自分ではうまくいきませんでした。
どなたかご教授していただけると幸いです。

該当のソースコード

test.py

from selenium import webdriver
 
driver = webdriver.Chrome("D:\chromedriver")
driver.get("https://www.google.co.jp")
#検索テキストボックスの要素をId属性名から取得


def send_keys_text(SELECTER,  VALUE, TEXT)
 element = driver.find_element_by_SELECTER(VALUE)
 element.send_keys(TEXT)

def send_keys(SELECTER,  VALUE, KEY)
 element = driver.find_element_by_SELECTER(VALUE)
 element.send_keys(Keys.KEY)

page01.py

import test

def page01_test:
 test.send_keys_text(’id’, ’realbox’,  ‘テスト’)

 test.send_keys(’id’, ’realbox’, ‘CONTROL,"a"’)
 test.send_keys(’id’, ’realbox’, ‘DELETE’)

補足情報

プログラミング超初心者、selenium超超初心者の小心者なので、優しく教えていただけるとたすかります。

Answer 1

大きく間違っていることが2つあります。

1
find_element_by_SELECTERとありますが、このような関数は存在しません。
SELECTERという引数を持ってくることによって、
find_element_by_idを実現したいなのだと思いますが、
これは固定の関数名であり、引数を与えるということ自体ができないです。
メタプログラミングをすれば別かもしれませんが。。。

2

from selenium import webdriver

driver = webdriver.Chrome("D:\chromedriver")
driver.get("https://www.google.co.jp")

をファイル内にクラスも関数も定義せずにそのまま書いていますが、
これではimport testを実行した瞬間に、
driverという変数が行方不明になってしまいます。

---

それからこの程度の関数でしたらわざわざ共通化する必要性を感じられません。

以下とりあえず動きそうなものあげますが、
先にドットインストールやUdemy等で
学習をしてからでないと、まともに目的を果たすのは難しいと思います。

---

# test.py

from selenium import webdriver
from selenium.webdriver.common.keys import Keys

SELECTOR_KEYS = {'id': '#', 'class': '.'} # この実装は拡張性が乏しく微妙です

class selenium_base(object):
    def __init__(self):
        self.driver = webdriver.Chrome("D:\chromedriver")
        self.driver.get("https://www.google.co.jp")

    def send_keys_text(self, SELECTER, VALUE, TEXT)
        self.driver.find_element_by_css_selector(SELECTOR_KEYS.get(SELECTER) + VALUE).send_keys(TEXT)

    def send_keys(self, SELECTER, VALUE, *KEYS)
        self.driver.find_element_by_css_selector(SELECTOR_KEYS.get(SELECTER) + VALUE).send_keys(eval('Keys.' + KEYS[0]) if len(KEYS[0]) == 1 else eval('Keys.' + KEYS[0] + ', ' + KEYS[1]))

# page01.py

import test

def page01_test:
    selenium_instance = test.selenium_base()
    selenium_instance.send_keys_text('id', 'realbox', 'テスト')
    selenium_instance.send_keys('id', 'realbox', 'CONTROL', 'a')
    selenium_instance.send_keys('id', 'realbox', 'DELETE')

page01_test

---

ついでに言うとgoogleはスクレイピング禁止です。
規約で明記されていますのでご注意を。

コメントを受けて追記

だとしたらそもそも分け方が間違っています。
page毎に関数を作り、テスト毎に値だけ渡してやるべきです。

下記例は

• pege01で色々とフォームを入力し、バリデーションが正しいとpage02に遷移する。
• page02で色々とフォームを入力し、バリデーションが正しいとpage03に遷移する。

という想定で作ってみました。

# test.py

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.select import Select

class selenium_base(object):
    def __init__(self):
        self.driver = webdriver.Chrome("D:\chromedriver")
        self.driver.get("page01の入力ページ")

    def page01(self, text1, text2, radio1, radio2, pulldown1, pulldown2):
        hoge1 = '/html/body/div/div[1]/pre/em[1]'
        hoge2 = '/html/body/div/div[18]' # hoge6まで適切なxpathを書いておく

        self.driver.find_element_by_xpath(hoge1).send_keys(text1)
        self.driver.find_element_by_xpath(hoge2).send_keys(text2)
        self.driver.find_elements_by_xpath(hoge3)[radio1].click()
        self.driver.find_elements_by_xpath(hoge4)[radio2].click()
        Select(self.driver.find_element_by_xpath(hoge5)).select_by_visible_text(pulldown1)
        Select(self.driver.find_element_by_xpath(hoge6)).select_by_visible_text(pulldown2)
        self.driver.find_element_by_class_name('btn-primary').click() # 最後に送信ボタンを押す

    def page02(self, text1, text2, radios):
        hoge1 = '/html/body/div/div[1]/pre/em[1]'
        hoge2 = '/html/body/div/div[18]'
        hoge3 = '/html/body/div/div[19]/p'

        self.driver.find_element_by_xpath(hoge1).send_keys(text1)
        self.driver.find_element_by_xpath(hoge2).send_keys(text2)
        for radio in radios:
            self.driver.find_elements_by_xpath(hoge3)[radio].click()
        self.driver.find_element_by_class_name('btn-primary').click() # 最後に送信ボタンを押す

# test_case.py

import test

# 正常に最終ページまで遷移するテスト
def success_test:
    selenium_instance = test.selenium_base()
    try:
        selenium_instance.page01('あああ', 'いいい', 1, 3, 'ううう', 'えええ')
        if selenium_instance.title != 'page02のタイトル':
            print('正常なケースを試したはずですが、page01からpage02に遷移できませんでした。')
            selenium_instance.quit()
            return
        selenium_instance.page02('あああ', 'いいい', [2, 3, 6, 7])
        if selenium_instance.title != 'page03のタイトル':
            print('正常なケースを試したはずですが、page02からpage03に遷移できませんでした。')
        else:
            print('正常にpage03まで遷移できました。')
    catch:
        print('success_test中にエラーが発生しました。')
    finally:
        selenium_instance.quit()
        print('success_testを終了します。')

# page01からpage02への遷移で失敗するテスト
def error_test1:
    selenium_instance = test.selenium_base()
    try:
        # text1は10文字以上だとエラーが出るシステムだとする
        selenium_instance.page01('あああああああああああああああ', 'いいい', 1, 3, 'ううう', 'えええ')
        if selenium_instance.title == 'page02のタイトル':
            print('異常なケースを試したはずですが、page01からpage02に遷移できてしまいました。')
        else:
            print('想定通りpage01でエラーが発生し先に進めませんでした。')
    catch:
        print('error_test1中にエラーが発生しました。')
    finally:
        selenium_instance.quit()
        print('error_test1を終了します。')

# page02からpage03への遷移で失敗するテスト
def error_test2:
    selenium_instance = test.selenium_base()
    try:
        selenium_instance.page01('あああ', 'いいい', 1, 3, 'ううう', 'えええ')
        if selenium_instance.title != 'page02のタイトル':
            print('正常なケースを試したはずですが、page01からpage02に遷移できませんでした。')
            selenium_instance.quit()
            return
        # ラジオボタンの選択肢は5つ以下しか選べないシステムだとする
        selenium_instance.page02('あああ', 'いいい', [2, 3, 6, 7, 10, 11])
        if selenium_instance.title == 'page03のタイトル':
            print('異常なケースを試したはずですが、page02からpage03に遷移できてしまいました。')
        else:
            print('想定通りpage02でエラーが発生し先に進めませんでした。')
    catch:
        print('error_test2中にエラーが発生しました。')
    finally:
        selenium_instance.quit()
        print('error_test2を終了します。')

success_test
error_test1
error_test2

pytestの書き方ではないですが必要だったら適宜やってみてください。

Comments

[ryan06]

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

現在テストを行おうとしているアプリでは複数ページに渡り大量のフォームに値を入力、操作が必要で
被らない様に要素の取得にId、xpathを多用しています。(idが振られていないものも多)
Seleniumリファレンスの関数をほとんど使うことになるので少しでも記述量が少なくなればと上記の様なことを試みました。

やはり一つ一つセレクタごとに関数を作るのが無難でしょうか？

[shirai]

追記しました

Answer 2

まず、test.pyで関数を定義するときにコロン（:）をつけ忘れています。pythonで関数にコロンを忘れるとエラーになります。

あと、page01.py内で文字列を囲むときのクオーテーション(')もおかしいです。ちゃんと半角入力の状態でコロンを正しく打たないとエラーになります。

プログラム書くときはメモ帳アプリとかでなく、VS Codeなどのテキストエディタをちゃんと用意して、プログラムをかけば上記のミスが減ると思いますので、使うことをおすすめします。

プログラミング初心者であればいきなりseleniumuを使わずにこちらのサイトで勉強されてはいかがでしょうか。すべて無料で自分もこのサイトでPythonのプログラミングの基礎を勉強しました。
Python入門

そして、seleniumの方ですが、find_element_by_SELECTERというメソッドは存在しません。
一度、find_element系のメソッドがまとまっている記事があるので、ここを参考にするといいと思います。
Pythonで、Seleniumのfind_element系メソッドとPageObjectsを試してみた
そもそもseleniumはHTMLを理解していたほうがプログラムを書きやすいのでHTMLについても多少は勉強しましょう。

Answer 3

提示したソースで何をしたいのかよく読み取れませんが、
CSSセレクターを指定して要素を取得するメソッドはすでにありますが、こちらは利用できないのですか？

https://www.seleniumqref.com/api/python/element_get/Python_find_element_by_css_selector.html

find_elementでも良いと思います。