from urllib.request import Request, urlopen

urls = [
    "https://epg.112114.xyz/pp.xml.gz",  # 第一个文件，扩展名为 .xml.gz
    "https://epg.pw/xmltv/epg_HK.xml"    # 第二个文件，扩展名为 .xml
]

# 遍历URL列表，下载文件
for i, url in enumerate(urls):
    # 根据URL确定文件名和扩展名
    if url.endswith('.xml.gz'):
        downloaded_file_name = f"pp.xml.gz"  # 第一个文件，保持 .xml.gz 扩展名
    else:
        downloaded_file_name = f"HK.xml"     # 第二个文件，保持 .xml 扩展名
  
    try:
        req = Request(url, headers={'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'})
        with urlopen(req) as response:
            data = response.read()
            with open(downloaded_file_name, 'wb') as file:
                file.write(data)
            print(f"成功下载文件: {downloaded_file_name} 来自 {url}")
    except Exception as e:
        print(f"下载文件时出错: {e} 对于URL: {url}")