在selenium中保存URL中的图像(Python)

时间:2016-03-18 11:12:45

标签: python python-2.7 selenium sendkeys

所以,我终于完成了我的简单脚本,从某个站点获取一些图像链接。现在,问题是硒没有保存这些图像。不,我不想要截图,他们没有达到目的。屏幕截图留下了很多空白区域,因为它看到了自己的分辨率。

无论如何,我试过这个发送“ctrl + S”然后按“Enter”:

saveas = ActionChains(driver).key_down(Keys.CONTROL).send_keys('s').key_up(Keys.CONTROL).key_down(Keys.ENTER).key_up(Keys.ENTER)
        saveas.perform()

现在,我在这里做错了什么?图片链接:http://imgcomic.naver.net/webtoon/654817/44/20160314180903_2cfdd685f7e4f2c93a54e819898d6fb1_IMAG01_2.jpg

2 个答案:

答案 0 :(得分:1)

Webdriver获取整页截图。但是,您可以尝试按照元素尺寸裁剪图像。以下java代码将帮助您

 driver.get("http://www.google.com");
    WebElement image = driver.findElement(By.id("hplogo"));   
    //Get entire page screenshot
    File screenshot = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
    BufferedImage  fullImg = ImageIO.read(screenshot);
    //Get the location of element on the page
    Point point = image.getLocation();
    //Get width and height of the element
    int imageWidth = image.getSize().getWidth();
    int imageHeight = image.getSize().getHeight();
    //Crop the entire page screenshot to get only element screenshot
    BufferedImage eleScreenshot= fullImg.getSubimage(point.getX(), point.getY(), imageWidth,
            imageHeight);
    ImageIO.write(eleScreenshot, "png", screenshot);
    //Copy the element screenshot to disk
    FileUtils.copyFile(screenshot, new File("E:\\selenium_desk\\GoogleLogo_screenshot1.png"));
    driver.close();

谢谢你, 穆拉利

答案 1 :(得分:-1)

您可以使用图片网址创建另一个webdriver实例并截取屏幕截图。不理想,但大部分时间都适合我。

fb = webdriver.Firefox()
fb.set_window_size(1200,2000) # Make sure the browser size is large enough
fb.get('IMAGE_URL')
imgs = fb.find_elements_by_tag_name('img')
imgs[0].screenshot('screenshot.png')

然后,如果尺寸很重要,您可以使用Pillow将图像转换为jpg。