通过正则表达式验证文件类型

import re

def validate_file_type(file_path):
    # 定义正则表达式
    image_pattern = r'\.(jpe?g|png|gif|bmp|webp)$'
    doc_pattern = r'\.(docx?|pdf|txt)$'
    audio_pattern = r'\.(mp3|wav|flac|aac)$'
    video_pattern = r'\.(mp4|avi|flv|mkv|mov|wmv)$'

    # 获取文件扩展名
    file_ext = file_path.split('.')[-1]

    # 验证文件类型
    if re.match(image_pattern, file_ext):
        return 'Image'
    elif re.match(doc_pattern, file_ext):
        return 'Document'
    elif re.match(audio_pattern, file_ext):
        return 'Audio'
    elif re.match(video_pattern, file_ext):
        return 'Video'
    else:
        return 'Unknown'