python调用mediainfo工具批量提取视频信息

写了2个脚本，分别是v1版本和v2版本

都是python调用mediainfo工具提取视频元数据信息

v1版本是使用pycharm中测试运行的，指定了视频路径

v2版本是最终交付给运营运行的，会把v2版本打成exe运行

先看v1版本

import os,subprocess,json,re,locale,sys

import xlwt,time,shutil

#获取当前文件所在绝对目录路径

# this_path=os.path.abspath('.')

# print('当前路径为----',this_path)

# dir_path=this_path

#视频文件所在目录

dir_path='I:\\3分钟便当'

# print(os.listdir(this_path))

print('---------------------------------')

print('--------------程序马上开始----------------')

# dir_path=this_path

#定义个列表存放每个文件绝对路径，便于后期操作

init_list=[]

# dir_path='F:\\玩具屋总视频'

#创建个方法，统计每个文件路径，并追加列表中。这里注释掉了递归,不获取子目录了，只获取dir_path下面的视频

def get_all_file(dir_path,init_list):

    for file in os.listdir(dir_path):

        # print(file)

        filepath=os.path.join(dir_path,file)

        # print(filepath)

        if os.path.isdir(filepath):

            print('遇到子目录---%s---此版本暂不提取子目录视频信息--'%(filepath))

            time.sleep(2)

            # get_all_file(filepath)

        else:

            if not file.endswith('exe'):

                init_list.append(filepath)

    return init_list

#执行上面方法，把每个文件绝对路径追加到列表中

file_list=get_all_file(dir_path,init_list)

print("文件读取完毕-----3秒后开始获取视频详细信息------------")

time.sleep(3)

#定义个方法，获取单个media文件的元数据，返回为字典数据

#此程序核心是调用了mediainfo工具来提取视频信息的

def get_media_info(file):

    pname='D:\mediainfo_i386\MediaInfo.exe "%s" --Output=JSON'%(file)

    result=subprocess.Popen(pname,shell=False,stdout=subprocess.PIPE).stdout

    list_std=result.readlines()

    str_tmp=''

    for item in list_std:

        str_tmp+=bytes.decode(item.strip())

    json_data=json.loads(str_tmp)

    return json_data

#定义个方法传递字典数据,返回自己想要的字段数据,返回值列表

def get_dict_data(json_data):

    #获取文件大小

    filesize=json_data['media']['track'][0]['FileSize']

    #获取码率

    malv=json_data['media']['track'][0]['OverallBitRate'][0:4]

    #获取播放时长

    duration=json_data['media']['track'][0]['Duration'].split('.')[0]

    #获取文件类型

    file_format=json_data['media']['track'][0]['Format']

    #获取帧宽

    samp_width=json_data['media']['track'][1]['Sampled_Width']

    #获取帧高

    samp_height=json_data['media']['track'][1]['Sampled_Height']

    return [filesize,malv,duration,file_format,samp_width,samp_height]

#定义个方法，获取文件名，它是key，它的value就是目标列表，返回值是个字典,参数是文件列表

dict_all={}

#定义个日志存提取失败视频文件名

f_fail=open('提取失败日志.log','a',encoding='utf-8') # 追加模式

def get_all_dict(file_list,f_fail):

    for file in file_list:

        filename=os.path.split(file)[1]

        print(filename)

        time.sleep(0.1)

        try:

            info_list=get_dict_data(get_media_info(file))

            dict_all[filename]=info_list

        except Exception as e:

            print(filename,'------提取此文件信息失败---------')

            f_fail.write(filename+'\r\n')

    f_fail.close()

get_all_dict(file_list,f_fail)

# for item in dict_all:

#     print(item,dict_all[item])

#创建一个excel表存放文件路径信息，第一列是目录，第二列是文件名

wb = xlwt.Workbook()

sh = wb.add_sheet('元数据')

#写第一行

row_count=0

sh.write(row_count,0,"文件名")

sh.write(row_count,1,"文件大小")

sh.write(row_count,2,"码率")

sh.write(row_count,3,"总时长")

sh.write(row_count,4,"视频格式")

sh.write(row_count,5,"帧宽")

sh.write(row_count,6,"帧高")

#批量写入视频信息

row_count=1

for item in dict_all:

    sh.write(row_count,0,item)

    sh.write(row_count,1,dict_all[item][0])

    sh.write(row_count,2,dict_all[item][1])

    sh.write(row_count,3,dict_all[item][2])

    sh.write(row_count,4,dict_all[item][3])

    sh.write(row_count,5,dict_all[item][4])

    sh.write(row_count,6,dict_all[item][5])

    row_count+=1

