背景

最近在做一个自动下载视频的小工具,其中就需要获取视频基础信息,如果分辨率、大小、播放时长等,研究发现ffprobe工具已经提供了非常全面的视频信息,用ffprobe分析视频很简单,且看我是怎么在python码中调用ffprobe来提取视频信息的

ffprobe 命令

我们先来看下ffprobe命令分析网络视频

ffprobe -print_format json -show_format -show_streams -v quiet https://1252524126.vod2.myqcloud.com/9764a7a5vodtransgzp1252524126/4e89d74d5285890792243257460/drm/voddrm.token.YTY4YThkNGE4ZDQ4NmVkNXRGWWRTTk1kMmwrUkF3VHZ6YUduS3diR1ZTRzlSZVJzSnVsajFBQ0VWREZtQ3VMbA.v.f230.m3u8?time=1592902124084

输出结果:

{
"streams": [
{
"index": 0,
"codec_name": "h264",
"codec_long_name": "H.264 / AVC / MPEG-4 AVC / MPEG-4 part 10",
"profile": "High",
"codec_type": "video",
"codec_time_base": "1/48",
"codec_tag_string": "[27][0][0][0]",
"codec_tag": "0x001b",
"width": 1280,
"height": 720,
"coded_width": 1280,
"coded_height": 720,
"has_b_frames": 2,
"sample_aspect_ratio": "1:1",
"display_aspect_ratio": "16:9",
"pix_fmt": "yuv420p",
"level": 31,
"chroma_location": "left",
"refs": 1,
"is_avc": "false",
"nal_length_size": "0",
"r_frame_rate": "24/1",
"avg_frame_rate": "24/1",
"time_base": "1/90000",
"start_pts": 7500,
"start_time": "0.083333",
"bits_per_raw_sample": "8",
"disposition": {
"default": 0,
"dub": 0,
"original": 0,
"comment": 0,
"lyrics": 0,
"karaoke": 0,
"forced": 0,
"hearing_impaired": 0,
"visual_impaired": 0,
"clean_effects": 0,
"attached_pic": 0,
"timed_thumbnails": 0
},
"tags": {
"variant_bitrate": "0"
}
},
{
"index": 1,
"codec_name": "aac",
"codec_long_name": "AAC (Advanced Audio Coding)",
"profile": "LC",
"codec_type": "audio",
"codec_time_base": "1/44100",
"codec_tag_string": "[15][0][0][0]",
"codec_tag": "0x000f",
"sample_fmt": "fltp",
"sample_rate": "44100",
"channels": 2,
"channel_layout": "stereo",
"bits_per_sample": 0,
"r_frame_rate": "0/0",
"avg_frame_rate": "0/0",
"time_base": "1/90000",
"start_pts": 3320,
"start_time": "0.036889",
"disposition": {
"default": 0,
"dub": 0,
"original": 0,
"comment": 0,
"lyrics": 0,
"karaoke": 0,
"forced": 0,
"hearing_impaired": 0,
"visual_impaired": 0,
"clean_effects": 0,
"attached_pic": 0,
"timed_thumbnails": 0
},
"tags": {
"variant_bitrate": "0"
}
}
],
"format": {
"filename": "https://1252524126.vod2.myqcloud.com/9764a7a5vodtransgzp1252524126/4e89d74d5285890792243257460/drm/voddrm.token.YTY4YThkNGE4ZDQ4NmVkNXRGWWRTTk1kMmwrUkF3VHZ6YUduS3diR1ZTRzlSZVJzSnVsNXRGWWRTTk1kMmwrUkF3VHZ6YUduSdiR1ZTRzlSZVJzSnVsajFBQ0VWREZtQ3VMbA.v.f230.m3u8?time=1592902124084",
"nb_streams": 2,
"nb_programs": 1,
"format_name": "hls",
"format_long_name": "Apple HTTP Live Streaming",
"start_time": "0.036889",
"duration": "225.005000",
"size": "14608",
"bit_rate": "519",
"probe_score": 100
}
}

这里面已经可以提取到视频的基本信息了,我们只需要通过python调用ffprobe命令就可以了

python调用ffprobe获取视频信息

import subprocess,json

class FFprobe():
def __init__(self):
self.filepath = ''
self._video_info = {}


def parse(self,filepath):
self.filepath = filepath
try:
res = subprocess.check_output(['ffprobe','-i',self.filepath,'-print_format','json','-show_format','-show_streams','-v','quiet'])
res = res.decode('utf8')
self._video_info = json.loads(res)
# print('_video_info ',self._video_info)
except Exception as e:
print(e)
raise Exception('获取视频信息失败')

def video_width_height(self):
streams = self._video_info['streams'][0]
return (streams['width'],streams['height'])

def video_filesize(self, format = 'gb'):
v_format = self._video_info['format']
size = int(v_format['size'])
kb=1024
mb=kb*1024
gb=mb*1024
tb=gb*1024
if size>=tb:
return "%.1f TB" % float(size / tb)
if size>=gb:
return "%.1f GB" % float(size / gb)
if size>=mb:
return "%.1f MB" % float(size / mb)
if size>=kb:
return "%.1f KB" % float(size / kb)

def video_full_frame(self):
stream = self._video_info['streams'][0]
return stream['nb_frames']

def video_time_length(self):
v_format = self._video_info['format']
return str(int(float(v_format['duration'])/3600)).__add__('小时').__add__(str(int(float(v_format['duration'])%3600/60))).__add__('分钟')

def video_info(self):
item = {
'path': self.filepath,
'height_width': self.video_width_height(),
'filesize': self.video_filesize(),
'time_length': self.video_time_length()
}
print('item = ', item)
return item


if __name__ == "__main__":
ffprobe = FFprobe()
ffprobe.parse('https://1252524126.vod2.myqcloud.com/9764a7a5vodtransgzp1252524126/4e89d74d5285890792243257460/drm/voddrm.token.MWVmNGRkjI2MTEzNDEzMVFxUFU4SEs3d3crUWREbTVTSk55YUhxRW1VTTBjVzMrTFN6T0VnbjhZZmZscTRVcQ.v.f230.m3u8?time=1592901564214')
print(ffprobe.video_info())

总结

当我们使用python不太好实现的时候,可以是外部程序,毕竟python调用外部程序还是比较简单的