import os
from pymediainfo import MediaInfo


def is_valid_mp4(file_path):
    """
    使用pymediainfo檢查MP4文件格式是否正確

    參數(shù):
        file_path: MP4文件路徑

    返回:
        tuple: (是否有效, 錯誤信息/詳細信息)
    """
    # 基礎檢查
    if not os.path.exists(file_path):
        return False, "文件不存在"

    if not os.path.isfile(file_path):
        return False, "路徑不是一個文件"

    file_size = os.path.getsize(file_path)
    if file_size == 0:
        return False, "文件大小為0字節(jié)，可能為空文件"

    try:
        # 分析媒體信息
        media_info = MediaInfo.parse(file_path)

        # 獲取文件格式信息
        general_track = next(t for t in media_info.tracks if t.track_type == "General")
        format_name = general_track.format

        if format_name not in ["MPEG-4", "MP4"]:
            return False, f"不是MP4格式，實際格式為: {format_name}"

        # 檢查是否包含視頻軌道
        video_tracks = [t for t in media_info.tracks if t.track_type == "Video"]
        if not video_tracks:
            return False, "文件不包含視頻軌道"

        video_track = video_tracks[0]
        video_codec = video_track.codec_id or video_track.codec

        # 檢查視頻編碼兼容性
        supported_video_codecs = ["avc1", "hev1", "hvc1", "mp4v", "vp09"]  # MP4常見視頻編碼
        if not any(codec in video_codec.lower() for codec in supported_video_codecs):
            return False, f"不支持的視頻編碼: {video_codec} (支持: {', '.join(supported_video_codecs)})"

        # 檢查音頻軌道（如果存在）
        audio_tracks = [t for t in media_info.tracks if t.track_type == "Audio"]
        if audio_tracks:
            audio_codec = audio_tracks[0].codec_id or audio_tracks[0].codec
            supported_audio_codecs = ["mp4a", "mp3", "ac-3", "opus"]  # MP4常見音頻編碼
            if not any(codec in audio_codec.lower() for codec in supported_audio_codecs):
                return False, f"不支持的音頻編碼: {audio_codec} (支持: {', '.join(supported_audio_codecs)})"

        # 收集詳細信息
        details = {
            "格式": format_name,
            "視頻編碼": video_codec,
            "分辨率": f"{video_track.width}x{video_track.height}",
            "時長": f"{video_track.duration / 1000:.2f}秒",
            "文件大小": f"{file_size / (1024 * 1024):.2f}MB"
        }

        return True, f"MP4格式正確。詳細信息: {details}"

    except StopIteration:
        return False, "無法解析文件結(jié)構(gòu)，可能不是媒體文件"
    except Exception as e:
        return False, f"格式錯誤或文件損壞: {str(e)}"


if __name__ == "__main__":
    # 替換為你的MP4文件路徑
    mp4_file = "D:\\VID_1931_040_ouput1.mp4"

    valid, message = is_valid_mp4(mp4_file)
    print(f"檢查結(jié)果: {'有效' if valid else '無效'}")
    print(f"詳情: {message}")

mp4_file = "你的MP4文件路徑"  # 例如："D:\\test.mp4" 或 "/home/user/video.mp4"