win32gui在python3.7中截屏会导致内存泄漏

Posted 2023-02-21

【中文标题】win32gui在python3.7中截屏会导致内存泄漏

【英文标题】：Get screnn shot in python3.7 by win32gui will cause memory leak

【发布时间】：2020-09-29 10:06:14

【问题描述】：

Win10、python3.7 我的代码是这样的：

import win32gui, win32ui, win32con, win32api

screen_w = 2880
screen_h = 1800
x0 = int(screen_w * 0.866)
y0 = int(screen_h * 0.14)
ww = 220
hh = 70

def window_capture(filename): 
    hwnd = 0
    hwndDC = win32gui.GetWindowDC(hwnd)
    mfcDC = win32ui.CreateDCFromHandle(hwndDC)
    saveDC = mfcDC.CreateCompatibleDC()
    saveBitMap = win32ui.CreateBitmap()
    saveBitMap.CreateCompatibleBitmap(mfcDC, ww, hh)
    saveDC.SelectObject(saveBitMap)
    saveDC.BitBlt((0, 0), (ww, hh), mfcDC, (x0, y0), win32con.SRCCOPY)
    saveBitMap.SaveBitmapFile(saveDC, filename)
    #------------------
    win32gui.DeleteObject(saveBitMap.GetHandle())
    saveDC.DeleteDC()
    mfcDC.DeleteDC()
    win32gui.ReleaseDC(hwnd, hwndDC)
    #------------------

count = 0
while 1:
    window_capture('temp.png')
    count += 1
    if int(count/1000) == float(count/1000):
        print(count)

首先，我没有添加 DeleteObject 和 DeleteDC 来释放内存。该过程将在大约 10000 次 window_capture 后停止。 但是在我添加它们之后，我仍然可以在任务管理器中看到内存增加。最后，在大约 130000 次之后，该过程再次停止。

这是否意味着我还没有释放所有的记忆？在我的项目中，我需要长时间捕获屏幕截图。我该如何解决这个问题？

谢谢！

【参考方案1】：

警告：我不懂 Python。

从 Win32 的角度来看，当 GDI 对象仍被选择到任何 DC（设备上下文）时，您不应该删除它。您在这里正是这样做的：

win32gui.DeleteObject(saveBitMap.GetHandle())

典型的 Win32 方案是将先前选择的对象的句柄保存到 DC 中（任何Select...() 函数都会返回该句柄），并在您完成自己的对象后将其重新选择到 DC 中。

如果您需要恢复多个对象，也可以使用SaveDC / RestoreDC 对。