百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术资源 > 正文

Python自动化办公——后台截图

off999 2024-12-03 00:21 18 浏览 0 评论

Python截图很简单,ImageGrab.grab()即可截取整个屏幕。但如果我们想后台截图呢?

所谓后台截图,就是当窗口最小化/非活动窗口时也能正确截图。比如你的软件/游戏,处于最小化状态、被其它窗口遮挡、部分区域处于屏幕外边,都可以截取到该软件/游戏此刻的图像。

调研

经过几个小时的研究、测试,整理出了如下3种方式,完整代码如下:

import win32gui
import win32ui
from PIL import Image
from ctypes import windll, byref, c_ubyte
from ctypes.wintypes import RECT, HWND
import numpy as np
import win32con


def fun1():
    hwnd = win32gui.FindWindow(None, '控制面板')

    #left, top, right, bot = win32gui.GetClientRect(hwnd)
    left, top, right, bot = win32gui.GetWindowRect(hwnd)
    w = right - left
    h = bot - top

    hwndDC = win32gui.GetWindowDC(hwnd)

    mfcDC = win32ui.CreateDCFromHandle(hwndDC)
    saveDC = mfcDC.CreateCompatibleDC()

    saveBitMap = win32ui.CreateBitmap()
    saveBitMap.CreateCompatibleBitmap(mfcDC, w, h)

    saveDC.SelectObject(saveBitMap)

    # result = windll.user32.PrintWindow(hwnd, saveDC.GetSafeHdc(), 1)
    result = windll.user32.PrintWindow(hwnd, saveDC.GetSafeHdc(), 0)

    bmpinfo = saveBitMap.GetInfo()
    bmpstr = saveBitMap.GetBitmapBits(True)

    im = Image.frombuffer(
        'RGB',
        (bmpinfo['bmWidth'], bmpinfo['bmHeight']),
        bmpstr, 'raw', 'BGRX', 0, 1)

    win32gui.DeleteObject(saveBitMap.GetHandle())
    saveDC.DeleteDC()
    mfcDC.DeleteDC()
    win32gui.ReleaseDC(hwnd, hwndDC)

    if result == 1:
        im.save("test1.png")


def fun2():
    handle = windll.user32.FindWindowW(None, "控制面板")
    # 排除缩放干扰
    windll.user32.SetProcessDPIAware()
    # 获取窗口客户区的大小
    r = RECT()
    windll.user32.GetClientRect(handle, byref(r))
    width, height = r.right, r.bottom
    print(width,height)
    # 开始截图
    dc = windll.user32.GetDC(handle)
    cdc = windll.gdi32.CreateCompatibleDC(dc)
    bitmap = windll.gdi32.CreateCompatibleBitmap(dc, width, height)
    windll.gdi32.SelectObject(cdc, bitmap)
    windll.gdi32.BitBlt(cdc, 0, 0, width, height, dc, 0, 0, win32con.SRCCOPY)
    # 截图是BGRA排列,因此总元素个数需要乘以4
    total_bytes = width * height * 4
    buffer = bytearray(total_bytes)
    # 要与c交互,需要创建c语言专用的数组类型byte_array
    byte_array = c_ubyte * total_bytes
    windll.gdi32.GetBitmapBits(bitmap, total_bytes, byte_array.from_buffer(buffer))
    windll.gdi32.DeleteObject(bitmap)
    windll.gdi32.DeleteObject(cdc)
    windll.user32.ReleaseDC(handle, dc)
    arr = np.frombuffer(buffer, dtype=np.uint8)
    image = Image.frombuffer(
        'RGB',
        (width, height),
        arr, 'raw', 'BGRX', 0, 1)
    image.save('test2.bmp')

def fun3():
    hwnd = windll.user32.FindWindowW(None, "控制面板")
    left, top, right, bot = win32gui.GetWindowRect(hwnd)
    width = right - left
    height = bot - top
    wDC = win32gui.GetWindowDC(hwnd)
    dcObj = win32ui.CreateDCFromHandle(wDC)
    cDC = dcObj.CreateCompatibleDC()
    dataBitMap = win32ui.CreateBitmap()
    dataBitMap.CreateCompatibleBitmap(dcObj, width, height)
    cDC.SelectObject(dataBitMap)
    cDC.BitBlt((0, 0), (width, height), dcObj, (0, 0), win32con.SRCCOPY)
    dataBitMap.SaveBitmapFile(cDC, 'test3.bmp')
    dcObj.DeleteDC()
    cDC.DeleteDC()
    win32gui.ReleaseDC(hwnd, wDC)
    win32gui.DeleteObject(dataBitMap.GetHandle())

if __name__ == "__main__":
    fun1()
    fun2()
    fun3()


如上,三种方法分别是fun1,fun2,fun3。

