学习笔记5：python实现手动截图并以时间戳命名自动保存的函数封装

只想飘一飘

已于 2023-12-29 09:30:04 修改

阅读量142

点赞数 1

文章标签： python 开发语言

于 2023-12-13 10:07:09 首次发布

原文链接：https://article.itxueyuan.com/amM798

版权

1.TK画布，无按钮，直接截图

import pyautogui
import tkinter as tk
import datetime

root = tk.Tk()
root.overrideredirect(True)         # 隐藏窗口的标题栏
root.attributes("-alpha", 0.1)      # 窗口透明度10%
# 窗口大小与位置
root.geometry("{}x{}+0+0".format(root.winfo_screenwidth(), root.winfo_screenheight()))
root.configure(bg="black")

# 再创建1个Canvas用于圈选
cv = tk.Canvas(root)
x, y = 0, 0
x_start, y_start = 0, 0
x_end, y_end = 0, 0
rec = ''


# 监听鼠标光标位置：
def move(event):
    global x, y, x_start, y_start
    new_x = (event.x-x)+cv.winfo_x()
    new_y = (event.y-y)+cv.winfo_y()
    s = "300x200+" + str(new_x)+"+" + str(new_y)
    cv.place(x=new_x - x_start, y=new_y - y_start)


# 创建子窗口：
def button_1(event):
    global x, y, x_start, y_start
    global rec
    x, y = event.x, event.y
    x_start, y_start = event.x, event.y
    x_start, y_start = event.x, event.y
    cv.configure(height=1)
    cv.configure(width=1)
    cv.config(highlightthickness=0)  # 无边框
    cv.place(x=event.x, y=event.y)
    rec = cv.create_rectangle(0, 0, 0, 0, width=0)


# 改变子窗口大小：
def b1_Motion(event):
    global x, y, x_start, y_start
    x, y = event.x, event.y
    cv.configure(height=event.y - y_start)
    cv.configure(width=event.x - x_start)
    cv.coords(rec, 0, 0, event.x-x_start, event.y-y_start)


# 松开鼠标，记录最后的光标位置：
def buttonRelease_1(event):
    global x_end, y_end
    x_end, y_end = event.x, event.y


# 截屏并保存图片
def button_3(event):
    global x_start, y_start, x_end, y_end
    cv.delete(rec)
    cv.place_forget()
    img = pyautogui.screenshot(region=[x_start, y_start, x_end-x_start, y_end-y_start])  # x,y,w,h
    timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
    img.save(f'screen_{timestamp}.png')
    sys_out(None)


# 退出程序
def sys_out(even):
    root.destroy()


def screen_by_manual():
    # 绑定事件
    cv.bind("<B1-Motion>", move)   # 鼠标左键移动->显示当前光标位置
    root.bind('<Escape>', sys_out)      # 键盘Esc键->退出
    root.bind("<Button-1>", button_1)  # 鼠标左键点击->显示子窗口
    root.bind("<B1-Motion>", b1_Motion)  # 鼠标左键移动->改变子窗口大小
    root.bind("<ButtonRelease-1>", buttonRelease_1)  # 鼠标左键释放->记录最后光标的位置
    root.bind("<Button-3>", button_3)  # 鼠标右键点击->截屏并保存图片
    root.mainloop()


if __name__ == '__main__':
    screen_by_manual()

参考链接： https://article.itxueyuan.com/amM798

2.TK按钮，点击按钮再开始截图

import tkinter
import tkinter.filedialog
import os
from PIL import ImageGrab
from time import sleep

#创建tkinter主窗口
root = tkinter.Tk()
#指定主窗口位置与大小
root.geometry('200x200+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()