import requests
from tqdm import tqdm
import re
import checkOutTime

# https://s1.zoubuting.com/20220622/qwZCSgqD/hls/index.m3u8
# 名侦探柯南：纯黑的恶梦
url_m3u8 = input('输入m3u8地址：')
video_name = input("请输入文件名")
video_name = video_name.replace(' ', '_')
headers = {
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; '
                  'x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/105.0.0.0 '
                  'Safari/537.36 '
}
response = requests.get(url=url_m3u8, headers=headers)
data_m3u8 = response.text
data_m3u8 = re.sub('#EXT.*', ' ', data_m3u8).split()
for ts in tqdm(data_m3u8):
    ts_content = checkOutTime.getData1(ts).content
    with open('E:\\anyCrawlers\\crawBySelenium\\video\\' + video_name+'.mp4', mode='ab') as video:
        video.write(ts_content)
print('下载完成！')