fun1主要步骤:

  • 使用 win32gui.GetWindowRect() 函数获取窗口的矩形区域;
  • 使用 win32gui.GetWindowDC() 函数获取窗口的设备上下文句柄;
  • 使用 win32ui.CreateDCFromHandle() 函数创建设备上下文对象;
  • 使用 win32ui.CreateCompatibleBitmap() 函数创建兼容位图对象;
  • 使用 windll.user32.PrintWindow() 函数将窗口内容复制到位图中;
  • 使用 Image.frombuffer() 函数将位图数据转换为 PIL.Image 对象;
  • 最后释放资源。

fun2主要步骤:

  • 使用 windll.user32.FindWindowW() 函数获取窗口句柄;
  • 使用 windll.user32.SetProcessDPIAware() 函数排除缩放干扰;
  • 使用 windll.user32.GetClientRect() 函数获取窗口客户区的大小;
  • 使用 windll.user32.GetDC() 函数获取窗口的设备上下文句柄;
  • 使用 windll.gdi32.CreateCompatibleDC() 函数创建兼容设备上下文对象;
  • 使用 windll.gdi32.CreateCompatibleBitmap() 函数创建兼容位图对象;
  • 使用 windll.gdi32.BitBlt() 函数将窗口内容复制到位图中;
  • 使用 windll.gdi32.GetBitmapBits() 函数获取位图数据;
  • 最后将位图数据转换为 PIL.Image 对象并保存。

fun3主要步骤:

  • 使用 windll.user32.FindWindowW() 函数获取窗口句柄;
  • 使用 win32gui.GetWindowRect() 函数获取窗口的矩形区域;
  • 使用 win32gui.GetWindowDC() 函数获取窗口的设备上下文句柄;
  • 使用 win32ui.CreateDCFromHandle() 函数创建设备上下文对象;
  • 使用 win32ui.CreateCompatibleBitmap() 函数创建兼容位图对象;
  • 使用 win32ui.CreateBitmap().SaveBitmapFile() 函数将窗口内容保存到位图文件中;
  • 最后释放资源。

从上面的分析可以看出,三个函数的实现方式有些不同,但都是通过获取窗口的设备上下文句柄,然后将窗口内容复制到位图中,最后将位图数据转换为 PIL.Image 对象并保存。其中,fun2() 函数使用的是 ctypes 库来调用 Windows API 函数,而 fun1() 和 fun3() 函数使用的是 win32gui 和 win32ui 库来实现。在使用时,可以根据实际情况选择合适的函数。

重要代码讲解:

1、win32gui.GetWindowRect(hwnd)与win32gui.GetClientRect(hwnd)

GetWindowRect:是获取程序整个窗口,包含标题栏,滚动条等。

GetClientRect:是获取客户区域,客户区域就是去掉标题栏、滚动条等剩余的区域

2、windll.user32.PrintWindow(hwnd, saveDC.GetSafeHdc(), 0)

最后一个参数传0表示复制整个窗口,传1表示复制客户区域。似乎可以尝试传2、3看看是什么效果。

3、im = Image.frombuffer('RGB',(bmpinfo['bmWidth'], bmpinfo['bmHeight']),bmpstr, 'raw', 'BGRX', 0, 1)

由于抓取的图是BGRX格式,直接保存颜色会出错,这行代码是将其转为RGB格式存储。

4、windll.user32.SetProcessDPIAware()

如果截图不正常,可能是由于系统DPI缩放导致的,加上这个代码再试试

5、windll.gdi32.BitBlt(cdc, 0, 0, width, height, dc, 0, 0, win32con.SRCCOPY)

windll.gdi32.BitBlt() 是 Windows GDI(图形设备接口)中的一个函数,用于将一个设备上的位图复制到另一个设备上。该函数可以将一个设备上的位图复制到另一个设备上。在给定的参数中,dc 是源设备的句柄,0 和 0 表示源设备中位图的左上角坐标为原点,width 和 height 表示要复制的位图的宽度和高度,cdc 是目标设备的句柄,SRCCOPY 表示直接复制源位图到目标位图。

在实际应用中,windll.gdi32.BitBlt() 函数常用于将一个窗口或控件的内容复制到另一个窗口或控件中,或者将一个设备上的位图复制到另一个设备上,例如将一个图像显示在屏幕上或者打印到打印机上。需要注意的是,windll.gdi32.BitBlt() 函数的性能较高,但使用不当可能会导致图像失真或者出现其他问题。

结果展示及结论

测试控制面板:

测试eclipse软件:

测试浏览器:

我先测试了Edge,Chrome,但都是黑图,猜测可能是uwp版本,程序无法截图。随后我尝试了360极速浏览器,终于正常截取到了。由于360浏览器自己把win的边框去掉了,所以三个方法截取的图片是一样的。

测试计算器:

