# coding=utf-8
import time
from selenium import webdriver
driver = webdriver.Chrome()
driver.maximize_window()
driver.implicitly_wait(6)
driver.get("https://www.baidu.com")
time.sleep(1)
driver.get_screenshot_as_file("C:\\baidu.png")
driver.quit()

import time
from PIL import ImageGrab
if __name__=="__main__": 
  pic = ImageGrab.grab()
  pic.save("picture.jpg")

#coding=utf-8
import time
import win32gui, win32ui, win32con, win32api
def window_capture(filename):
 hwnd = 0 # 窗口的編號(hào)，0號(hào)表示當(dāng)前活躍窗口
 # 根據(jù)窗口句柄獲取窗口的設(shè)備上下文DC（Divice Context）
 hwndDC = win32gui.GetWindowDC(hwnd)
 
 # 根據(jù)窗口的DC獲取mfcDC
 mfcDC = win32ui.CreateDCFromHandle(hwndDC)
 
 # mfcDC創(chuàng)建可兼容的DC
 saveDC = mfcDC.CreateCompatibleDC()
 
 # 創(chuàng)建bigmap準(zhǔn)備保存圖片
 saveBitMap = win32ui.CreateBitmap()
 
 # 獲取監(jiān)控器信息
 MoniterDev = win32api.EnumDisplayMonitors(None, None)
 w = MoniterDev[0][2][2]
 h = MoniterDev[0][2][3]
 
 # print w,h　　　#圖片大小
 # 為bitmap開(kāi)辟空間
 saveBitMap.CreateCompatibleBitmap(mfcDC, w, h)
 
 # 高度saveDC，將截圖保存到saveBitmap中
 saveDC.SelectObject(saveBitMap)
 
 # 截取從左上角（0，0）長(zhǎng)寬為（w，h）的圖片
 saveDC.BitBlt((0, 0), (w, h), mfcDC, (0, 0), win32con.SRCCOPY)
 saveBitMap.SaveBitmapFile(saveDC, filename)
 
if __name__=="__main__": 
  beg = time.time()
  for i in range(1):
   image_name="haha%s.jpg"%i
   window_capture(image_name)
  end = time.time()
  print(end - beg)