z230

Euni/euni_stahni.py

@@ -376,21 +376,23 @@ def stahni_dokument(s, url, out_dir: Path, label=""):
     return ("staženo", dest.name)
 
 
-def stahni_video(embed, out_dir: Path, referer):
-    """Stáhne video přes yt-dlp; soukromé/nedostupné přeskočí. Vrací (stav, info)."""
+def stahni_video(embed, out_dir: Path, referer, fmt="bestvideo*+bestaudio/best",
+                 frags=10):
+    """Stáhne video přes yt-dlp; soukromé/nedostupné přeskočí. Vrací (stav, info, fp)."""
     if sv is None:
-        return ("chyba", "modul stahni_video není dostupný")
+        return ("chyba", "modul stahni_video není dostupný", None)
     try:
         import yt_dlp
         from yt_dlp.utils import DownloadError
     except ImportError:
-        return ("chyba", "yt-dlp není nainstalován")
+        return ("chyba", "yt-dlp není nainstalován", None)
 
     out_dir.mkdir(parents=True, exist_ok=True)
     ff_dir = sv.priprav_ffmpeg()
     opts = {
         "outtmpl": str(out_dir / "%(title)s [%(id)s].%(ext)s"),
-        "format": "bestvideo*+bestaudio/best",
+        "format": fmt,
+        "concurrent_fragment_downloads": frags,  # paralelní HLS fragmenty = rychlejší
         "merge_output_format": "mp4",
         "logger": sv._TichyLogger(),
         "progress_hooks": [sv._progress_hook],
@@ -458,6 +460,10 @@ def main():
                    help="přeskočí scrape, použije existující euni_kurzy.json")
     p.add_argument("--no-videos", action="store_true", help="nestahovat videa")
     p.add_argument("--no-docs", action="store_true", help="nestahovat dokumenty")
+    p.add_argument("--video-format", default="bestvideo*+bestaudio/best",
+                   help="yt-dlp formát videa (např. \"bestvideo[height<=720]+bestaudio/best\")")
+    p.add_argument("--frags", type=int, default=10,
+                   help="počet paralelně stahovaných HLS fragmentů videa (default 10)")
     p.add_argument("--limit", type=int, default=0, help="jen prvních N kurzů (test)")
     p.add_argument("--out", default=str(SKRIPT_DIR / "stazeno"), help="výstupní adresář")
     p.add_argument("--json", default=str(SKRIPT_DIR / "euni_kurzy.json"),
@@ -602,7 +608,8 @@ def main():
         if not a.no_videos:
             for v in c.get("videos", []):
                 klic = material_klic("video", v)[0]
-                stav, info, fp = stahni_video(v["embed"], folder / "videa", c["url"])
+                stav, info, fp = stahni_video(v["embed"], folder / "videa", c["url"],
+                                              fmt=a.video_format, frags=a.frags)
                 if stav == "staženo":
                     stat["vid_ok"] += 1
                     print(f"    [VIDEO] {info}")