import os
import subprocess
from urllib.parse import urlparse

import yt_dlp

def download_video(url, output_folder):
    ydl_opts = {
        'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
        'outtmpl': os.path.join(output_folder, '%(title)s.%(ext)s'),
    }

    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
        ydl.download([url])

def convert_to_mp4(input_file, output_folder):
    output_file = os.path.join(output_folder, os.path.splitext(os.path.basename(input_file))[0] + '.mp4')
    subprocess.run(['ffmpeg', '-i', input_file, '-c', 'copy', output_file])

def main():
    video_urls = input("请输入视频链接（多个链接用逗号分隔）: ").split(',')
    output_folder = 'mp4'

    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    for url in video_urls:
        url = url.strip()
        if not url:
            continue

        try:
            print(f"正在下载视频：{url}")
            download_video(url, output_folder)
        except Exception as e:
            print(f"下载视频时发生错误：{e}")

    # 转换非 mp4 格式文件为 mp4
    for filename in os.listdir(output_folder):
        if filename.lower().endswith(('.mkv', '.flv', '.webm', '.avi', '.mov', '.wmv', '.3gp')):
            input_file = os.path.join(output_folder, filename)
            convert_to_mp4(input_file, output_folder)
            os.remove(input_file)

    print("所有视频下载和转换完成，输出目录：", os.path.abspath(output_folder))

if __name__ == "__main__":
    main()