import os import requests

headers = { 'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/69.0.3497.100 Safari/537.36', 'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,/;q=0.8', 'Accept-Language': 'en-US,en;q=0.9,ja-JP;q=0.8,ja;q=0.7', 'Cache-Control': 'no-cache', 'Connection': 'keep-alive', 'Pragma': 'no-cache', 'Upgrade-Insecure-Requests': '1' }

输入包含.m3u8文件的目录路径

dir_path = input('请输入.m3u8文件所在目录的路径：')

遍历目录下所有的.m3u8文件

for file_name in os.listdir(dir_path): if file_name.endswith('.m3u8'): print(f'处理文件: {file_name}') file_path = os.path.join(dir_path, file_name) with open(file_path, 'r') as f: lines = f.readlines()

    # 处理以 '#YUMING|https' 开头的第一行
    if lines[0].startswith('#YUMING|https'):
        url = lines[0][lines[0].index('https'):lines[0].index('/-/')+3]
        response = requests.post(url + 'playlist.m3u8', headers=headers)
        if response.status_code == 200:
            print('成功：' + str(response.status_code))
            new_url = response.url
            new_url = new_url[:new_url.index('playlist.m3u8')]

            # 将所有以 'media_' 开头的行替换为新的 URL
            new_lines = []
            for line in lines:
                if line.startswith('media_'):
                    new_line = new_url + line.strip()
                    print(f'Replacing {line.strip()} with {new_line}')
                    new_lines.append(new_line)
                elif not line.startswith('#YUMING|https'):
                    new_lines.append(line)

            # 将新的行写回原文件（第一行以 '#YUMING|https' 开头的行除外）
            with open(file_path, 'w') as f:
                for line in new_lines[1:]:
                    f.write(line)
        else:
            print('失效：' + str(response.status_code))

    else:
        # 处理其他情况下的 URL
        if 'http' in lines[1]:
            url = lines[1][lines[1].index('http'):lines[1].index('/-/')+3]
        else:
            url = lines[8][lines[8].index('http'):lines[8].index('/-/')+3]

        response = requests.post(url + 'playlist.m3u8', headers=headers)
        if response.status_code == 200:
            print('成功：' + str(response.status_code))
            new_url = response.url
            new_url = new_url[:new_url.index('playlist.m3u8')]

            # 将所有 URL 在 "http" 和 "/-/" 之间的部分替换为新的 URL
            new_lines = []
            for line in lines:
                if 'media_' in line:
                    new_line = line[:line.index('media_')] + new_url + line[line.index('media_'):].strip()
                    print(f'Replacing {line.strip()} with {new_line}')
                    new_lines.append(new_line)
                else:
                    new_lines.append(line)

            # 将新的行写回原文件
            with open(file_path, 'w') as f:
                for line in new_lines:
                    if not line.startswith('#YUMING|https'):
                        f.write(line)
        else:
            print('失效：' + str(response.status_code)