截屏时提示的一个原始实现

admin

不包含截屏提示在截图中：

1. import ctypes
   
2. import win32gui
   
3. import win32con
   
4. import time
   
5. import cv2
   
6. 
   
7. # 弹出提示窗口的函数
   
8. def show_toast(message, duration=5):
   
9.     # 创建一个简单的弹出窗口
   
10.     win32gui.MessageBox(0, message, "截屏提示", win32con.MB_ICONINFORMATION)
    
11. 
    
12. # 触发截图的函数
    
13. def take_screenshot():
    
14.     # 获取屏幕尺寸
    
15.     screen = ctypes.windll.user32.GetDesktopWindow()
    
16.     screen_width = ctypes.windll.user32.GetSystemMetrics(0)
    
17.     screen_height = ctypes.windll.user32.GetSystemMetrics(1)
    
18.    
    
19.     # 创建设备上下文
    
20.     hdc = ctypes.windll.user32.GetWindowDC(screen)
    
21.     hdc_driver = ctypes.windll.gdi32.CreateCompatibleDC(hdc)
    
22.     hbitmap = ctypes.windll.gdi32.CreateCompatibleBitmap(hdc, screen_width, screen_height)
    
23.     hold_bitmap = ctypes.c_long()
    
24.     ctypes.windll.gdi32.SelectObject(hdc_driver, hbitmap)
    
25.    
    
26.     # 复制屏幕到位图
    
27.     ctypes.windll.gdi32.BitBlt(hdc_driver, 0, 0, screen_width, screen_height, hdc, 0, 0, 13369376)
    
28.    
    
29.     # 保存位图到文件
    
30.     bmp_info = ctypes.wintypes.BITMAPINFO()
    
31.     bmp_info.bmiHeader.biSize = ctypes.sizeof(bmp_info.bmiHeader)
    
32.     status = ctypes.windll.gdi32.GetDIBits(hdc_driver, hbitmap, 0, screen_height, None, ctypes.byref(bmp_info), win32con.DIB_RGB_COLORS)
    
33.    
    
34.     # 释放资源
    
35.     ctypes.windll.gdi32.DeleteObject(hbitmap)
    
36.     ctypes.windll.gdi32.DeleteDC(hdc_driver)
    
37.     ctypes.windll.user32.ReleaseDC(screen, hdc)
    
38.    
    
39.     # 读取截图数据到Python图像库
    
40.     img = cv2.imdecode(img_array, cv2.IMREAD_COLOR)
    
41.    
    
42.     # 保存图像到文件
    
43.     cv2.imwrite('screenshot.png', img)
    
44. 
    
45. # 调用函数，触发截图并显示提示窗口
    
46. show_toast("注意截屏！")
    
47. time.sleep(1)  # 等待提示窗口显示完成
    
48. take_screenshot()
    

复制代码

这个实现是最原始的复杂的，下面还有一个实现比较简单（用到Python第三方库）

admin

实现同样功能的第三方库：