スクレイピングをする時に使うブラウザをRailsの開発環境と本番環境で切り替えたかったので方法を記録しておきます。

Rails.envで条件わけをしてprependメソッドで呼び出せるように設定します。

'lib/selenium_web_driver_extension.rb'
module SeleniumWebDriverExtension
  def for_rails
    case Rails.env
    when 'development' then
      self.for(:firefox)
    when 'production' then
      self.for(:chrome)
    else
      raise
    end
  end
end

Selenium::WebDriver.singleton_class.prepend(SeleniumWebDriverExtension)