1.概述

在一個測試項目中境蜕，有一個需求是要在測試過程中轧粟，播放一段living video streaming. 上網(wǎng)查了一下柿祈，可以利用ffmpeg,opencv等技術來實現(xiàn)苫昌。但是實驗后發(fā)現(xiàn)效果不太理想，而測試用例里建議是使用vlc來播放這個視頻流捕虽，故采用python-vlc模塊慨丐，來實現(xiàn)。經(jīng)測試泄私，效果滿足測試需求房揭。

2.opencv demo

雖然沒有采用opencv的方案，但是發(fā)現(xiàn)這個 demo在對http視頻流方面還是可以使用的晌端。

import numpy as np
import cv2

#cap = cv2.VideoCapture(0)
cap = cv2.VideoCapture(r'http://admin:admin@192.168.2.161:8081')
# 定義解碼器并創(chuàng)建VideoWrite對象
# linux: XVID捅暴、X264; windows:DIVX
# 20.0指定一分鐘的幀數(shù)
fourcc = cv2.VideoWriter_fourcc(*'DIVX')
out = cv2.VideoWriter('output.avi', fourcc, 20.0, (640, 480))

while (cap.isOpened()):
    ret, frame = cap.read()
    if ret == True:
        frame = cv2.flip(frame, 0)

        # 寫入幀
        out.write(frame)

        cv2.imshow('frame', frame)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
    else:
        break

# 釋放內(nèi)存
cap.release()
out.release()
cv2.destroyAllWindows()

3.python-vlc demo

這個demo就是項目中所用到的用于播放rtsp視頻流的模塊，主要難點可能會出現(xiàn)在環(huán)境搭建上咧纠。

import os
import vlc
import time
def get_player(url='rtsp://192.168.2.134:8554/live'):
    Instance = vlc.Instance()
    player = Instance.media_player_new()
    Media = Instance.media_new(url)
    Media.get_mrl()
    player.set_media(Media)
    player.play()
    return player

def get_player_state(player):
      is_playing = player.is_playing()#playing:1 ;other:0
      will_play = player.will_play()
      get_state = player.get_state()
      print 'is_playing: ',is_playing
      print 'will_play: ', will_play
      print 'get_state: ', get_state

def get_player_para(player):
    length = player.get_length()
    width = player.video_get_width()
    size = player.video_get_size()
    title = player.video_get_title_description()
    video_track = player.video_get_track_description()
    audio_track = player.audio_get_track_description()
    fps = player.get_fps()
    rate = player.get_rate()
    track_count = player.video_get_track_count()
    track = player.video_get_track()
    print 'length: ',length
    print 'width: ',width
    print 'size: ', size
    print 'title: ',title
    print 'video_track: ',video_track
    print 'audio_track: ',audio_track
    print 'fps: ',fps
    print 'rate: ', rate
    print 'track_count: ',track_count
    print 'track: ',track

def vlc_main():
    timestamp = time.time()
    while time.time()-timestamp < 400:
        player = get_player()
        time.sleep(1)  # must delay at least one second
        if player.is_playing():
            time.sleep(10)
            get_player_para(player)
            player.video_take_snapshot(num=0, psz_filepath=os.getcwd(), i_width=1080, i_height=360)
            #time.sleep(10)
            player.release()
            return
        player.release()
    player.release()
    print '###########################'
    print 'Fail: cannot find any video'
    print '###########################'

if __name__ == '__main__':
    vlc_main()