最全的python截屏方法

#!/usr/bin/env python #coding=utf-8  
import os import platform 
import tempfile 
import shutil  
from PIL import Image  
PATH = lambda p: os.path.abspath(p) 
TEMP_FILE = PATH(tempfile.gettempdir() + "/temp_screen.png")  
class Appium_Extend(object):     
def __init__(self, driver):         
self.driver = driver  
    
def get_screenshot_by_element(self, element):        
 #先截取整个屏幕，存储至系统临时目录下         
self.driver.get_screenshot_as_file(TEMP_FILE)          
#获取元素bounds         
location = element.location         
size = element.size         
box = (location["x"], location["y"], location["x"] + size["width"], location["y"] + size["height"])          
#截取图片         
image = Image.open(TEMP_FILE)         
newImage = image.crop(box)         
newImage.save(TEMP_FILE)          
return self     
 
def get_screenshot_by_custom_size(self, start_x, start_y, end_x, end_y):         
#自定义截取范围         
self.driver.get_screenshot_as_file(TEMP_FILE)         
box = (start_x, start_y, end_x, end_y)         
image = Image.open(TEMP_FILE)         
newImage = image.crop(box)         
newImage.save(TEMP_FILE)          
return self 
     
def write_to_file( self, dirPath, imageName, form = "png"):         
#将截屏文件复制到指定目录下         
if not os.path.isdir(dirPath):             
os.makedirs(dirPath)         
shutil.copyfile(TEMP_FILE, PATH(dirPath + "/" + imageName + "." + form)) 
     
def load_image(self, image_path):         
#加载目标图片供对比用         
if os.path.isfile(image_path):             
load = Image.open(image_path)             
return load         
else:             
raise Exception("%s is not exist" %image_path)      
#http://testerhome.com/topics/202     


def same_as(self, load_image, percent):         
#对比图片，percent值设为0，则100%相似时返回True，设置的值越大，相差越大         
import math         
import operator          
image1 = Image.open(TEMP_FILE)         
image2 = load_image         
 histogram1 = image1.histogram()         
histogram2 = image2.histogram()          
differ = math.sqrt(reduce(operator.add, list(map(lambda a,b: (a-b)**2,    histogram1, histogram2)))/len(histogram1))        
 if differ <= percent:             
return True         
else:             
return False