win.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import time, os, signal, sys
import hashlib

import win32gui, win32ui, win32con, win32api

def save_screen(prefix, w, h):
    pass
def window_capture(dpath):
    
    # 截屏函数,调用方法window_capture('d:\\') ,参数为指定保存的目录 
    # 返回图片文件名,文件名格式:日期.jpg 如:2009328224853.jpg 
    
    hwnd = 0 
    hwndDC = win32gui.GetWindowDC(hwnd)
    mfcDC=win32ui.CreateDCFromHandle(hwndDC)
    saveDC=mfcDC.CreateCompatibleDC()
    saveBitMap = win32ui.CreateBitmap()
    MoniterDev=win32api.EnumDisplayMonitors(None,None)
    cnt = len(MoniterDev)
    print(str(cnt)+' moniter')
    if cnt == 1:
        w = MoniterDev[0][2][2]
        h = MoniterDev[0][2][3]
        #print w,h　　　＃图片大小  
        saveBitMap.CreateCompatibleBitmap(mfcDC, w, h)
        saveDC.SelectObject(saveBitMap)
        saveDC.BitBlt((0,0),(w, h) , mfcDC, (0,0), win32con.SRCCOPY)
        cc=time.gmtime()
        bmpname=str(cc[0])+str(cc[1])+str(cc[2])+str(cc[3]+8)+str(cc[4])+str(cc[5])+'.bmp'
        print(bmpname)
        saveBitMap.SaveBitmapFile(saveDC, dpath+bmpname)
    elif cnt > 1:
        i = 0
        for m in MoniterDev:
            x, y = m[2][0], m[2][1]
            w = m[2][2] - x
            h = m[2][3] - y
            saveBitMap.CreateCompatibleBitmap(mfcDC, w, h)
            saveDC.SelectObject(saveBitMap)
            # print x,y,w,h # 图片大小
            saveDC.BitBlt((0,0), (w, h), mfcDC, (x,y), win32con.SRCCOPY)
            idpath = dpath+str(i)
            if not os.path.exists(idpath):
                os.mkdir(idpath)
            cc=time.gmtime()
            bmpname= idpath+'\\'+str(cc[0])+str(cc[1])+str(cc[2])+str(cc[3]+8)+str(cc[4])+str(cc[5])+'.bmp'
            print(bmpname)
            saveBitMap.SaveBitmapFile(saveDC, bmpname)
            i += 1