seleniumでgoogle chromeが一瞬で落ちる問題

Question

質問内容

seleniumを用いてWeb画面上の操作をしたいと思ったのですが，chormeが立ち上がってすぐに落ちてしまう事があります．
簡単に言えば，ソースコード(1)のように関数内にseleniumの記述をするとそういう現状になります．一方，ソースコード(2)の様にするとchormeは落ちる事なくプログラムを実行できます．
ChormeドライバーとChormeのバージョンは一致しており，(1)のプログラムにおいてもエラーは出ませんでした．
この原因がわかる方は教えていただけると助かります．

ソースコード(1)

def shop_bot():
    driver = webdriver.Chrome('パスを記入')
    driver.get('サイトのURLを記入')
    btn = driver.find_element_by_class_name('purchaseButton')
    btn.click()

if __name__ == '__main__':
    shop_bot()

ソースコード(2)

if __name__ == '__main__':
    driver = webdriver.Chrome('パスを記入')
    driver.get('サイトのURLを記入')
    btn = driver.find_element_by_class_name('purchaseButton')
    btn.click()

Answer 1

関数から抜ける際にdriverが破棄されますが、そのタイミングが異なっているからかもしれません。
関数化したほうの最後にtime.sleep(5)など追加して少し待ってみてはどうでしょう。