store video_id in LiveStreamError

pytube/__main__.py

@@ -22,7 +22,7 @@ from pytube import Stream
 from pytube import StreamQuery
 from pytube.extract import apply_descrambler, apply_signature, get_ytplayer_config
 from pytube.helpers import install_proxy
-from pytube.exceptions import VideoUnavailable
+from pytube.exceptions import VideoUnavailable, LiveStreamError
 from pytube.monostate import OnProgress, OnComplete, Monostate
 
 logger = logging.getLogger(__name__)
@@ -162,17 +162,17 @@ class YouTube:
         which blocks for long periods of time.
 
         :rtype: None
-
         """
         self.watch_html = request.get(url=self.watch_url)
         if self.watch_html is None:
             raise VideoUnavailable(video_id=self.video_id)
         self.age_restricted = extract.is_age_restricted(self.watch_html)
-        if not self.age_restricted and (
-            "yt-badge-live" in self.watch_html
-            or "This video is private" in self.watch_html
-        ):
-            raise VideoUnavailable(video_id=self.video_id)
+
+        if not self.age_restricted:
+            if "yt-badge-live" in self.watch_html:
+                raise LiveStreamError(self.video_id)
+            if "This video is private" in self.watch_html:
+                raise VideoUnavailable(video_id=self.video_id)
 
         if self.age_restricted:
             if not self.embed_html:

pytube/exceptions.py

@@ -35,6 +35,15 @@ class RegexMatchError(ExtractError):
 class LiveStreamError(ExtractError):
     """Video is a live stream."""
 
+    def __init__(self, video_id: str):
+        """
+        :param str video_id:
+            A YouTube video identifier.
+        """
+        super().__init__(f"{video_id} is streaming live and cannot be loaded")
+
+        self.video_id = video_id
+
 
 class VideoUnavailable(PytubeError):
     """Video is unavailable."""

pytube/extract.py

@@ -231,7 +231,7 @@ def apply_signature(config_args: Dict, fmt: str, js: str) -> None:
             url: str = stream["url"]
         except KeyError:
             if live_stream:
-                raise LiveStreamError("Video is currently being streamed live")
+                raise LiveStreamError("UNKNOWN")
         # 403 Forbidden fix.
         if "signature" in url or (
             "s" not in stream and ("&sig=" in url or "&lsig=" in url)

tests/test_extract.py

@@ -77,3 +77,8 @@ def test_mime_type_codec():
 def test_mime_type_codec_with_no_match_should_error():
     with pytest.raises(RegexMatchError):
         extract.mime_type_codec("audio/webm")
+
+
+def test_get_ytplayer_config_with_no_match_should_error():
+    with pytest.raises(RegexMatchError):
+        extract.get_ytplayer_config("")