如何检查文件是否为视频

我正试图在Python 2.7中创建一个程序，该程序具有以下要求：

枚举Windows中的所有视频文件
使用以下命令打开每个视频文件：os.open(filepath, os.O_RDWR)
处理文件

我目前的算法是：

拥有所有视频文件扩展名的列表。我从这里得到的
循环浏览所有文件
如果文件具有视频扩展名，则对其进行处理

这种方法有以下缺点：

某些文件扩展名（如".dat"或".am"）用于非媒体文件
某些扩展名为".dat"的文件被其他程序（如Antivirus）使用

我应该做什么来制作一个满足上述要求的程序？

这是我当前的代码：

def is_video_file(filename):
    video_file_extensions = (
'.264', '.3g2', '.3gp', '.3gp2', '.3gpp', '.3gpp2', '.3mm', '.3p2', '.60d', '.787', '.89', '.aaf', '.aec', '.aep', '.aepx',
'.aet', '.aetx', '.ajp', '.ale', '.am', '.amc', '.amv', '.amx', '.anim', '.aqt', '.arcut', '.arf', '.asf', '.asx', '.avb',
'.avc', '.avd', '.avi', '.avp', '.avs', '.avs', '.avv', '.axm', '.bdm', '.bdmv', '.bdt2', '.bdt3', '.bik', '.bin', '.bix',
'.bmk', '.bnp', '.box', '.bs4', '.bsf', '.bvr', '.byu', '.camproj', '.camrec', '.camv', '.ced', '.cel', '.cine', '.cip',
'.clpi', '.cmmp', '.cmmtpl', '.cmproj', '.cmrec', '.cpi', '.cst', '.cvc', '.cx3', '.d2v', '.d3v', '.dat', '.dav', '.dce',
'.dck', '.dcr', '.dcr', '.ddat', '.dif', '.dir', '.divx', '.dlx', '.dmb', '.dmsd', '.dmsd3d', '.dmsm', '.dmsm3d', '.dmss',
'.dmx', '.dnc', '.dpa', '.dpg', '.dream', '.dsy', '.dv', '.dv-avi', '.dv4', '.dvdmedia', '.dvr', '.dvr-ms', '.dvx', '.dxr',
'.dzm', '.dzp', '.dzt', '.edl', '.evo', '.eye', '.ezt', '.f4p', '.f4v', '.fbr', '.fbr', '.fbz', '.fcp', '.fcproject',
'.ffd', '.flc', '.flh', '.fli', '.flv', '.flx', '.gfp', '.gl', '.gom', '.grasp', '.gts', '.gvi', '.gvp', '.h264', '.hdmov',
'.hkm', '.ifo', '.imovieproj', '.imovieproject', '.ircp', '.irf', '.ism', '.ismc', '.ismv', '.iva', '.ivf', '.ivr', '.ivs',
'.izz', '.izzy', '.jss', '.jts', '.jtv', '.k3g', '.kmv', '.ktn', '.lrec', '.lsf', '.lsx', '.m15', '.m1pg', '.m1v', '.m21',
'.m21', '.m2a', '.m2p', '.m2t', '.m2ts', '.m2v', '.m4e', '.m4u', '.m4v', '.m75', '.mani', '.meta', '.mgv', '.mj2', '.mjp',
'.mjpg', '.mk3d', '.mkv', '.mmv', '.mnv', '.mob', '.mod', '.modd', '.moff', '.moi', '.moov', '.mov', '.movie', '.mp21',
'.mp21', '.mp2v', '.mp4', '.mp4v', '.mpe', '.mpeg', '.mpeg1', '.mpeg4', '.mpf', '.mpg', '.mpg2', '.mpgindex', '.mpl',
'.mpl', '.mpls', '.mpsub', '.mpv', '.mpv2', '.mqv', '.msdvd', '.mse', '.msh', '.mswmm', '.mts', '.mtv', '.mvb', '.mvc',
'.mvd', '.mve', '.mvex', '.mvp', '.mvp', '.mvy', '.mxf', '.mxv', '.mys', '.ncor', '.nsv', '.nut', '.nuv', '.nvc', '.ogm',
'.ogv', '.ogx', '.osp', '.otrkey', '.pac', '.par', '.pds', '.pgi', '.photoshow', '.piv', '.pjs', '.playlist', '.plproj',
'.pmf', '.pmv', '.pns', '.ppj', '.prel', '.pro', '.prproj', '.prtl', '.psb', '.psh', '.pssd', '.pva', '.pvr', '.pxv',
'.qt', '.qtch', '.qtindex', '.qtl', '.qtm', '.qtz', '.r3d', '.rcd', '.rcproject', '.rdb', '.rec', '.rm', '.rmd', '.rmd',
'.rmp', '.rms', '.rmv', '.rmvb', '.roq', '.rp', '.rsx', '.rts', '.rts', '.rum', '.rv', '.rvid', '.rvl', '.sbk', '.sbt',
'.scc', '.scm', '.scm', '.scn', '.screenflow', '.sec', '.sedprj', '.seq', '.sfd', '.sfvidcap', '.siv', '.smi', '.smi',
'.smil', '.smk', '.sml', '.smv', '.spl', '.sqz', '.srt', '.ssf', '.ssm', '.stl', '.str', '.stx', '.svi', '.swf', '.swi',
'.swt', '.tda3mt', '.tdx', '.thp', '.tivo', '.tix', '.tod', '.tp', '.tp0', '.tpd', '.tpr', '.trp', '.ts', '.tsp', '.ttxt',
'.tvs', '.usf', '.usm', '.vc1', '.vcpf', '.vcr', '.vcv', '.vdo', '.vdr', '.vdx', '.veg','.vem', '.vep', '.vf', '.vft',
'.vfw', '.vfz', '.vgz', '.vid', '.video', '.viewlet', '.viv', '.vivo', '.vlab', '.vob', '.vp3', '.vp6', '.vp7', '.vpj',
'.vro', '.vs4', '.vse', '.vsp', '.w32', '.wcp', '.webm', '.wlmp', '.wm', '.wmd', '.wmmp', '.wmv', '.wmx', '.wot', '.wp3',
'.wpl', '.wtv', '.wve', '.wvx', '.xej', '.xel', '.xesc', '.xfl', '.xlmv', '.xmv', '.xvid', '.y4m', '.yog', '.yuv', '.zeg',
'.zm1', '.zm2', '.zm3', '.zmv'  )
if filename.endswith((video_file_extensions)):
    return True
