使用Python验证mp3文件的最佳方法

Question

我需要检测一个文件是否是有效的mp3文件。目前为止，我已经找到了两种解决方案，包括:

1. 这个解决方案 来自Peter Carroll。

2. 使用 try-catch 表达式：

try:
    _ = librosa.get_duration(filename=mp3_file_path)
    return True
except:
    return False

以上两种解决方案都能行得通，但可能存在一些缺点，第一种方案看起来过于复杂，而第二种方案则太慢（需要几秒钟的时间，具体取决于音频的长度）。所以我想知道是否有更好的方法来使用Python验证mp3文件？

谢谢。

- xtluo

---

我认为Peter的解决方案是可行的。你可以使用 struct 来改进一下，这可能会使它更符合Python风格。通常二进制文件很复杂，读取它们的代码也很复杂，但你不应该担心这些。你只需编写一次代码，测试所有已知情况，然后再也不用看它了。 - Jose A. García

@JoseA。我认为这两个解决方案的验证过程可能是相同的。它们都试图将文件解码为常规mp3文件，直到发生某些异常。因此，也许更快的音频处理库是一种方法，另一个解决方案就像你提到的那样，是改进Peter的代码。谢谢。 - xtluo

据我所知，Peter的代码只读取了mp3文件的前两帧。因此它不会查看整个文件，甚至不会解码音频。它只检查文件格式是否正确。而且它应该很快。 - Jose A. García

Answer 1

尝试使用eyed3。

In [8]: %timeit eyed3.load("some.mp3").info.time_secs                                                    
1.94 ms ± 43.8 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

In [10]: %timeit eyed3.load("120mb_video.mp4")                                                        
4.43 ms ± 26.6 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

基本上，如果eyed3.load返回None，则该文件不是有效的mp3文件。即使对于一个120mb的文件，它的平均运行时间也只有4.5毫秒。

- altunyurt

Answer 2

尝试这段代码。它会从文件夹中删除任何不是有效mp3的文件，因此只剩下有效的mp3。

import os



def isMp3Valid(file_path):
    is_valid = False
    f = open(file_path, 'r')
    block = f.read(1024)
    frame_start = block.find(chr(255))
    block_count = 0  # abort after 64k
    while len(block) > 0 and frame_start == -1 and block_count < 64:
        block = f.read(1024)
        frame_start = block.find(chr(255))
        block_count += 1

    if frame_start > -1:
        frame_hdr = block[frame_start:frame_start + 4]
        is_valid = frame_hdr[0] == chr(255)

        mpeg_version = ''
        layer_desc = ''
        uses_crc = False
        bitrate = 0
        sample_rate = 0
        padding = False
        frame_length = 0

        if is_valid:
            is_valid = ord(frame_hdr[1]) & 0xe0 == 0xe0  # validate the rest of the frame_sync bits exist

        if is_valid:
            if ord(frame_hdr[1]) & 0x18 == 0:
                mpeg_version = '2.5'
            elif ord(frame_hdr[1]) & 0x18 == 0x10:
                mpeg_version = '2'
            elif ord(frame_hdr[1]) & 0x18 == 0x18:
                mpeg_version = '1'
            else:
                is_valid = False

        if is_valid:
            if ord(frame_hdr[1]) & 6 == 2:
                layer_desc = 'Layer III'
            elif ord(frame_hdr[1]) & 6 == 4:
                layer_desc = 'Layer II'
            elif ord(frame_hdr[1]) & 6 == 6:
                layer_desc = 'Layer I'
            else:
                is_valid = False

        if is_valid:
            uses_crc = ord(frame_hdr[1]) & 1 == 0

            bitrate_chart = [
                [0, 0, 0, 0, 0],
                [32, 32, 32, 32, 8],
                [64, 48, 40, 48, 16],
                [96, 56, 48, 56, 24],
                [128, 64, 56, 64, 32],
                [160, 80, 64, 80, 40],
                [192, 96, 80, 96, 40],
                [224, 112, 96, 112, 56],
                [256, 128, 112, 128, 64],
                [288, 160, 128, 144, 80],
                [320, 192, 160, 160, 96],
                [352, 224, 192, 176, 112],
                [384, 256, 224, 192, 128],
                [416, 320, 256, 224, 144],
                [448, 384, 320, 256, 160]]
            bitrate_index = ord(frame_hdr[2]) >> 4
            if bitrate_index == 15:
                is_valid = False
            else:
                bitrate_col = 0
                if mpeg_version == '1':
                    if layer_desc == 'Layer I':
                        bitrate_col = 0
                    elif layer_desc == 'Layer II':
                        bitrate_col = 1
                    else:
                        bitrate_col = 2
                else:
                    if layer_desc == 'Layer I':
                        bitrate_col = 3
                    else:
                        bitrate_col = 4
                bitrate = bitrate_chart[bitrate_index][bitrate_col]
                is_valid = bitrate > 0

        if is_valid:
            sample_rate_chart = [
                [44100, 22050, 11025],
                [48000, 24000, 12000],
                [32000, 16000, 8000]]
            sample_rate_index = (ord(frame_hdr[2]) & 0xc) >> 2
            if sample_rate_index != 3:
                sample_rate_col = 0
                if mpeg_version == '1':
                    sample_rate_col = 0
                elif mpeg_version == '2':
                    sample_rate_col = 1
                else:
                    sample_rate_col = 2
                sample_rate = sample_rate_chart[sample_rate_index][sample_rate_col]
            else:
                is_valid = False

        if is_valid:
            padding = ord(frame_hdr[2]) & 1 == 1

            padding_length = 0
            if layer_desc == 'Layer I':
                if padding:
                    padding_length = 4
                frame_length = (12 * bitrate * 1000 / sample_rate + padding_length) * 4
            else:
                if padding:
                    padding_length = 1
                frame_length = 144 * bitrate * 1000 / sample_rate + padding_length
            is_valid = frame_length > 0

            # Verify the next frame
            if (frame_start + frame_length < len(block)):
                is_valid = block[frame_start + frame_length] == chr(255)
            else:
                offset = (frame_start + frame_length) - len(block)
                block = f.read(1024)
                if len(block) > offset:
                    is_valid = block[offset] == chr(255)
                else:
                    is_valid = False

    f.close()
    return is_valid



#You can remove this part if you don't want the deletion part. Hope it helps
for file, dir, files in os.walk("download"):
    for f in files:
        try:
            if not isMp3Valid("{}/{}".format(file, f)):
                os.remove("{}/{}".format(file, f))
        except Exception as e:
            print(e)

- Brian Brix