wip

.gitignore

@@ -15,5 +15,4 @@ tmp/
 /*.pyz
 /pyrightconfig.json
 /book
-*.mp4
-*.mkv
+/*.mkv

CHANGELOG.md

@@ -3,18 +3,6 @@ twitch-dl changelog
 
 <!-- Do not edit. This file is automatically generated from changelog.yaml.-->
 
-### [2.5.0 (2024-08-30)](https://github.com/ihabunek/twitch-dl/releases/tag/2.5.0)
-
-* Add support for HD video qualities (#163)
-
-### [2.4.0 (2024-08-30)](https://github.com/ihabunek/twitch-dl/releases/tag/2.4.0)
-
-* Add `clips --target-dir` option. Use in conjunction with `--download` to
-  specify target directory.
-* Fix a crash when downloading clips (#160)
-* Handle video URLs which contain the channel name (#162)
-* Don't stop downloading clips if one download fails
-
 ### [2.3.1 (2024-05-19)](https://github.com/ihabunek/twitch-dl/releases/tag/2.3.1)
 
 * Fix fetching access token (#155, thanks @KryptonicDragon)

changelog.yaml

@@ -1,16 +1,3 @@
-2.5.0:
-  date: 2024-08-30
-  changes:
-    - "Add support for HD video qualities (#163)"
-
-2.4.0:
-  date: 2024-08-30
-  changes:
-    - "Add `clips --target-dir` option. Use in conjunction with `--download` to specify target directory."
-    - "Fix a crash when downloading clips (#160)"
-    - "Handle video URLs which contain the channel name (#162)"
-    - "Don't stop downloading clips if one download fails"
-
 2.3.1:
   date: 2024-05-19
   changes:

docs/changelog.md

@@ -3,18 +3,6 @@ twitch-dl changelog
 
 <!-- Do not edit. This file is automatically generated from changelog.yaml.-->
 
-### [2.5.0 (2024-08-30)](https://github.com/ihabunek/twitch-dl/releases/tag/2.5.0)
-
-* Add support for HD video qualities (#163)
-
-### [2.4.0 (2024-08-30)](https://github.com/ihabunek/twitch-dl/releases/tag/2.4.0)
-
-* Add `clips --target-dir` option. Use in conjunction with `--download` to
-  specify target directory.
-* Fix a crash when downloading clips (#160)
-* Handle video URLs which contain the channel name (#162)
-* Don't stop downloading clips if one download fails
-
 ### [2.3.1 (2024-05-19)](https://github.com/ihabunek/twitch-dl/releases/tag/2.3.1)
 
 * Fix fetching access token (#155, thanks @KryptonicDragon)

docs/commands/clips.md

@@ -43,11 +43,6 @@ twitch-dl clips [OPTIONS] CHANNEL_NAME
     <td>Period from which to return clips Possible values: <code>last_day</code>, <code>last_week</code>, <code>last_month</code>, <code>all_time</code>. [default: <code>all_time</code>]</td>
 </tr>
 
-<tr>
-    <td class="code">-t, --target-dir</td>
-    <td>Target directory when downloading clips [default: <code>.</code>]</td>
-</tr>
-
 <tr>
     <td class="code">--json</td>
     <td>Print data as JSON rather than human readable text</td>

pyproject.toml

@@ -22,7 +22,7 @@ classifiers = [
 dependencies = [
     "click>=8.0.0,<9.0.0",
     "httpx>=0.17.0,<1.0.0",
-    "m3u8>=3.0.0,<7.0.0",
+    "m3u8>=3.0.0,<5.0.0",
 ]
 
 [tool.setuptools]

tests/test_api.py

@@ -9,7 +9,7 @@ from twitchdl import twitch
 from twitchdl.commands.download import get_clip_authenticated_url
 from twitchdl.commands.videos import get_game_ids
 from twitchdl.exceptions import ConsoleError
-from twitchdl.playlists import enumerate_vods, load_m3u8, parse_playlists
+from twitchdl.playlists import parse_playlists
 
 TEST_CHANNEL = "bananasaurus_rex"
 
@@ -37,10 +37,6 @@ def test_get_videos():
     playlist_txt = httpx.get(playlist_url).text
     assert playlist_txt.startswith("#EXTM3U")
 
-    playlist_m3u8 = load_m3u8(playlist_txt)
-    vods = enumerate_vods(playlist_m3u8)
-    assert vods[0].path == "0.ts"
-
 
 def test_get_clips():
     """

tests/test_download.py

@@ -0,0 +1,90 @@
+from decimal import Decimal
+
+from twitchdl.commands.download import filter_vods
+from twitchdl.playlists import Vod
+
+VODS = [
+    Vod(index=1, path="1.ts", duration=Decimal("10.0")),
+    Vod(index=2, path="2.ts", duration=Decimal("10.0")),
+    Vod(index=3, path="3.ts", duration=Decimal("10.0")),
+    Vod(index=4, path="4.ts", duration=Decimal("10.0")),
+    Vod(index=5, path="5.ts", duration=Decimal("10.0")),
+    Vod(index=6, path="6.ts", duration=Decimal("10.0")),
+    Vod(index=7, path="7.ts", duration=Decimal("10.0")),
+    Vod(index=8, path="8.ts", duration=Decimal("10.0")),
+    Vod(index=9, path="9.ts", duration=Decimal("10.0")),
+    Vod(index=10, path="10.ts", duration=Decimal("3.15")),
+]
+
+
+def test_filter_vods_no_start_no_end():
+    vods, start_offset, duration = filter_vods(VODS, None, None)
+    assert vods == VODS
+    assert start_offset == Decimal("0")
+    assert duration == Decimal("93.15")
+
+
+def test_filter_vods_start():
+    # Zero offset
+    vods, start_offset, duration = filter_vods(VODS, 0, None)
+    assert [v.index for v in vods] == [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
+    assert start_offset == Decimal("0")
+    assert duration == Decimal("93.15")
+
+    # Mid-vod
+    vods, start_offset, duration = filter_vods(VODS, 13, None)
+    assert [v.index for v in vods] == [2, 3, 4, 5, 6, 7, 8, 9, 10]
+    assert start_offset == Decimal("3.0")
+    assert duration == Decimal("80.15")
+
+    # Between vods
+    vods, start_offset, duration = filter_vods(VODS, 50, None)
+    assert [v.index for v in vods] == [6, 7, 8, 9, 10]
+    assert start_offset == Decimal("0")
+    assert duration == Decimal("43.15")
+
+    # Close to end
+    vods, start_offset, duration = filter_vods(VODS, 93, None)
+    assert [v.index for v in vods] == [10]
+    assert start_offset == Decimal("3.0")
+    assert duration == Decimal("0.15")
+
+
+def test_filter_vods_end():
+    # Zero offset
+    vods, start_offset, duration = filter_vods(VODS, 0, None)
+    assert [v.index for v in vods] == [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
+    assert start_offset == Decimal("0")
+    assert duration == Decimal("93.15")
+
+    # Mid-vod
+    vods, start_offset, duration = filter_vods(VODS, None, 56)
+    assert [v.index for v in vods] == [1, 2, 3, 4, 5, 6]
+    assert start_offset == Decimal("0")
+    assert duration == Decimal("56")
+
+    # Between vods
+    vods, start_offset, duration = filter_vods(VODS, None, 30)
+    assert [v.index for v in vods] == [1, 2, 3]
+    assert start_offset == Decimal("0")
+    assert duration == Decimal("30")
+
+
+def test_filter_vods_start_end():
+    # Zero offset
+    vods, start_offset, duration = filter_vods(VODS, 0, 0)
+    assert [v.index for v in vods] == [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
+    assert start_offset == Decimal("0")
+    assert duration == Decimal("93.15")
+
+    # Mid-vod
+    vods, start_offset, duration = filter_vods(VODS, 32, 56)
+    assert [v.index for v in vods] == [4, 5, 6]
+    assert start_offset == Decimal("2")
+    assert duration == Decimal("24")
+
+    # Between vods
+    vods, start_offset, duration = filter_vods(VODS, 20, 60)
+    assert [v.index for v in vods] == [3, 4, 5, 6]
+    assert start_offset == Decimal("0")
+    assert duration == Decimal("40")

tests/test_patterns.py

@@ -7,7 +7,6 @@ TEST_VIDEO_PATTERNS = [
     ("702689313", "https://twitch.tv/videos/702689313"),
     ("702689313", "https://www.twitch.tv/videos/702689313"),
     ("702689313", "https://m.twitch.tv/videos/702689313"),
-    ("2223719525", "https://www.twitch.tv/r0dn3y/video/2223719525"),
 ]
 
 TEST_CLIP_PATTERNS = {

tests/test_progress.py

@@ -8,8 +8,8 @@ def test_initial_values():
     assert progress.progress_perc == 0
     assert progress.remaining_time is None
     assert progress.speed is None
-    assert progress.file_count == 10
-    assert progress.downloaded_count == 0
+    assert progress.vod_count == 10
+    assert progress.vod_downloaded_count == 0
 
 
 def test_downloaded():
@@ -96,16 +96,16 @@ def test_vod_downloaded_count():
     progress.start(2, 100)
     progress.start(3, 100)
 
-    assert progress.downloaded_count == 0
+    assert progress.vod_downloaded_count == 0
 
     progress.advance(1, 100)
     progress.end(1)
-    assert progress.downloaded_count == 1
+    assert progress.vod_downloaded_count == 1
 
     progress.advance(2, 100)
     progress.end(2)
-    assert progress.downloaded_count == 2
+    assert progress.vod_downloaded_count == 2
 
     progress.advance(3, 100)
     progress.end(3)
-    assert progress.downloaded_count == 3
+    assert progress.vod_downloaded_count == 3

twitchdl/cli.py

@@ -2,14 +2,12 @@ import logging
 import platform
 import re
 import sys
-from pathlib import Path
 from typing import Optional, Tuple
 
 import click
 
 from twitchdl import __version__
 from twitchdl.entities import DownloadOptions
-from twitchdl.naming import DEFAULT_OUTPUT_TEMPLATE
 from twitchdl.twitch import ClipsPeriod, VideosSort, VideosType
 
 # Tweak the Click context
@@ -81,12 +79,11 @@ def validate_rate(_ctx: click.Context, _param: click.Parameter, value: str) -> O
 
 
 @click.group(context_settings=CONTEXT)
-@click.option("--debug/--no-debug", default=False, help="Enable debug logging to stderr")
-@click.option("--verbose/--no-verbose", default=False, help="More verbose debug logging")
+@click.option("--debug/--no-debug", default=False, help="Log debug info to stderr")
 @click.option("--color/--no-color", default=sys.stdout.isatty(), help="Use ANSI color in output")
 @click.version_option(package_name="twitch-dl")
 @click.pass_context
-def cli(ctx: click.Context, color: bool, debug: bool, verbose: bool):
+def cli(ctx: click.Context, color: bool, debug: bool):
     """twitch-dl - twitch.tv downloader
 
     https://twitch-dl.bezdomni.net/
@@ -94,7 +91,7 @@ def cli(ctx: click.Context, color: bool, debug: bool, verbose: bool):
     ctx.color = color
 
     if debug:
-        logging.basicConfig(level=logging.DEBUG if verbose else logging.INFO)
+        logging.basicConfig(level=logging.DEBUG)
         logging.getLogger("httpx").setLevel(logging.WARN)
         logging.getLogger("httpcore").setLevel(logging.WARN)
 
@@ -142,18 +139,6 @@ def cli(ctx: click.Context, color: bool, debug: bool, verbose: bool):
     default="all_time",
     type=click.Choice(["last_day", "last_week", "last_month", "all_time"]),
 )
-@click.option(
-    "-t",
-    "--target-dir",
-    help="Target directory when downloading clips",
-    type=click.Path(
-        file_okay=False,
-        readable=False,
-        writable=True,
-        path_type=Path,
-    ),
-    default=Path(),
-)
 @json_option
 def clips(
     channel_name: str,
@@ -164,14 +149,10 @@ def clips(
     limit: Optional[int],
     pager: Optional[int],
     period: ClipsPeriod,
-    target_dir: Path,
 ):
     """List or download clips for given CHANNEL_NAME."""
     from twitchdl.commands.clips import clips
 
-    if not target_dir.exists():
-        target_dir.mkdir(parents=True, exist_ok=True)
-
     clips(
         channel_name,
         all=all,
@@ -181,7 +162,6 @@ def clips(
         limit=limit,
         pager=pager,
         period=period,
-        target_dir=target_dir,
     )
 
 
@@ -249,7 +229,7 @@ def clips(
     "-o",
     "--output",
     help="Output file name template. See docs for details.",
-    default=DEFAULT_OUTPUT_TEMPLATE,
+    default="{date}_{id}_{channel_login}_{title_slug}.{format}",
 )
 @click.option(
     "-q",

twitchdl/commands/clips.py

@@ -1,15 +1,13 @@
 import re
 import sys
 from os import path
-from pathlib import Path
-from typing import Callable, Generator, List, Optional
+from typing import Callable, Generator, Optional
 
 import click
 
 from twitchdl import twitch, utils
 from twitchdl.commands.download import get_clip_authenticated_url
-from twitchdl.entities import VideoQuality
-from twitchdl.http import download_file
+from twitchdl.download import download_file
 from twitchdl.output import green, print_clip, print_clip_compact, print_json, print_paged, yellow
 from twitchdl.twitch import Clip, ClipsPeriod
 
@@ -24,7 +22,6 @@ def clips(
     limit: Optional[int] = None,
     pager: Optional[int] = None,
     period: ClipsPeriod = "all_time",
-    target_dir: Path = Path(),
 ):
     # Set different defaults for limit for compact display
     default_limit = 40 if compact else 10
@@ -38,7 +35,7 @@ def clips(
         return print_json(list(generator))
 
     if download:
-        return _download_clips(target_dir, generator)
+        return _download_clips(generator)
 
     print_fn = print_clip_compact if compact else print_clip
 
@@ -48,8 +45,8 @@ def clips(
     return _print_all(generator, print_fn, all)
 
 
-def _target_filename(clip: Clip, video_qualities: List[VideoQuality]):
-    url = video_qualities[0]["sourceURL"]
+def _target_filename(clip: Clip):
+    url = clip["videoQualities"][0]["sourceURL"]
     _, ext = path.splitext(url)
     ext = ext.lstrip(".")
 
@@ -70,27 +67,16 @@ def _target_filename(clip: Clip, video_qualities: List[VideoQuality]):
     return f"{name}.{ext}"
 
 
-def _download_clips(target_dir: Path, generator: Generator[Clip, None, None]):
-    if not target_dir.exists():
-        target_dir.mkdir(parents=True, exist_ok=True)
-
+def _download_clips(generator: Generator[Clip, None, None]):
     for clip in generator:
-        # videoQualities can be null in some circumstances, see:
-        # https://github.com/ihabunek/twitch-dl/issues/160
-        if not clip["videoQualities"]:
-            continue
+        target = _target_filename(clip)
 
-        target = target_dir / _target_filename(clip, clip["videoQualities"])
-
-        if target.exists():
+        if path.exists(target):
             click.echo(f"Already downloaded: {green(target)}")
         else:
-            try:
-                url = get_clip_authenticated_url(clip["slug"], "source")
-                click.echo(f"Downloading: {yellow(target)}")
-                download_file(url, target)
-            except Exception as ex:
-                click.secho(ex, err=True, fg="red")
+            url = get_clip_authenticated_url(clip["slug"], "source")
+            click.echo(f"Downloading: {yellow(target)}")
+            download_file(url, target)
 
 
 def _print_all(

twitchdl/commands/download.py

@@ -1,32 +1,34 @@
 import asyncio
+import os
 import platform
 import re
-import shlex
 import shutil
 import subprocess
 import tempfile
+from decimal import Decimal
+from os import path
 from pathlib import Path
-from typing import List, Optional
+from typing import Dict, List, Optional, Tuple
 from urllib.parse import urlencode, urlparse
 
 import click
 import httpx
 
 from twitchdl import twitch, utils
+from twitchdl.download import download_file
 from twitchdl.entities import DownloadOptions
 from twitchdl.exceptions import ConsoleError
-from twitchdl.http import download_all, download_file
-from twitchdl.naming import clip_filename, video_filename
+from twitchdl.http import download_all
 from twitchdl.output import blue, bold, green, print_log, yellow
 from twitchdl.playlists import (
-    enumerate_vods,
-    get_init_sections,
+    Vod,
     load_m3u8,
     make_join_playlist,
     parse_playlists,
+    parse_vods,
     select_playlist,
 )
-from twitchdl.twitch import Chapter, ClipAccessToken, Video
+from twitchdl.twitch import Chapter, Clip, ClipAccessToken, Video
 
 
 def download(ids: List[str], args: DownloadOptions):
@@ -50,16 +52,27 @@ def download_one(video: str, args: DownloadOptions):
     raise ConsoleError(f"Invalid input: {video}")
 
 
-def _join_vods(playlist_path: Path, target: Path, overwrite: bool, video: Video):
+def _join_vods(
+    playlist_path: str,
+    target: str,
+    overwrite: bool,
+    video: Video,
+    start_offset: int,
+    duration: int,
+):
     description = video["description"] or ""
     description = description.strip()
 
-    command: List[str] = [
+    command = [
         "ffmpeg",
         "-i",
-        str(playlist_path),
+        playlist_path,
         "-c",
         "copy",
+        "-ss",
+        str(start_offset),
+        "-to",
+        str(duration),
         "-metadata",
         f"artist={video['creator']['displayName']}",
         "-metadata",
@@ -73,19 +86,19 @@ def _join_vods(playlist_path: Path, target: Path, overwrite: bool, video: Video)
         "warning",
         f"file:{target}",
     ]
-
     if overwrite:
         command.append("-y")
 
-    click.secho(f"{shlex.join(command)}", dim=True)
+    click.secho(f"{' '.join(command)}", dim=True)
+
     result = subprocess.run(command)
     if result.returncode != 0:
         raise ConsoleError("Joining files failed")
 
 
-def _concat_vods(vod_paths: List[Path], target: Path):
+def _concat_vods(vod_paths: List[str], target: str):
     tool = "type" if platform.system() == "Windows" else "cat"
-    command = [tool] + [str(p) for p in vod_paths]
+    command = [tool] + vod_paths
 
     with open(target, "wb") as target_file:
         result = subprocess.run(command, stdout=target_file)
@@ -93,12 +106,71 @@ def _concat_vods(vod_paths: List[Path], target: Path):
             raise ConsoleError(f"Joining files failed: {result.stderr}")
 
 
-def _crete_temp_dir(base_uri: str) -> Path:
+def get_video_placeholders(video: Video, format: str) -> Dict[str, str]:
+    date, time = video["publishedAt"].split("T")
+    game = video["game"]["name"] if video["game"] else "Unknown"
+
+    return {
+        "channel": video["creator"]["displayName"],
+        "channel_login": video["creator"]["login"],
+        "date": date,
+        "datetime": video["publishedAt"],
+        "format": format,
+        "game": game,
+        "game_slug": utils.slugify(game),
+        "id": video["id"],
+        "time": time,
+        "title": utils.titlify(video["title"]),
+        "title_slug": utils.slugify(video["title"]),
+    }
+
+
+def _video_target_filename(video: Video, args: DownloadOptions):
+    subs = get_video_placeholders(video, args.format)
+
+    try:
+        return args.output.format(**subs)
+    except KeyError as e:
+        supported = ", ".join(subs.keys())
+        raise ConsoleError(f"Invalid key {e} used in --output. Supported keys are: {supported}")
+
+
+def _clip_target_filename(clip: Clip, args: DownloadOptions):
+    date, time = clip["createdAt"].split("T")
+    game = clip["game"]["name"] if clip["game"] else "Unknown"
+
+    url = clip["videoQualities"][0]["sourceURL"]
+    _, ext = path.splitext(url)
+    ext = ext.lstrip(".")
+
+    subs = {
+        "channel": clip["broadcaster"]["displayName"],
+        "channel_login": clip["broadcaster"]["login"],
+        "date": date,
+        "datetime": clip["createdAt"],
+        "format": ext,
+        "game": game,
+        "game_slug": utils.slugify(game),
+        "id": clip["id"],
+        "slug": clip["slug"],
+        "time": time,
+        "title": utils.titlify(clip["title"]),
+        "title_slug": utils.slugify(clip["title"]),
+    }
+
+    try:
+        return args.output.format(**subs)
+    except KeyError as e:
+        supported = ", ".join(subs.keys())
+        raise ConsoleError(f"Invalid key {e} used in --output. Supported keys are: {supported}")
+
+
+def _crete_temp_dir(base_uri: str) -> str:
     """Create a temp dir to store downloads if it doesn't exist."""
     path = urlparse(base_uri).path.lstrip("/")
     temp_dir = Path(tempfile.gettempdir(), "twitch-dl", path)
     temp_dir.mkdir(parents=True, exist_ok=True)
-    return temp_dir
+    return str(temp_dir)
 
 
 def _get_clip_url(access_token: ClipAccessToken, quality: Optional[str]) -> str:
@@ -161,10 +233,10 @@ def _download_clip(slug: str, args: DownloadOptions) -> None:
     duration = utils.format_duration(clip["durationSeconds"])
     click.echo(f"Found: {green(title)} by {yellow(user)}, playing {blue(game)} ({duration})")
 
-    target = Path(clip_filename(clip, args.output))
+    target = _clip_target_filename(clip, args)
     click.echo(f"Target: {blue(target)}")
 
-    if not args.overwrite and target.exists():
+    if not args.overwrite and path.exists(target):
         response = click.prompt("File exists. Overwrite? [Y/n]", default="Y", show_default=False)
         if response.lower().strip() != "y":
             raise click.Abort()
@@ -193,10 +265,10 @@ def _download_video(video_id: str, args: DownloadOptions) -> None:
 
     click.echo(f"Found: {blue(video['title'])} by {yellow(video['creator']['displayName'])}")
 
-    target = Path(video_filename(video, args.format, args.output))
+    target = _video_target_filename(video, args)
     click.echo(f"Output: {blue(target)}")
 
-    if not args.overwrite and target.exists():
+    if not args.overwrite and path.exists(target):
         response = click.prompt("File exists. Overwrite? [Y/n]", default="Y", show_default=False)
         if response.lower().strip() != "y":
             raise click.Abort()
@@ -216,7 +288,8 @@ def _download_video(video_id: str, args: DownloadOptions) -> None:
     print_log("Fetching playlist...")
     vods_text = http_get(playlist.url)
     vods_m3u8 = load_m3u8(vods_text)
-    vods = enumerate_vods(vods_m3u8, start, end)
+    all_vods = parse_vods(vods_m3u8)
+    vods, start_offset, duration = filter_vods(all_vods, start, end)
 
     if args.dry_run:
         click.echo("Dry run, video not downloaded.")
@@ -226,32 +299,19 @@ def _download_video(video_id: str, args: DownloadOptions) -> None:
     target_dir = _crete_temp_dir(base_uri)
 
     # Save playlists for debugging purposes
-    with open(target_dir / "playlists.m3u8", "w") as f:
+    with open(path.join(target_dir, "playlists.m3u8"), "w") as f:
         f.write(playlists_text)
-    with open(target_dir / "playlist.m3u8", "w") as f:
+    with open(path.join(target_dir, "playlist.m3u8"), "w") as f:
         f.write(vods_text)
 
-    init_sections = get_init_sections(vods_m3u8)
-    for uri in init_sections:
-        print_log(f"Downloading init section {uri}...")
-        download_file(f"{base_uri}{uri}", target_dir / uri)
-
-    print_log(f"Downloading {len(vods)} VODs using {args.max_workers} workers to {target_dir}")
+    click.echo(f"\nDownloading {len(vods)} VODs using {args.max_workers} workers to {target_dir}")
 
     sources = [base_uri + vod.path for vod in vods]
-    targets = [target_dir / f"{vod.index:05d}.ts" for vod in vods]
-
-    asyncio.run(
-        download_all(
-            zip(sources, targets),
-            args.max_workers,
-            rate_limit=args.rate_limit,
-            count=len(vods),
-        )
-    )
+    targets = [os.path.join(target_dir, f"{vod.index:05d}.ts") for vod in vods]
+    asyncio.run(download_all(sources, targets, args.max_workers, rate_limit=args.rate_limit))
 
     join_playlist = make_join_playlist(vods_m3u8, vods, targets)
-    join_playlist_path = target_dir / "playlist_downloaded.m3u8"
+    join_playlist_path = path.join(target_dir, "playlist_downloaded.m3u8")
     join_playlist.dump(join_playlist_path)  # type: ignore
     click.echo()
 
@@ -265,17 +325,44 @@ def _download_video(video_id: str, args: DownloadOptions) -> None:
         _concat_vods(targets, target)
     else:
         print_log("Joining files...")
-        _join_vods(join_playlist_path, target, args.overwrite, video)
+        _join_vods(join_playlist_path, target, args.overwrite, video, start_offset, duration)
 
     click.echo()
 
     if args.keep:
-        click.echo(f"Temporary files not deleted: {yellow(target_dir)}")
+        click.echo(f"Temporary files not deleted: {target_dir}")
     else:
         print_log("Deleting temporary files...")
         shutil.rmtree(target_dir)
 
-    click.echo(f"Downloaded: {green(target)}")
+    click.echo(f"\nDownloaded: {green(target)}")
+
+
+def filter_vods(
+    vods: List[Vod], start: Optional[int], end: Optional[int]
+) -> Tuple[List[Vod], Decimal, Decimal]:
+    vod_start = Decimal(0)
+    start_offset = Decimal(0)
+    end_offset = Decimal(0)
+    filtered_vods: List[Vod] = []
+
+    for vod in vods:
+        vod_end = vod_start + vod.duration
+
+        if (not start or vod_end > start) and (not end or vod_start < end):
+            filtered_vods.append(vod)
+
+        if start and start > vod_start and start < vod_end:
+            start_offset = start - vod_start
+
+        if end and end > vod_start and end < vod_end:
+            end_offset = vod_end - end
+
+        vod_start = vod_end
+
+    filtered_vod_duration = sum(v.duration for v in filtered_vods)
+    duration = filtered_vod_duration - start_offset - end_offset
+    return filtered_vods, start_offset, duration
 
 
 def http_get(url: str) -> str:

twitchdl/commands/info.py

@@ -4,9 +4,9 @@ import click
 import m3u8
 
 from twitchdl import twitch, utils
+from twitchdl.commands.download import get_video_placeholders
 from twitchdl.exceptions import ConsoleError
-from twitchdl.naming import video_placeholders
-from twitchdl.output import bold, dim, print_clip, print_json, print_log, print_table, print_video
+from twitchdl.output import bold, print_clip, print_json, print_log, print_table, print_video
 from twitchdl.playlists import parse_playlists
 from twitchdl.twitch import Chapter, Clip, Video
 
@@ -55,19 +55,9 @@ def video_info(video: Video, playlists: str, chapters: List[Chapter]):
     click.echo()
     print_video(video)
 
-    click.echo("Playlists:\n")
-
-    playlist_headers = ["Name", "Group", "Resolution", "URL"]
-    playlist_data = [
-        [
-            f"{p.name} {dim('source')}" if p.is_source else p.name,
-            p.group_id,
-            f"{p.resolution}",
-            p.url,
-        ]
-        for p in parse_playlists(playlists)
-    ]
-    print_table(playlist_headers, playlist_data)
+    click.echo("Playlists:")
+    for p in parse_playlists(playlists):
+        click.echo(f"{bold(p.name)} {p.url}")
 
     if chapters:
         click.echo()
@@ -77,7 +67,7 @@ def video_info(video: Video, playlists: str, chapters: List[Chapter]):
             duration = utils.format_time(chapter["durationMilliseconds"] // 1000)
             click.echo(f'{start} {bold(chapter["description"])} ({duration})')
 
-    placeholders = video_placeholders(video, format="mkv")
+    placeholders = get_video_placeholders(video, format="mkv")
     placeholders = [[f"{{{k}}}", v] for k, v in placeholders.items()]
     click.echo("")
     print_table(["Placeholder", "Value"], placeholders)
@@ -108,8 +98,5 @@ def clip_info(clip: Clip):
     click.echo()
     click.echo("Download links:")
 
-    if clip["videoQualities"]:
-        for q in clip["videoQualities"]:
-            click.echo(f"{bold(q['quality'])} [{q['frameRate']} fps] {q['sourceURL']}")
-    else:
-        click.echo("No download URLs found")
+    for q in clip["videoQualities"]:
+        click.echo(f"{bold(q['quality'])} [{q['frameRate']} fps] {q['sourceURL']}")

twitchdl/download.py

@@ -0,0 +1,37 @@
+import os
+
+import httpx
+
+from twitchdl.exceptions import ConsoleError
+
+CHUNK_SIZE = 1024
+CONNECT_TIMEOUT = 5
+RETRY_COUNT = 5
+
+
+def _download(url: str, path: str):
+    tmp_path = path + ".tmp"
+    size = 0
+    with httpx.stream("GET", url, timeout=CONNECT_TIMEOUT) as response:
+        with open(tmp_path, "wb") as target:
+            for chunk in response.iter_bytes(chunk_size=CHUNK_SIZE):
+                target.write(chunk)
+                size += len(chunk)
+
+    os.rename(tmp_path, path)
+    return size
+
+
+def download_file(url: str, path: str, retries: int = RETRY_COUNT):
+    if os.path.exists(path):
+        from_disk = True
+        return (os.path.getsize(path), from_disk)
+
+    from_disk = False
+    for _ in range(retries):
+        try:
+            return (_download(url, path), from_disk)
+        except httpx.RequestError:
+            pass
+
+    raise ConsoleError(f"Failed downloading after {retries} attempts: {url}")

twitchdl/entities.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import Any, List, Literal, Mapping, Optional, TypedDict
+from typing import Any, Mapping, Optional
 
 
 @dataclass
@@ -20,73 +20,6 @@ class DownloadOptions:
     max_workers: int
 
 
-ClipsPeriod = Literal["last_day", "last_week", "last_month", "all_time"]
-VideosSort = Literal["views", "time"]
-VideosType = Literal["archive", "highlight", "upload"]
-
-
-class AccessToken(TypedDict):
-    signature: str
-    value: str
-
-
-class User(TypedDict):
-    login: str
-    displayName: str
-
-
-class Game(TypedDict):
-    id: str
-    name: str
-
-
-class VideoQuality(TypedDict):
-    frameRate: str
-    quality: str
-    sourceURL: str
-
-
-class ClipAccessToken(TypedDict):
-    id: str
-    playbackAccessToken: AccessToken
-    videoQualities: List[VideoQuality]
-
-
-class Clip(TypedDict):
-    id: str
-    slug: str
-    title: str
-    createdAt: str
-    viewCount: int
-    durationSeconds: int
-    url: str
-    videoQualities: Optional[List[VideoQuality]]
-    game: Game
-    broadcaster: User
-
-
-class Video(TypedDict):
-    id: str
-    title: str
-    description: str
-    publishedAt: str
-    broadcastType: str
-    lengthSeconds: int
-    game: Game
-    creator: User
-
-
-class Chapter(TypedDict):
-    id: str
-    durationMilliseconds: int
-    positionMilliseconds: int
-    type: str
-    description: str
-    subDescription: str
-    thumbnailURL: str
-    game: Game
-
-
 # Type for annotating decoded JSON
 # TODO: make data classes for common structs
 Data = Mapping[str, Any]

twitchdl/http.py

@@ -3,12 +3,10 @@ import logging
 import os
 import time
 from abc import ABC, abstractmethod
-from pathlib import Path
-from typing import Iterable, Optional, Tuple
+from typing import List, Optional
 
 import httpx
 
-from twitchdl.exceptions import ConsoleError
 from twitchdl.progress import Progress
 
 logger = logging.getLogger(__name__)
@@ -73,7 +71,7 @@ async def download(
     client: httpx.AsyncClient,
     task_id: int,
     source: str,
-    target: Path,
+    target: str,
     progress: Progress,
     token_bucket: TokenBucket,
 ):
@@ -98,12 +96,12 @@ async def download_with_retries(
     semaphore: asyncio.Semaphore,
     task_id: int,
     source: str,
-    target: Path,
+    target: str,
     progress: Progress,
     token_bucket: TokenBucket,
 ):
     async with semaphore:
-        if target.exists():
+        if os.path.exists(target):
             size = os.path.getsize(target)
             progress.already_downloaded(task_id, size)
             return
@@ -121,13 +119,13 @@ async def download_with_retries(
 
 
 async def download_all(
-    source_targets: Iterable[Tuple[str, Path]],
+    sources: List[str],
+    targets: List[str],
     workers: int,
     *,
-    count: Optional[int] = None,
     rate_limit: Optional[int] = None,
 ):
-    progress = Progress(count)
+    progress = Progress(len(sources))
     token_bucket = LimitingTokenBucket(rate_limit) if rate_limit else EndlessTokenBucket()
     async with httpx.AsyncClient(timeout=TIMEOUT) as client:
         semaphore = asyncio.Semaphore(workers)
@@ -141,36 +139,6 @@ async def download_all(
                 progress,
                 token_bucket,
             )
-            for task_id, (source, target) in enumerate(source_targets)
+            for task_id, (source, target) in enumerate(zip(sources, targets))
         ]
         await asyncio.gather(*tasks)
-
-
-def download_file(url: str, target: Path, retries: int = RETRY_COUNT) -> None:
-    """Download URL to given target path with retries"""
-    error_message = ""
-    for r in range(retries):
-        try:
-            retry_info = f" (retry {r})" if r > 0 else ""
-            logger.info(f"Downloading {url} to {target}{retry_info}")
-            return _do_download_file(url, target)
-        except httpx.HTTPStatusError as ex:
-            logger.error(ex)
-            error_message = f"Server responded with HTTP {ex.response.status_code}"
-        except httpx.RequestError as ex:
-            logger.error(ex)
-            error_message = str(ex)
-
-    raise ConsoleError(f"Failed downloading after {retries} attempts: {error_message}")
-
-
-def _do_download_file(url: str, target: Path) -> None:
-    tmp_path = Path(str(target) + ".tmp")
-
-    with httpx.stream("GET", url, timeout=TIMEOUT, follow_redirects=True) as response:
-        response.raise_for_status()
-        with open(tmp_path, "wb") as f:
-            for chunk in response.iter_bytes(chunk_size=CHUNK_SIZE):
-                f.write(chunk)
-
-    os.rename(tmp_path, target)

twitchdl/naming.py

@@ -1,72 +0,0 @@
-import os
-from typing import Dict
-
-from twitchdl import utils
-from twitchdl.entities import Clip, Video
-from twitchdl.exceptions import ConsoleError
-
-DEFAULT_OUTPUT_TEMPLATE = "{date}_{id}_{channel_login}_{title_slug}.{format}"
-
-
-def video_filename(video: Video, format: str, output: str) -> str:
-    subs = video_placeholders(video, format)
-    return _format(output, subs)
-
-
-def video_placeholders(video: Video, format: str) -> Dict[str, str]:
-    date, time = video["publishedAt"].split("T")
-    game = video["game"]["name"] if video["game"] else "Unknown"
-
-    return {
-        "channel": video["creator"]["displayName"],
-        "channel_login": video["creator"]["login"],
-        "date": date,
-        "datetime": video["publishedAt"],
-        "format": format,
-        "game": game,
-        "game_slug": utils.slugify(game),
-        "id": video["id"],
-        "time": time,
-        "title": utils.titlify(video["title"]),
-        "title_slug": utils.slugify(video["title"]),
-    }
-
-
-def clip_filename(clip: Clip, output: str):
-    subs = clip_placeholders(clip)
-    return _format(output, subs)
-
-
-def clip_placeholders(clip: Clip) -> Dict[str, str]:
-    date, time = clip["createdAt"].split("T")
-    game = clip["game"]["name"] if clip["game"] else "Unknown"
-
-    if clip["videoQualities"]:
-        url = clip["videoQualities"][0]["sourceURL"]
-        _, ext = os.path.splitext(url)
-        ext = ext.lstrip(".")
-    else:
-        ext = "mp4"
-
-    return {
-        "channel": clip["broadcaster"]["displayName"],
-        "channel_login": clip["broadcaster"]["login"],
-        "date": date,
-        "datetime": clip["createdAt"],
-        "format": ext,
-        "game": game,
-        "game_slug": utils.slugify(game),
-        "id": clip["id"],
-        "slug": clip["slug"],
-        "time": time,
-        "title": utils.titlify(clip["title"]),
-        "title_slug": utils.slugify(clip["title"]),
-    }
-
-
-def _format(output: str, subs: Dict[str, str]) -> str:
-    try:
-        return output.format(**subs)
-    except KeyError as e:
-        supported = ", ".join(subs.keys())
-        raise ConsoleError(f"Invalid key {e} used in --output. Supported keys are: {supported}")

twitchdl/output.py

@@ -6,7 +6,7 @@ from typing import Any, Callable, Generator, List, Optional, TypeVar
 import click
 
 from twitchdl import utils
-from twitchdl.entities import Clip, Video
+from twitchdl.twitch import Clip, Video
 
 T = TypeVar("T")
 
@@ -46,8 +46,11 @@ def print_table(headers: List[str], data: List[List[str]]):
     underlines = ["-" * width for width in widths]
 
     def print_row(row: List[str]):
-        parts = (ljust(cell, widths[idx]) for idx, cell in enumerate(row))
-        click.echo("  ".join(parts).strip())
+        for idx, cell in enumerate(row):
+            width = widths[idx]
+            click.echo(ljust(cell, width), nl=False)
+            click.echo("  ", nl=False)
+        click.echo()
 
     print_row(headers)
     print_row(underlines)
@@ -105,12 +108,11 @@ def print_video(video: Video):
     if channel or playing:
         click.echo(" ".join([channel, playing]))
 
+    if video["description"]:
+        click.echo(f"Description: {video['description']}")
+
     click.echo(f"Published {blue(published_at)}  Length: {blue(length)} ")
     click.secho(url, italic=True)
-
-    if video["description"]:
-        click.echo(f"\nDescription:\n{video['description']}")
-
     click.echo()
 
 

twitchdl/playlists.py

@@ -3,8 +3,8 @@ Parse and manipulate m3u8 playlists.
 """
 
 from dataclasses import dataclass
-from pathlib import Path
-from typing import Generator, List, Optional, OrderedDict, Set
+from decimal import Decimal
+from typing import Generator, List, Optional, OrderedDict
 
 import click
 import m3u8
@@ -28,7 +28,7 @@ class Vod:
     """Ordinal number of the VOD in the playlist"""
     path: str
     """Path part of the VOD URL"""
-    duration: int
+    duration: Decimal
     """Segment duration in seconds"""
 
 
@@ -54,35 +54,17 @@ def load_m3u8(playlist_m3u8: str) -> m3u8.M3U8:
     return m3u8.loads(playlist_m3u8)
 
 
-def enumerate_vods(
-    document: m3u8.M3U8,
-    start: Optional[int] = None,
-    end: Optional[int] = None,
-) -> List[Vod]:
-    """Extract VODs for download from document."""
-    vods = []
-    vod_start = 0
-
-    for index, segment in enumerate(document.segments):
-        vod_end = vod_start + segment.duration
-
-        # `vod_end > start` is used here becuase it's better to download a bit
-        # more than a bit less, similar for the end condition
-        start_condition = not start or vod_end > start
-        end_condition = not end or vod_start < end
-
-        if start_condition and end_condition:
-            vods.append(Vod(index, segment.uri, segment.duration))
-
-        vod_start = vod_end
-
-    return vods
+def parse_vods(document: m3u8.M3U8) -> List[Vod]:
+    return [
+        Vod(index, segment.uri, Decimal(segment.duration))
+        for index, segment in enumerate(document.segments)
+    ]
 
 
 def make_join_playlist(
     playlist: m3u8.M3U8,
     vods: List[Vod],
-    targets: List[Path],
+    targets: List[str],
 ) -> m3u8.Playlist:
     """
     Make a modified playlist which references downloaded VODs
@@ -94,7 +76,7 @@ def make_join_playlist(
     playlist.segments.clear()
     for segment in org_segments:
         if segment.uri in path_map:
-            segment.uri = str(path_map[segment.uri].name)
+            segment.uri = path_map[segment.uri]
             playlist.segments.append(segment)
 
     return playlist
@@ -169,12 +151,3 @@ def _playlist_key(playlist: Playlist) -> int:
         pass
 
     return MAX
-
-
-def get_init_sections(playlist: m3u8.M3U8) -> Set[str]:
-    # TODO: we're ignoring initi_section.base_uri and bytes
-    return set(
-        segment.init_section.uri
-        for segment in playlist.segments
-        if segment.init_section is not None
-    )

twitchdl/progress.py

@@ -32,7 +32,7 @@ class Sample(NamedTuple):
 
 
 class Progress:
-    def __init__(self, file_count: Optional[int] = None):
+    def __init__(self, vod_count: int):
         self.downloaded: int = 0
         self.estimated_total: Optional[int] = None
         self.last_printed: Optional[float] = None
@@ -42,8 +42,8 @@ class Progress:
         self.samples: Deque[Sample] = deque(maxlen=1000)
         self.speed: Optional[float] = None
         self.tasks: Dict[TaskId, Task] = {}
-        self.file_count = file_count
-        self.downloaded_count: int = 0
+        self.vod_count = vod_count
+        self.vod_downloaded_count: int = 0
 
     def start(self, task_id: int, size: int):
         if task_id in self.tasks:
@@ -68,7 +68,7 @@ class Progress:
 
         self.tasks[task_id] = Task(task_id, size)
         self.progress_bytes += size
-        self.downloaded_count += 1
+        self.vod_downloaded_count += 1
         self.print()
 
     def abort(self, task_id: int):
@@ -89,15 +89,13 @@ class Progress:
                 f"Taks {task_id} ended with {task.downloaded}b downloaded, expected {task.size}b."
             )
 
-        self.downloaded_count += 1
+        self.vod_downloaded_count += 1
         self.print()
 
     def _recalculate(self):
-        if self.tasks and self.file_count:
-            self.estimated_total = int(mean(t.size for t in self.tasks.values()) * self.file_count)
-        else:
-            self.estimated_total = None
-
+        self.estimated_total = (
+            int(mean(t.size for t in self.tasks.values()) * self.vod_count) if self.tasks else None
+        )
         self.speed = self._calculate_speed()
         self.progress_perc = (
             int(100 * self.progress_bytes / self.estimated_total) if self.estimated_total else 0
@@ -123,15 +121,14 @@ class Progress:
     def print(self):
         now = time.time()
 
-        # Don't print more often than 10 times per second
-        if self.last_printed and now - self.last_printed < 0.1:
+        # Don't print more often than 5 times per second
+        if self.last_printed and now - self.last_printed < 0.2:
             return
 
         self._recalculate()
 
         clear_line()
-        total_label = f"/{self.file_count}" if self.file_count else ""
-        click.echo(f"Downloaded {self.downloaded_count}{total_label} VODs", nl=False)
+        click.echo(f"Downloaded {self.vod_downloaded_count}/{self.vod_count} VODs", nl=False)
         click.secho(f" {self.progress_perc}%", fg="blue", nl=False)
 
         if self.estimated_total is not None:

twitchdl/twitch.py

@@ -2,28 +2,83 @@
 Twitch API access.
 """
 
+import json
 import logging
-import random
 import time
-from typing import Any, Dict, Generator, List, Mapping, Optional, Tuple, Union
+from typing import Any, Dict, Generator, List, Literal, Mapping, Optional, Tuple, TypedDict, Union
 
 import click
 import httpx
 
 from twitchdl import CLIENT_ID
-from twitchdl.entities import (
-    AccessToken,
-    Chapter,
-    Clip,
-    ClipAccessToken,
-    ClipsPeriod,
-    Data,
-    Video,
-    VideosSort,
-    VideosType,
-)
+from twitchdl.entities import Data
 from twitchdl.exceptions import ConsoleError
-from twitchdl.utils import format_size
+
+ClipsPeriod = Literal["last_day", "last_week", "last_month", "all_time"]
+VideosSort = Literal["views", "time"]
+VideosType = Literal["archive", "highlight", "upload"]
+
+
+class AccessToken(TypedDict):
+    signature: str
+    value: str
+
+
+class User(TypedDict):
+    login: str
+    displayName: str
+
+
+class Game(TypedDict):
+    id: str
+    name: str
+
+
+class VideoQuality(TypedDict):
+    frameRate: str
+    quality: str
+    sourceURL: str
+
+
+class ClipAccessToken(TypedDict):
+    id: str
+    playbackAccessToken: AccessToken
+    videoQualities: List[VideoQuality]
+
+
+class Clip(TypedDict):
+    id: str
+    slug: str
+    title: str
+    createdAt: str
+    viewCount: int
+    durationSeconds: int
+    url: str
+    videoQualities: List[VideoQuality]
+    game: Game
+    broadcaster: User
+
+
+class Video(TypedDict):
+    id: str
+    title: str
+    description: str
+    publishedAt: str
+    broadcastType: str
+    lengthSeconds: int
+    game: Game
+    creator: User
+
+
+class Chapter(TypedDict):
+    id: str
+    durationMilliseconds: int
+    positionMilliseconds: int
+    type: str
+    description: str
+    subDescription: str
+    thumbnailURL: str
+    game: Game
 
 
 class GQLError(click.ClickException):
@@ -80,23 +135,22 @@ logger = logging.getLogger(__name__)
 
 
 def log_request(request: httpx.Request):
-    logger.info(f"--> {request.method} {request.url}")
+    logger.debug(f"--> {request.method} {request.url}")
     if request.content:
         logger.debug(f"--> {request.content}")
 
 
-def log_response(response: httpx.Response, duration_seconds: float):
+def log_response(response: httpx.Response, duration: float):
     request = response.request
-    duration = f"{int(1000 * duration_seconds)}ms"
-    size = format_size(len(response.content))
-    logger.info(f"<-- {request.method} {request.url} HTTP {response.status_code} {duration} {size}")
+    duration_ms = int(1000 * duration)
+    logger.debug(f"<-- {request.method} {request.url} HTTP {response.status_code} {duration_ms}ms")
     if response.content:
         logger.debug(f"<-- {response.content}")
 
 
-def gql_persisted_query(query: Data):
+def gql_post(query: str):
     url = "https://gql.twitch.tv/gql"
-    response = authenticated_post(url, json=query)
+    response = authenticated_post(url, content=query)
     gql_raise_on_error(response)
     return response.json()
 
@@ -184,18 +238,22 @@ def get_clip(slug: str) -> Optional[Clip]:
 
 
 def get_clip_access_token(slug: str) -> ClipAccessToken:
-    query = {
+    query = f"""
+    {{
         "operationName": "VideoAccessToken_Clip",
-        "variables": {"slug": slug},
-        "extensions": {
-            "persistedQuery": {
+        "variables": {{
+            "slug": "{slug}"
+        }},
+        "extensions": {{
+            "persistedQuery": {{
                 "version": 1,
-                "sha256Hash": "36b89d2507fce29e5ca551df756d27c1cfe079e2609642b4390aa4c35796eb11",
-            }
-        },
-    }
+                "sha256Hash": "36b89d2507fce29e5ca551df756d27c1cfe079e2609642b4390aa4c35796eb11"
+            }}
+        }}
+    }}
+    """
 
-    response = gql_persisted_query(query)
+    response = gql_post(query.strip())
     return response["data"]["clip"]
 
 
@@ -267,6 +325,23 @@ def channel_clips_generator(
     return _generator(clips, limit)
 
 
+def channel_clips_generator_old(channel_id: str, period: ClipsPeriod, limit: int):
+    cursor = ""
+    while True:
+        clips = get_channel_clips(channel_id, period, limit, after=cursor)
+
+        if not clips["edges"]:
+            break
+
+        has_next = clips["pageInfo"]["hasNextPage"]
+        cursor = clips["edges"][-1]["cursor"] if has_next else None
+
+        yield clips, has_next
+
+        if not cursor:
+            break
+
+
 def get_channel_videos(
     channel_id: str,
     limit: int,
@@ -392,12 +467,8 @@ def get_playlists(video_id: str, access_token: AccessToken) -> str:
             "allow_audio_only": "true",
             "allow_source": "true",
             "player": "twitchweb",
-            "platform": "web",
-            "supported_codecs": "av1,h265,h264",
-            "p": random.randint(1000000, 10000000),
         },
     )
-
     response.raise_for_status()
     return response.content.decode("utf-8")
 
@@ -432,7 +503,7 @@ def get_video_chapters(video_id: str) -> List[Chapter]:
         },
     }
 
-    response = gql_persisted_query(query)
+    response = gql_post(json.dumps(query))
     return list(_chapter_nodes(response["data"]["video"]["moments"]))
 
 

twitchdl/utils.py

@@ -85,7 +85,6 @@ def titlify(value: str) -> str:
 VIDEO_PATTERNS = [
     r"^(?P<id>\d+)?$",
     r"^https://(www\.|m\.)?twitch\.tv/videos/(?P<id>\d+)(\?.+)?$",
-    r"^https://(www\.|m\.)?twitch\.tv/\w+/video/(?P<id>\d+)(\?.+)?$",
 ]
 
 CLIP_PATTERNS = [