使用Python實現(xiàn)屏幕錄制與鍵盤監(jiān)聽功能

更新時間：2024年12月26日 10:49:26 作者：Leuanghing

在Python中,我們可以借助多個強大的庫來實現(xiàn)豐富的功能,比如屏幕錄制和鍵盤監(jiān)聽,今天,我們將通過結合PIL、OpenCV、pynput以及threading等庫,來實現(xiàn)一個簡單的屏幕錄制工具,該工具能夠通過監(jiān)聽鍵盤事件來控制錄制的開始與結束,需要的朋友可以參考下

一、引言

在Python中，我們可以借助多個強大的庫來實現(xiàn)豐富的功能，比如屏幕錄制和鍵盤監(jiān)聽。今天，我們將通過結合PIL（Python Imaging Library的分支Pillow）、OpenCV、pynput以及threading等庫，來實現(xiàn)一個簡單的屏幕錄制工具，該工具能夠通過監(jiān)聽鍵盤事件來控制錄制的開始與結束。

二、所需庫介紹

PIL（Pillow）：用于屏幕截圖。
OpenCV：用于視頻文件的創(chuàng)建與寫入。
pynput：用于監(jiān)聽鍵盤事件。
threading：用于并行執(zhí)行代碼，實現(xiàn)屏幕錄制與鍵盤監(jiān)聽的同時進行。

三、代碼詳解

導入所需庫

import time
import threading
from datetime import datetime
from PIL import ImageGrab
import numpy as np
from cv2.cv2 import cvtColor, VideoWriter_fourcc, VideoWriter, COLOR_RGB2BGR, VideoCapture, \
                  CAP_PROP_FPS, CAP_PROP_FRAME_COUNT, CAP_PROP_FRAME_WIDTH, CAP_PROP_FRAME_HEIGHT
from pynput import keyboard

定義視頻錄制函數(shù)

def video_record():
    # ...（省略了部分代碼）
    while True:
        if flag:
            # ...（省略了部分代碼）
            break
        im = ImageGrab.grab()  # 截圖
        imm = cvtColor(np.array(im), COLOR_RGB2BGR)  # 轉換顏色模式
        video.write(imm)  # 寫入視頻文件

這個函數(shù)首先獲取當前時間作為文件名，然后初始化視頻寫入對象。接著，它會顯示一個3秒的倒計時，之后開始錄制屏幕。在錄制過程中，它會不斷檢查一個全局變量flag，如果flag變?yōu)?code>True，則停止錄制并釋放視頻寫入對象。

定義鍵盤監(jiān)聽函數(shù)

def on_press(key):
    global flag
    if key == keyboard.Key.esc:
        flag = True
        return False

這個函數(shù)會監(jiān)聽鍵盤事件，當按下esc鍵時，將全局變量flag設置為True，并返回False以停止鍵盤監(jiān)聽。

定義視頻信息打印函數(shù)

def video_info():
    # ...（省略了部分代碼）
    print('幀率=%.1f' % fps)
    # ...（省略了部分代碼）
    print('推薦幀率=%.2f' % (fps*((int(count)/fps)/(final_time-start_time))))

這個函數(shù)會在錄制結束后讀取視頻文件的信息，包括幀率、幀數(shù)、分辨率以及錄制時間等，并打印出來。同時，它還會計算并打印一個推薦幀率，這個推薦幀率是基于實際錄制時間與視頻文件中的幀數(shù)計算得出的。

主函數(shù)

if __name__ == '__main__':
    flag = False
    th = threading.Thread(target=video_record)
    th.start()

    with keyboard.Listener(on_press=on_press) as listener:
        listener.join()

    time.sleep(1)  # 等待視頻釋放過后
    video_info()

在主函數(shù)中，我們首先初始化全局變量flag為False，然后創(chuàng)建一個線程來運行video_record函數(shù)。接著，我們使用pynput庫的keyboard.Listener來監(jiān)聽鍵盤事件，并將on_press函數(shù)作為回調函數(shù)。當鍵盤監(jiān)聽結束時（即按下esc鍵后），主線程會等待1秒以確保視頻文件被正確釋放，然后調用video_info函數(shù)來打印視頻信息。

四、完整代碼