#

wb.save("元数据统计.xls")

#

在看v2版本

v2版本是对当前目录下的视频进行提取

import os,subprocess,json,re,locale,sys

import xlwt,time,shutil

#获取当前文件所在绝对目录路径

this_path=os.path.abspath('.')

print('当前路径为----',this_path)

dir_path=this_path

# print(os.listdir(this_path))

print('---------------------------------')

print('--------------程序马上开始----------------')

# dir_path=this_path

#定义个列表存放每个文件路径，便于后期操作

init_list=[]

# dir_path='I:\\3分钟便当'

# dir_path='F:\\玩具屋总视频'

#创建个方法，统计每个文件路径，并追加列表中。用到了递归,这里不获取子目录了

def get_all_file(dir_path,init_list):

    for file in os.listdir(dir_path):

        # print(file)

        filepath=os.path.join(dir_path,file)

        # print(filepath)

        if os.path.isdir(filepath):

            print('遇到子目录---%s---此版本暂不提取子目录视频信息--'%(filepath))

            time.sleep(2)

            # get_all_file(filepath)

        else:

            if not file.endswith('exe'):

                init_list.append(filepath)

    return init_list

#执行上面方法，把每个文件绝对路径追加到列表中

file_list=get_all_file(dir_path,init_list)

print("文件读取完毕-----3秒后开始获取视频详细信息------------")

time.sleep(3)

#定义个方法，获取单个media文件的元数据，返回为字典数据

def get_media_info(file):

    pname='D:\mediainfo_i386\MediaInfo.exe "%s" --Output=JSON'%(file)

    result=subprocess.Popen(pname,shell=False,stdout=subprocess.PIPE).stdout

    list_std=result.readlines()

    str_tmp=''

    for item in list_std:

        str_tmp+=bytes.decode(item.strip())

    json_data=json.loads(str_tmp)

    return json_data

#定义个方法传递字典数据,返回自己想要的字段数据,返回值列表

def get_dict_data(json_data):

    #获取文件大小

    filesize=json_data['media']['track'][0]['FileSize']

    #获取码率

    malv=json_data['media']['track'][0]['OverallBitRate'][0:4]

    #获取播放时长

    duration=json_data['media']['track'][0]['Duration'].split('.')[0]

    #获取文件类型

    file_format=json_data['media']['track'][0]['Format']

    #获取帧宽

    samp_width=json_data['media']['track'][1]['Sampled_Width']

    #获取帧高

    samp_height=json_data['media']['track'][1]['Sampled_Height']

    return [filesize,malv,duration,file_format,samp_width,samp_height]

#定义个方法，获取文件名，它是key，它的value就是目标列表，返回值是个字典,参数是文件列表

dict_all={}

f_fail=open('提取失败日志.log','a',encoding='utf-8') # 追加模式

def get_all_dict(file_list,f_fail):

    for file in file_list:

        filename=os.path.split(file)[1]

        print(filename)

        time.sleep(0.1)

        try:

            info_list=get_dict_data(get_media_info(file))

            dict_all[filename]=info_list

        except Exception as e:

            print(filename,'------提取此文件信息失败---------')

            f_fail.write(filename+'\r\n')

    f_fail.close()

get_all_dict(file_list,f_fail)

# for item in dict_all:

#     print(item,dict_all[item])

#创建一个excel表存放文件路径信息，第一列是目录，第二列是文件名

wb = xlwt.Workbook()

sh = wb.add_sheet('元数据')

#写第一行

row_count=0

sh.write(row_count,0,"文件名")

sh.write(row_count,1,"文件大小")

sh.write(row_count,2,"码率")

sh.write(row_count,3,"总时长")

sh.write(row_count,4,"视频格式")

sh.write(row_count,5,"帧宽")

sh.write(row_count,6,"帧高")

row_count=1

for item in dict_all:

    sh.write(row_count,0,item)

    sh.write(row_count,1,dict_all[item][0])

    sh.write(row_count,2,dict_all[item][1])

    sh.write(row_count,3,dict_all[item][2])

    sh.write(row_count,4,dict_all[item][3])

    sh.write(row_count,5,dict_all[item][4])

    sh.write(row_count,6,dict_all[item][5])

    row_count+=1

#

wb.save("元数据统计.xls")