您当前的位置:首页 > 计算机 > 编程开发 > Python

Python实现计算机屏幕任意区域截图

时间:09-07来源:作者:点击数:

程序功能与用法:运行后有个主窗体,上面有个按钮,单击后开始截图,鼠标坐标落下开始截图,鼠标左键抬起表示截图结束,然后弹出对话框提示保存截图文件。

本文要点在于Python扩展库pillow提供的ImageGrab支持全屏幕或指定区域的截图。

import tkinter

import tkinter.filedialog

import os

from PIL import ImageGrab

from time import sleep

#创建tkinter主窗口

root = tkinter.Tk()

#指定主窗口位置与大小

root.geometry('100x40+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)

        #屏幕尺寸

        screenWidth = root.winfo_screenwidth()

        screenHeight = root.winfo_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.image = tkinter.PhotoImage(file=png)

        self.canvas.create_image(screenWidth//2, screenHeight//2, image=self.image)

        #鼠标左键按下的位置

        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)

            #考虑鼠标左键从右下方按下而从左上方抬起的截图

            left, right = sorted([self.X.get(), event.x])

            top, bottom = sorted([self.Y.get(), event.y])

            pic = ImageGrab.grab((left+1, top+1, right, bottom))

            #弹出保存截图对话框

            fileName = tkinter.filedialog.asksaveasfilename(title='保存截图', filetypes=[('JPG files', '*.jpg')])

            if fileName:

                pic.save(fileName+'.jpg')

            #关闭当前窗口

            self.top.destroy()

        self.canvas.bind('<ButtonRelease-1>', onLeftButtonUp)

        self.canvas.pack(fill=tkinter.BOTH, expand=tkinter.YES)

    #开始截图

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)

    #截图结束,恢复主窗口,并删除临时的全屏幕截图文件

    root.state('normal')

    os.remove(filename)

buttonCapture = tkinter.Button(root, text='截图', command=buttonCaptureClick)

buttonCapture.place(x=10, y=10, width=80, height=20)

#启动消息主循环

root.mainloop()

方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门