云存视频下载方法

from urllib.parse import urlparse
import requests

def get_m3u8(url):
    r = requests.get(url)
    if (r.status_code != 200):
        return None
    return r.content.decode("utf-8")

def make_ts_list(url, m3u8):
    ts_list = []
    m3u8_url = urlparse(url)
    url_head = m3u8_url.scheme + '://' + m3u8_url.hostname
    m3u8_lines = m3u8.split("\\n")
    for each_line in m3u8_lines:
        if not each_line.startswith('#') and each_line != '':
            ts_list.append('%s%s' %(url_head, each_line))
    return ts_list

def download_ts(ts_list):
    for each_ts in ts_list:
        ts_url = urlparse(each_ts)
        pos = ts_url.path.rfind('/') + 1
        filename = ts_url.path[pos:]
        print("download " + filename)
        r = requests.get(each_ts)
        if (r.status_code != 200):
            return
        with open(filename, "wb") as fw:
            fw.write(r.content)
    print("download finish")

def main():
    aim_url = "https://zylcb.iotvideo.tencentcs.com/timeshift/live/timeshift.m3u8"
    m3u8 = get_m3u8(aim_url)
    ts_list = make_ts_list(aim_url, m3u8)
    download_ts(ts_list)

if (__name__ == "__main__"):
    main()