51Testing软件测试论坛

 找回密码
 (注-册)加入51Testing

QQ登录

只需一步,快速开始

微信登录,快人一步

手机号码,快捷登录

查看: 4195|回复: 0
打印 上一主题 下一主题

[转贴] Python+Selenium操作浏览器全屏截图

[复制链接]
  • TA的每日心情
    无聊
    3 天前
  • 签到天数: 1050 天

    连续签到: 1 天

    [LV.10]测试总司令

    跳转到指定楼层
    1#
    发表于 2021-5-20 09:55:45 | 只看该作者 回帖奖励 |倒序浏览 |阅读模式
     最近碰到一个问题,需求是把做好的报告发到领导邮箱(只能发邮箱,不接受邮箱链接等方式),大家都知道outlook对js是不支持的,只支持最简单的html样式,最后选定的方法是截图,但是报告可能很长啊,怎么能一次性全截下来呢,请大家往下看!
      首先本次实现是用的python3.6+selenium+chrome,python的安装和python库的安装就不多说了。用代码控制chrome的时候需要下一个driver,下载地址:http://chromedriver.storage.googleapis.com/index.html
      下载之前要注意看看你电脑上装的chrome版本号,把下好的driver放到跟chrome.exe一个文件夹就行了。

      接下来直接上代码:
      # -*- coding: utf-8 -*-
      from selenium import webdriver
      from selenium.webdriver.chrome.options import Options
      import os
      import time
      def get_image(url, pic_name):
      #chromedriver的路径
          chromedriver = r"C:\Users\name\AppData\Local\Google\Chrome\Application\chromedriver.exe"
          os.environ["webdriver.chrome.driver"] = chromedriver
      #设置chrome开启的模式,headless就是无界面模式
      #一定要使用这个模式,不然截不了全页面,只能截到你电脑的高度
          chrome_options = Options()
          chrome_options.add_argument('headless')
          driver = webdriver.Chrome(chromedriver,chrome_options=chrome_options)
      #控制浏览器写入并转到链接
          driver.get(url)
          time.sleep(1)
      #接下来是全屏的关键,用js获取页面的宽高,如果有其他需要用js的部分也可以用这个方法
          width = driver.execute_script("return document.documentElement.scrollWidth")
          height = driver.execute_script("return document.documentElement.scrollHeight")
          print(width,height)
      #将浏览器的宽高设置成刚刚获取的宽高
          driver.set_window_size(width, height)
          time.sleep(1)
      #截图并关掉浏览器
          driver.save_screenshot(pic_name)
          driver.close()
      #你输入的参数
      url = 'https://movie.douban.com/top250'
      pic_name = r'D:\image.png'
      get_image(url, pic_name)

      很简单的几行代码,随意截任何网站的图片,通过这个方式也可以截网页任意部位的图,怎么样你学会了吗?
    分享到:  QQ好友和群QQ好友和群 QQ空间QQ空间 腾讯微博腾讯微博 腾讯朋友腾讯朋友
    收藏收藏
    回复

    使用道具 举报

    本版积分规则

    关闭

    站长推荐上一条 /1 下一条

    小黑屋|手机版|Archiver|51Testing软件测试网 ( 沪ICP备05003035号 关于我们

    GMT+8, 2024-11-24 11:36 , Processed in 0.066706 second(s), 24 queries .

    Powered by Discuz! X3.2

    © 2001-2024 Comsenz Inc.

    快速回复 返回顶部 返回列表