From ca3ea20283e77f82750800f0862f37ca64e0082e Mon Sep 17 00:00:00 2001
From: simon <simobilleter@gmail.com>
Date: Mon, 2 May 2022 09:48:52 +0700
Subject: [PATCH] add socket_timeout when passing check_formats to yt-dlp

---
 tubearchivist/home/src/download/queue.py          | 1 +
 tubearchivist/home/src/download/yt_dlp_handler.py | 1 +
 tubearchivist/home/src/index/generic.py           | 1 +
 3 files changed, 3 insertions(+)

diff --git a/tubearchivist/home/src/download/queue.py b/tubearchivist/home/src/download/queue.py
index 5ec4a19..57d520b 100644
--- a/tubearchivist/home/src/download/queue.py
+++ b/tubearchivist/home/src/download/queue.py
@@ -128,6 +128,7 @@ class PendingList(PendingIndex):
         "noplaylist": True,
         "writethumbnail": True,
         "simulate": True,
+        "socket_timeout": 3,
     }
 
     def __init__(self, youtube_ids=False):
diff --git a/tubearchivist/home/src/download/yt_dlp_handler.py b/tubearchivist/home/src/download/yt_dlp_handler.py
index 9712f95..816e58c 100644
--- a/tubearchivist/home/src/download/yt_dlp_handler.py
+++ b/tubearchivist/home/src/download/yt_dlp_handler.py
@@ -281,6 +281,7 @@ class VideoDownloader:
             "writethumbnail": False,
             "noplaylist": True,
             "check_formats": "selected",
+            "socket_timeout": 3,
         }
 
     def _build_obs_user(self):
diff --git a/tubearchivist/home/src/index/generic.py b/tubearchivist/home/src/index/generic.py
index c4dcac5..f384ec7 100644
--- a/tubearchivist/home/src/index/generic.py
+++ b/tubearchivist/home/src/index/generic.py
@@ -23,6 +23,7 @@ class YouTubeItem:
         "default_search": "ytsearch",
         "skip_download": True,
         "check_formats": "selected",
+        "socket_timeout": 3,
         "noplaylist": True,
     }