三张均为黑图,我就不上传了,因为win10的计算器是uwp实现的,程序无法截取。

结论

1、当窗口被遮挡或处于屏幕外边时,可以正确截图。当窗口最小化时,不能正确截图。

2、fun1与fun3结果相同,包含标题栏,fun2未包含标题栏。

3、fun1和fun2生成的文件较小,fun3文件较大,约是前者的4-20倍。

4、uwp的程序,无法使用该方式截图。

相关推荐

让 Python 代码飙升330倍:从入门到精通的四种性能优化实践

花下猫语:性能优化是每个程序员的必修课,但你是否想过,除了更换算法,还有哪些“大招”?这篇文章堪称典范,它将一个普通的函数,通过四套组合拳,硬生生把性能提升了330倍!作者不仅展示了“术”,更传授...

7 段不到 50 行的 Python 脚本,解决 7 个真实麻烦:代码、场景与可复制

“本文整理自开发者AbdurRahman在Stackademic的真实记录,所有代码均经过最小化删减,确保在50行内即可运行。每段脚本都对应一个日常场景,拿来即用,无需额外依赖。一、在朋...

Python3.14:终于摆脱了GIL的限制

前言Python中最遭人诟病的设计之一就是GIL。GIL(全局解释器锁)是CPython的一个互斥锁,确保任何时刻只有一个线程可以执行Python字节码,这样可以避免多个线程同时操作内部数据结...

Python Web开发实战:3小时从零搭建个人博客

一、为什么选Python做Web开发?Python在Web领域的优势很突出:o开发快:Django、Flask这些框架把常用功能都封装好了,不用重复写代码,能快速把想法变成能用的产品o需求多:行业...

图解Python编程:从入门到精通系列教程(附全套速查表)

引言本系列教程展开讲解Python编程语言,Python是一门开源免费、通用型的脚本编程语言,它上手简单,功能强大,它也是互联网最热门的编程语言之一。Python生态丰富,库(模块)极其丰富,这使...

Python 并发编程实战:从基础到实战应用

并发编程是提升Python程序效率的关键技能,尤其在处理多任务场景时作用显著。本文将系统介绍Python中主流的并发实现方式,帮助你根据场景选择最优方案。一、多线程编程(threading)核...

吴恩达亲自授课,适合初学者的Python编程课程上线

吴恩达教授开新课了,还是亲自授课!今天,人工智能著名学者、斯坦福大学教授吴恩达在社交平台X上发帖介绍了一门新课程——AIPythonforBeginners,旨在从头开始讲授Python...

Python GUI 编程:tkinter 初学者入门指南——Ttk 小部件

在本文中,将介绍Tkinter.ttk主题小部件,是常规Tkinter小部件的升级版本。Tkinter有两种小部件:经典小部件、主题小部件。Tkinter于1991年推出了经典小部件,...

Python turtle模块编程实践教程

一、模块概述与核心概念1.1turtle模块简介定义:turtle是Python标准库中的2D绘图模块,基于Logo语言的海龟绘图理念实现。核心原理:坐标系系统:原点(0,0)位于画布中心X轴:向右...

Python 中的asyncio 编程入门示例-1

Python的asyncio库是用于编写并发代码的,它使用async/await语法。它为编写异步程序提供了基础,通过非阻塞调用高效处理I/O密集型操作,适用于涉及网络连接、文件I/O...

30天学会Python,开启编程新世界

在当今这个数字化无处不在的时代,Python凭借其精炼的语法架构、卓越的性能以及多元化的应用领域,稳坐编程语言排行榜的前列。无论是投身于数据分析、人工智能的探索,还是Web开发的构建,亦或是自动化办公...

Python基础知识(IO编程)

1.文件读写读写文件是Python语言最常见的IO操作。通过数据盘读写文件的功能都是由操作系统提供的,读写文件就是请求操作系统打开一个文件对象(通常称为文件描述符),然后,通过操作系统提供的接口从这个...

Python零基础到精通,这8个入门技巧让你少走弯路,7天速通编程!

Python学习就像玩积木,从最基础的块开始,一步步搭建出复杂的作品。我记得刚开始学Python时也是一头雾水,走了不少弯路。现在回头看,其实掌握几个核心概念,就能快速入门这门编程语言。来聊聊怎么用最...

一文带你了解Python Socket 编程

大家好,我是皮皮。前言Socket又称为套接字,它是所有网络通信的基础。网络通信其实就是进程间的通信,Socket主要是使用IP地址,协议,端口号来标识一个进程。端口号的范围为0~65535(用户端口...

Python-面向对象编程入门

面向对象编程是一种非常流行的编程范式(programmingparadigm),所谓编程范式就是程序设计的方法论,简单的说就是程序员对程序的认知和理解以及他们编写代码的方式。类和对象面向对象编程:把...

取消回复欢迎 发表评论: