Skip to content

Downloader

VideoDownloader

Bases: BaseModel

Methods:

Name Description
check_if_tiktok
download

output_folder 

output_folder: str = Field(default='./data/downloads', description='Download folder')

quality_formats 

quality_formats: dict[str, str]

check_if_tiktok 

check_if_tiktok(url: str) -> bool
Source code in src/utils/downloader.py 
27
28
def check_if_tiktok(self, url: str) -> bool:
    return bool(tiktok_pattern.match(url))

download 

download(url: str, quality: str = 'best') -> tuple[str, Path]
Source code in src/utils/downloader.py 
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
def download(self, url: str, quality: str = "best") -> tuple[str, Path]:
    filename_stem = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
    match = shorts_pattern.search(url)
    if match:
        found_name = match.group(1)
        if isinstance(found_name, str):
            filename_stem = found_name

    output_path = Path(self.output_folder)
    if not output_path.exists():
        output_path.mkdir(parents=True, exist_ok=True)

    # 獲取所選畫質的格式設定
    format_option = self.quality_formats.get(quality, "best")
    is_audio_only = quality == "audio"
    outtmpl = f"{output_path.as_posix()}/{filename_stem}.%(ext)s"

    ydl_opts = {
        "format": format_option,
        "outtmpl": outtmpl,
        "quiet": False,
        "continuedl": True,
        "noplaylist": True,
        "restrictfilenames": True,
        "http_headers": {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36",
            "Accept-Language": "en-US,en;q=0.9",
            "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8",
            "Connection": "keep-alive",
            "Upgrade-Insecure-Requests": "1",
            "Sec-Fetch-Dest": "document",
            "Sec-Fetch-Mode": "navigate",
            "Sec-Fetch-Site": "none",
            "Sec-Fetch-User": "?1",
        },
        # "writeinfojson": True,
    }

    # 如果是音訊模式，轉換成 mp3
    if is_audio_only:
        ydl_opts.update({
            "postprocessors": [
                {
                    "key": "FFmpegExtractAudio",
                    "preferredcodec": "mp3",
                    "preferredquality": "192",
                }
            ]
        })

    with YoutubeDL(ydl_opts) as ydl:
        info = ydl.extract_info(url, download=True)
        title = info.get("title", "")
        filename = Path(ydl.prepare_filename(info))
        # 修正音訊模式下的副檔名
        if is_audio_only and filename.suffix != ".mp3":
            filename = filename.with_suffix(".mp3")
    return title, filename