実行結果とは別にbool型の戻り値がある場合のスマートなコードの書き方

Question

driver = webdriver.Chrome()
url = 'https://www.google.com/'
driver.get(url)
driver.save_screenshot('./googleスクショ.png')

実行するとスクショが保存されるが、それとは別に「True」の戻り値がある。

「False」の時にメッセージを表示させたいのですが

driver = webdriver.Chrome()
url = 'https://www.google.com/'
driver.get(url)
if driver.save_screenshot('./googleスクショ.png'):
    driver.save_screenshot('./googleスクショ.png')
else:
    print('メッセージ')

とすると「driver.save_screenshot('./googleスクショ.png')」を２回書くことになるので
もう少しスマートにするにはどうすればいいですか？

Answer 1

driver.save_screenshot('./googleスクショ.png') or print('メッセージ')

でどうでしょう。

Answer 2

if not driver.save_screenshot('./googleスクショ.png'):
    print('メッセージ')

ではだめですか？

Comments

[Mitsuki_0]

ご回答頂きありがとうございました。

Answer 3

このようにしてはどうでしょうか。

if not driver.save_screenshot('./googleスクショ.png'):
    print('保存失敗')

または

success = driver.save_screenshot('./googleスクショ.png')
if not success:
    print('保存失敗')

Comments

[Mitsuki_0]

できました、ありがとうございます。