时间戳

首页 > 解决方案 > 有没有办法使用 Ruby 在 Centos 中使用 headless chromedriver 下载文件？

ruby - 有没有办法使用 Ruby 在 Centos 中使用 headless chromedriver 下载文件？

问题描述

我尝试使用无头 chrome 下载文件，但该文件似乎没有在任何地方下载。我可以看到它实际上是一种安全功能，可以限制无头文件下载，但是，在 Ruby 中是否有相同的解决方法？尝试了下面的代码，但没有运气。

download_path = "#{Pathname.pwd}/test-data/downloaded"
options = Selenium::WebDriver::Chrome::Options.new
options.add_argument("--disable-dev-shm-usage");
options.add_argument('--headless') #Declaring the browser to run in headless mode
options.add_preference(
  :download, directory_upgrade: true,
  prompt_for_download: false,
  default_directory: download_path
)
options.add_preference(:browser, set_download_behavior: { behavior: 'allow' })
@driver = Selenium::WebDriver.for :chrome, options: options #Browser object initialization
set_screen_resolution(1400, 900)
$driver = @driver

bridge = @driver.send(:bridge)
path = '/session/:session_id/chromium/send_command'
path[':session_id'] = bridge.session_id
bridge.http.call(:post, path, cmd: 'Page.setDownloadBehavior',
                        params: {
                          behavior: 'allow',
                          downloadPath: download_path
                        })

我希望使用无头 chrome 下载文件，但它没有发生。

标签： rubyselenium-webdrivergoogle-chrome-headless

解决方案

当您单击下载链接并且如果它在文件开始下载之前在单独的选项卡中打开时，那么您还需要将上述脚本应用到新打开的选项卡，因为您只为当前选项卡设置了会话 ID而不是新打开的标签。因此，在尝试下载文件之前，请尝试将此脚本应用于新打开的选项卡。我相信它会起作用的。

def download_file(label, download_path)
  ele = Locator.new(:xpath, "//ul[@class='cnt']//div[text()='#{label}']/..//a")
  download_url = get_attribute(ele.get_how, ele.get_what, "href")
  @driver.execute_script("window.open()")
  @driver.switch_to.window(@driver.window_handles.last)
  bridge = @driver.send(:bridge)
  path = '/session/:session_id/chromium/send_command'
  path[':session_id'] = bridge.session_id
  bridge.http.call(:post, path, {
    "cmd" => 'Page.setDownloadBehavior',
    "params" => {
      "behavior" => 'allow',
      "downloadPath" => download_path
    }
  })
  @driver.get(download_url)
  @driver.close
  @driver.switch_to.window(@driver.window_handles.last)
end

推荐阅读