使用capybara将图像转换为base64

时间:2016-04-05 14:01:00

标签: ruby-on-rails ruby selenium capybara poltergeist

我目前正在使用capybara来运行一些报废任务以及现场测试。我一直在使用水豚下载图像/文件时遇到困难。我发现的所有文档都只是简单按钮,表单,链接交互的指南。

如果有人知道如何将网页上的图片下载/转换为base64格式,我们将非常感激。

2 个答案:

答案 0 :(得分:2)

此示例使用Capybara / Selenium从网页中提取图像:

require 'capybara'

JS_GET_IMAGE = "
  var ele = arguments[0], callback = arguments[1], img = new Image();
  img.crossOrigin = 'Anonymous';
  img.onload = function(){
    var cnv = document.createElement('CANVAS');
    cnv.width = this.width;
    cnv.height = this.height;
    cnv.getContext('2d').drawImage(this, 0, 0);
    var type = this.src.endsWith('png') ? 'png' : 'jpeg';
    callback(cnv.toDataURL('image/' + type).substring(22));
  };
  var src = ele.src || window.getComputedStyle(ele).backgroundImage;
  img.src = /https?:/.test(src) ? src.match(/https?:[^\"')]+/)[0] : callback(''); "


session = Capybara::Session.new(:selenium)
driver = session.driver.browser
driver.manage.timeouts.script_timeout = 5000

# navigate to google
session.visit "https://www.google.co.uk/"

# get the logo element
ele = session.find(:css, '#hplogo img:nth-child(1)')

# get the logo as base64 string
imgBase64 = driver.execute_async_script(JS_GET_IMAGE, ele.native)

# save to a file
file = File.new("C:\\temp\\image." + (imgBase64[0] == 'i' ? 'png' : 'jpg'), 'wb')
file.write(Base64.decode64(imgBase64))
file.close

答案 1 :(得分:0)

只是通过capybara gem看了一下,找到了一个.render_base64和save_screenshot方法,它可以将图像保存到png或jpg文件中,之后我可以裁剪我想要的部分。方法可以在这里找到:https://github.com/teampoltergeist/poltergeist