《自拍教程68》Python調用攝像頭？ 幹嘛？ 乾正事！！！

案例故事:
場景一：反覆重啟Android終端產品100次，每重啟一次錄一個視頻;
場景二：做壓力測試比如Monkey一晚上，我們需要涉及長時間錄像;
場景三：做自動化測試的時候，跑一條自動化用例，錄製每條用例執行的整個過程視頻。
許多需要長時間的壓測的場景，我們都可以通過攝像頭錄像記錄下被測試設備的運行情況，


本篇主要介紹如何通過Python + ffmpeg + 電腦攝像頭實現錄像功能。

準備階段

1. ffmpeg的下載地址及其介紹可參考：ffmpeg - 音視頻圖像編解碼工具
   

2. 確保ffmpeg.exe已經加入Path環境變數里的某個路徑下。

3. 確保攝像頭驅動已經Ok，在Windows設備管理器"圖像設備"里可以找到這個設備。
   
   
   

Python面向對象類形式

由於涉及設備初始化，開始錄像，停止錄像等功能。
更建議直接用面向對象類形式來定義這個模塊。

# coding=utf-8

import os
import time
import subprocess


class FFmpegRecorder(object):
    def __init__(self, camera_device="None"):
        self.__restore_ffmpeg() # 先重置ffmpeg
        self.camera_device = camera_device
        self.__ffmpeg_p = None

    def __restore_ffmpeg(self):
        '''防止ffmpeg正在錄製視頻，導致視頻設備被占用'''
        os.system("taskkill /im ffmpeg.exe /f")

    def start_record(self, output_file):
        command = "ffmpeg -f dshow -s 640x480 -i video=\"%s\" -vcodec libxvid -y %s" % (self.camera_device, output_file)
        self.__ffmpeg_p = subprocess.Popen(command)

    def stop_record(self):
        if self.__ffmpeg_p != None:
            self.__ffmpeg_p.terminate()
            time.sleep(1)
        else:
            print("please run start_record() firstly")


if __name__ == "__main__":
    CAMERA_DEVICE = "USB "  # 常量， 自行去Windows設備管理器查看
    f_obj = FFmpegRecorder(CAMERA_DEVICE)
    for i in range(100):
        f_obj.start_record("%s.avi" % i)
        time.sleep(20) #錄製一個20s的視頻
        f_obj.stop_record()
        time.sleep(1)

運行效果視頻

視頻上傳失敗，跳轉至官網查看本視頻效果

擴展閱讀

以上案例只能錄製無聲視頻，如果想錄製有聲視頻，
且想分別在Window操作系統及Linux操作系統端實現，可參考以下命令：

小提示： 因為錄製的視頻是xvid視頻編碼, 尾碼是.avi的文件，
視頻播放過程中，如果出現拖動進度條無效或異常，可以考慮使用VLC player,
很推薦各位音視頻測試的同學使用這款播放器，支持視頻播放，支持線上流媒體播放。

小提示： ffmpeg從攝像頭獲取圖片並編碼的過程中，如果經常出現error或者frame dropped (丟幀），
可以考慮降低解析度，比如從1920x1080降低到640x480的解析度。

更多更好的原創文章，請訪問官方網站：www.zipython.com
自拍教程（自動化測試Python教程，武散人編著）
原文鏈接：https://www.zipython.com/#/detail?id=d460b2754b5b45edad1dc7d2706cf94e
也可關註“武散人”微信訂閱號，隨時接受文章推送。