介绍一种能自动截图的python代码

博主在做毕业设计的时候总是需要截取自己做的网页的图，但是用各种截图方法（QQ、微信、Ctrl+Alt+S）总是很不方便，因此我就想能不能写一个代码，当我按压键盘某一个键的时候能触发这个截图功能？说干就干，我做了如下思考。

问：怎么保证我一点击的时候就截图，不点击就不截图呢？

答：用while循环，一直监听是否触发某个事件，一旦触发就执行代码。

编不下去了，其实我没有思考，就直接写了，有一部分也是copy别人的，我只是在上面做了改动，效果是，运行代码后就能保存Svg格式的图和png格式的图。个人感觉png比svg更清晰。

代码如下：

import cv2
import time
import os
import keyboard
from PIL import ImageGrab
import numpy as np
import matplotlib.pyplot as plt

# 获取当前目录
BASE_DIR = os.getcwd()


# 截屏函数
def screenshot(event):
    if event.name == 'y':
        local_tm = time.localtime(time.time())
        # 格式化日期: 年月日
        date_str = time.strftime("%Y%m%d", local_tm)
        # 以日期建立文件夹
        dir_name = "Screenshot_" + date_str
        dir_path = os.path.join(BASE_DIR, dir_name)
        # 不存在这一天的文件夹, 则建立它
        if not os.path.exists(dir_path):
            os.mkdir(dir_path)
        # 格式化时间: 年月日_时分秒
        YMD_HMS_str = time.strftime("%Y%m%d_%H%M%S", local_tm)
        # 文件保存为.jpg (也可以是png或者其他格式)
        file_name = YMD_HMS_str + ".svg"
        file_path = os.path.join(dir_path, file_name)
        # 截屏, 抓取当前屏幕的快照，返回一个模式为"RGB"的图像
        im = ImageGrab.grab()
        im.save(file_path.replace(".svg", ".png"),dpi=(300, 300))
        plt.imshow(np.asarray(im))
        plt.axis('off')  # 不显示坐标轴y
        plt.savefig(file_path, dpi=300, format='svg',bbox_inches='tight')  # 图片保存的模式是紧凑的，即图片两边没有多余的空白
        # im.save(file_path)
        # 输出已经保存的截图文件, 以及截图尺寸 (比如: 560, 1440)
        print("save: ", file_path, " size:", im.size)


# 基于 Python Imaging Library(PIL) 的截屏实现
if __name__ == '__main__':
    # print("Now, Run Auto_Screenshot: ", BASE_DIR)
    # if cv2.WaitKey(1) == ord('y'):
    #     screenshot()
    keyboard.on_press(screenshot)
    keyboard.wait()