pythonでseleniumを実行中に外部ファイルを読み込む

Question

python3でseleniumを実行中に外部ファイルを読み込みたいです

input = driver.find_element_by_id("sa").send_keys(u"銀座")
input = driver.find_element_by_id("sk").send_keys(u"イタリアン")
銀座の箇所にa.txtの内容を入れて、イタリアンの箇所にb.txtの内容を入れてseleniumを実行したい。
a.txtとb.txtはすべての組み合わせでお願いします

a.txt

銀座
立川
新橋
有楽町

b.txt

イタリアン
フレンチ
焼き鳥
焼肉

Answer 1

全体として以下の手順で処理できます。

1. a.txt の各行から成る list_a と、b.txt の各行から成る list_b を作成。その後
2. list_a と list_b から、すべての組み合わせを要素とする list_ab を作成。
3. list_ab をループ処理して send_keys() を実行。

list_a と list_b は以下の方法で作成できます。

list_a = open("a.txt").read().splitlines()
list_b = open("b.txt").read().splitlines()

list_ab は以下の方法で作成できます。

list_ab = [(a,b) for a in list_a for b in list_b]

ここまでできればあとは通常のループ処理ですので、できますね！

ご参考になれば。

Comments

[DaichiYasuda]

ありがとうございます！
ループ処理をリスト内包表記で書きたいのですが、
教えていただいてもよろしいでしょうか？

Answer 2

すでにループを回す回答が付いてますが楽しそうなので私も書いてみました。

from itertools import product

with open('a.txt', encoding='utf8') as fa, open('b.txt', encoding='utf8') as fb:
    for a, b in product(fa, fb):
        a, b = map(str.rstrip, (a, b))
        if not a or not b:
            continue
        input = driver.find_element_by_id("sa").send_keys(a)
        input = driver.find_element_by_id("sk").send_keys(b)