用python实现选择截图区域
- 2020 年 1 月 10 日
- 筆記
一直想用python实现一个类似QQ截图的功能，但不直接截图，而是返回截图的区域，以下是代码
# -*- coding:utf-8 -*- import tkinter import tkinter.filedialog import os from PIL import ImageGrab from time import sleep from tkinter import StringVar, IntVar #创建tkinter主窗口 root = tkinter.Tk() #指定主窗口位置与大小 root.geometry('200x80+400+300') #不允许改变窗口大小 root.resizable(False, False) class MyCapture: def __init__(self, png): #变量X和Y用来记录鼠标左键按下的位置 self.X = tkinter.IntVar(value=0) self.Y = tkinter.IntVar(value=0) self.selectPosition=None #屏幕尺寸 screenWidth = root.winfo_screenwidth() #print(screenWidth) screenHeight = root.winfo_screenheight() #print(screenHeight) #创建顶级组件容器 self.top = tkinter.Toplevel(root, width=screenWidth, height=screenHeight) #不显示最大化、最小化按钮 self.top.overrideredirect(True) self.canvas = tkinter.Canvas(self.top,bg='white', width=screenWidth, height=screenHeight) #显示全屏截图，在全屏截图上进行区域截图 self.p_w_picpath = tkinter.PhotoImage(file=png) self.canvas.create_p_w_picpath(screenWidth//2, screenHeight//2, p_w_picpath=self.p_w_picpath) #鼠标左键按下的位置 def onLeftButtonDown(event): self.X.set(event.x) self.Y.set(event.y) #开始截图 self.sel = True self.canvas.bind('<Button-1>', onLeftButtonDown) #鼠标左键移动，显示选取的区域 def onLeftButtonMove(event): if not self.sel: return global lastDraw try: #删除刚画完的图形，要不然鼠标移动的时候是黑乎乎的一片矩形 self.canvas.delete(lastDraw) except Exception as e: pass lastDraw = self.canvas.create_rectangle(self.X.get(), self.Y.get(), event.x, event.y, outline='black') self.canvas.bind('<B1-Motion>', onLeftButtonMove) #获取鼠标左键抬起的位置，保存区域截图 def onLeftButtonUp(event): self.sel = False try: self.canvas.delete(lastDraw) except Exception as e: pass sleep(0.1) #考虑鼠标左键从右下方按下而从左上方抬起的截图 myleft, myright = sorted([self.X.get(), event.x]) mytop, mybottom = sorted([self.Y.get(), event.y]) self.selectPosition=(myleft,myright,mytop,mybottom) # pic = ImageGrab.grab((left+1, top+1, right, bottom)) # # #弹出保存截图对话框 # # fileName = tkinter.filedialog.asksaveasfilename(title='保存截图', filetypes=[('JPG files', '*.jpg')]) # # if fileName: # # pic.save(fileName+'.jpg') #关闭当前窗口 #print(left, ' ', top,' ',right,' ',bottom) self.top.destroy() self.canvas.bind('<ButtonRelease-1>', onLeftButtonUp) self.canvas.pack(fill=tkinter.BOTH, expand=tkinter.YES) #开始截图 text = StringVar() text.set('old') def buttonCaptureClick(): #最小化主窗口 #root.state('icon') #sleep(0.2) filename = 'temp.png' im = ImageGrab.grab() im.save(filename) im.close() #显示全屏幕截图 w = MyCapture(filename) buttonCapture.wait_window(w.top) text.set(str(w.selectPosition)) #print(w.myleft,w.mybottom) #截图结束，恢复主窗口，并删除临时的全屏幕截图文件 #label.config(text='Hello') root.state('normal') os.remove(filename) label=tkinter.Label(root,textvariable=text) label.place(x=10, y=30, width=160, height=20) label.config(text='New test') buttonCapture = tkinter.Button(root, text='截图', command=buttonCaptureClick) buttonCapture.place(x=10, y=10, width=160, height=20) #启动消息主循环 #root.update() root.mainloop()