def get_drives():
    import string
    from ctypes import windll
    import ctypes
    from ctypes.wintypes import MAX_PATH
    drives = []
    bitmask = windll.kernel32.GetLogicalDrives()
    for letter in string.uppercase:
        if bitmask & 1:
            letter = letter + ':\'
            drives.append(letter)
        bitmask >>= 1
    return drives
def process(filepath):
    try:
        video_file = os.open(filepath, os.O_RDWR)
        # Do processing

def process_all_videos():
    for dirve in drives:
        for root, dirs, files in os.walk(dirve):
            for filename in files:
                if is_video_file(filename):
                    try:
                        filepath = os.path.join(root, filename)
                        process(filepath)

def main():
    process_all_videos()
if __name__ == '__main__':
    main()

尝试使用python-magic，它通过读取文件的内容（通常是前1024个字节）来更好地检测文件的类型。

此时，有一种更简单的方法可以检查任何给定文件是否包含有效的视频流：使用Python的MediaInfo包装器加载文件并检查视频流的存在。

fileInfo = MediaInfo.parse('some/file/name.ext')
for track in fileInfo.tracks:
    if track.track_type == "Video":
        # success!

我不知道这样做的效率，但它至少应该和使用ffmpeg一样快。至少，它有一个很好的API，还有很多其他很好的信息，你可以很容易地用它提取

这是问题的有效解决方案

import mimetypes
if mimetypes.guess_type('file_path')[0].startswith('video'):
   print('It is a video')```

我遇到了同样的问题，这是我的解决方案：

import magic
mime = magic.Magic(mime=True)
filename = mime.from_file(path_to_video)
if filename.find('video') != -1:
   print('it is video')

我可能会使用ffmpeg/libavcodec打印出文件信息，并解析输出以检查文件是否包含视频流。

然后，您可以避免有一个庞大的文件扩展名列表，这也将解决DAT文件的问题。

这里有一个使用ffmpeg:的工作解决方案

import ffmpeg
def fileHasVideoStream(file_path):
    video_stream = ffmpeg.probe(file_path, select_streams='v')['streams']
    if video_stream:
        return True
    else:
        return False

如果没有找到视频流，则video_stream将是一个空列表，并且我们返回False。

我在代码中使用ffmpeg获取元数据，所以我使用相同的包来检查文件是否是视频：

import ffmpeg
def check_if_video(path):
    metadata = get_metadata(path)
    return metadata['codec_type'] == 'video'
def get_metadata(path):
    return ffmpeg.probe(path, select_streams = "v")['streams'][0]

相关内容

最新更新

热门标签：