使用ffprobe分析视频,python代码调用ffprobe获取视频信息

背景

最近在做一个自动下载视频的小工具,其中就需要获取视频基础信息,如果分辨率、大小、播放时长等,研究发现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调用外部程序还是比较简单的