stream expiration property

pytube/request.py

@@ -8,7 +8,6 @@ from typing import Iterable, Dict, Optional
 from urllib.request import Request
 from urllib.request import urlopen
 
-
 logger = logging.getLogger(__name__)
 
 
@@ -54,7 +53,7 @@ def stream(
         response = _execute_request(url, method="GET", headers={"Range": range_header})
         if file_size == range_size:
             try:
-                content_range = response.info()['Content-Range']
+                content_range = response.info()["Content-Range"]
                 file_size = int(content_range.split("/")[1])
             except (KeyError, IndexError, ValueError) as e:
                 logger.error(e)

pytube/streams.py

@@ -9,9 +9,11 @@ has been renamed to accommodate DASH (which serves the audio and video
 separately).
 """
 
+from datetime import datetime
 import logging
 import os
 from typing import Dict, Tuple, Optional, BinaryIO
+from urllib.parse import parse_qs
 
 from pytube import extract
 from pytube import request
@@ -169,6 +171,11 @@ class Stream:
 
         return self.filesize
 
+    @property
+    def expiration(self) -> datetime:
+        expire = parse_qs(self.url.split("?")[1])["expire"][0]
+        return datetime.fromtimestamp(int(expire))
+
     @property
     def default_filename(self) -> str:
         """Generate filename based on the video title.

tests/test_streams.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 import os
 import random
+from datetime import datetime
 from unittest import mock
 from unittest.mock import MagicMock
 
@@ -59,6 +60,10 @@ def test_title(cipher_signature):
     assert cipher_signature.title == expected
 
 
+def test_expiration(cipher_signature):
+    assert cipher_signature.streams[0].expiration == datetime(2020, 1, 15, 21, 12, 5)
+
+
 def test_caption_tracks(presigned_video):
     assert len(presigned_video.caption_tracks) == 13