#!/usr/bin/env python
# -*- coding = utf-8 -*-
# @Time : 2021/10/6 17:43
# @Author : Leuanghing Chen
# @Blog : https://blog.csdn.net/weixin_46153372?spm=1010.2135.3001.5421
# @File : python實現(xiàn)錄屏功能.py
# @Software : PyCharm
# 發(fā)布程序：pyinstaller -F -w -i F:\python_demo\2.ico F:\python_demo\python實現(xiàn)錄屏功能.py

import time
import threading
from datetime import datetime
from PIL import ImageGrab
import numpy as np
from cv2.cv2 import cvtColor, VideoWriter_fourcc, VideoWriter, COLOR_RGB2BGR, VideoCapture,\
                    CAP_PROP_FPS, CAP_PROP_FRAME_COUNT, CAP_PROP_FRAME_WIDTH, CAP_PROP_FRAME_HEIGHT
from pynput import keyboard


# 錄入視頻
def video_record():
    # 加了global，則可以在函數(shù)內部對函數(shù)外的對象進行操作了，也可以改變它的值了
    global name
    name = datetime.now().strftime('%Y-%m-%d %H-%M-%S')               # 當前的時間（當文件名）
    screen = ImageGrab.grab()                                         # 獲取當前屏幕
    width, high = screen.size                                         # 獲取當前屏幕的大小
    fourcc = VideoWriter_fourcc('X', 'V', 'I', 'D')                   # MPEG-4編碼,文件后綴可為.avi .asf .mov等
    video = VideoWriter('%s.avi' % name, fourcc, 15, (width, high))   # （文件名，編碼器，幀率，視頻寬高）
    for i in range(3):
        time.sleep(1)
        print("倒計時：{}".format(3-i))
    print('開始錄制!')
    global start_time
    start_time = time.time()

    while True:
        if flag:
            print("錄制結束！")
            global final_time
            final_time = time.time()
            video.release()                                     # 釋放
            break
        im = ImageGrab.grab()  # 圖片為RGB模式
        imm = cvtColor(np.array(im), COLOR_RGB2BGR)             # 轉為opencv的BGR模式
        video.write(imm)                                        # 寫入
        # time.sleep(5)                                         # 等待5秒再次循環(huán)


# 結束監(jiān)聽按鍵，按下esc結束錄屏
def on_press(key):
    global flag
    # esc鍵
    if key == keyboard.Key.esc:
        flag = True                                             # 改變
        return False                                            # 返回False，鍵盤監(jiān)聽結束！


# 視頻信息
def video_info():
    video = VideoCapture('%s.avi' % name)                       # 記得文件名加格式不要錯！
    fps = video.get(CAP_PROP_FPS)
    count = video.get(CAP_PROP_FRAME_COUNT)
    size = (int(video.get(CAP_PROP_FRAME_WIDTH)), int(video.get(CAP_PROP_FRAME_HEIGHT)))
    print('幀率=%.1f' % fps)
    print('幀數(shù)=%.1f' % count)
    print('分辨率', size)
    print('視頻時間=%.3f秒' % (int(count)/fps))
    print('錄制時間=%.3f秒' % (final_time-start_time))
    print('推薦幀率=%.2f' % (fps*((int(count)/fps)/(final_time-start_time))))


if __name__ == '__main__':
    flag = False
    th = threading.Thread(target=video_record)
    th.start()

    with keyboard.Listener(on_press=on_press) as listener:
        listener.join()

    time.sleep(1)  # 等待視頻釋放過后
    video_info()

五、總結

通過結合多個Python庫，我們實現(xiàn)了一個簡單的屏幕錄制工具，該工具能夠實時錄制屏幕并通過監(jiān)聽鍵盤事件來控制錄制的開始與結束。這個工具不僅可以用于個人娛樂或學習，還可以作為開發(fā)屏幕錄制功能應用程序的基礎。希望這篇博文能幫助你更好地理解如何使用Python來實現(xiàn)屏幕錄制與鍵盤監(jiān)聽功能。

以上就是使用Python實現(xiàn)屏幕錄制與鍵盤監(jiān)聽功能的詳細內容，更多關于Python屏幕錄制與鍵盤監(jiān)聽的資料請關注腳本之家其它相關文章！

您可能感興趣的文章: