reformatting
Browse files- pytube/itags.py +28 -2
- tests/contrib/test_playlist.py +8 -2
pytube/itags.py
CHANGED
@@ -98,9 +98,32 @@ _3D = [82, 83, 84, 85, 100, 101, 102]
|
|
98 |
LIVE = [91, 92, 93, 94, 95, 96, 132, 151]
|
99 |
DASH_MP4_VIDEO = [133, 134, 135, 136, 137, 138, 160, 212, 264, 266, 298, 299]
|
100 |
DASH_MP4_AUDIO = [139, 140, 141, 256, 258, 325, 328]
|
101 |
-
DASH_WEBM_VIDEO = [
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
102 |
DASH_WEBM_AUDIO = [171, 172, 249, 250, 251]
|
103 |
|
|
|
104 |
def get_format_profile(itag: int) -> Dict:
|
105 |
"""Get additional format information for a given itag.
|
106 |
|
@@ -119,5 +142,8 @@ def get_format_profile(itag: int) -> Dict:
|
|
119 |
"is_3d": itag in _3D,
|
120 |
"is_hdr": itag in HDR,
|
121 |
"fps": 60 if itag in _60FPS else 30,
|
122 |
-
"is_dash": itag in DASH_MP4_VIDEO
|
|
|
|
|
|
|
123 |
}
|
|
|
98 |
LIVE = [91, 92, 93, 94, 95, 96, 132, 151]
|
99 |
DASH_MP4_VIDEO = [133, 134, 135, 136, 137, 138, 160, 212, 264, 266, 298, 299]
|
100 |
DASH_MP4_AUDIO = [139, 140, 141, 256, 258, 325, 328]
|
101 |
+
DASH_WEBM_VIDEO = [
|
102 |
+
167,
|
103 |
+
168,
|
104 |
+
169,
|
105 |
+
170,
|
106 |
+
218,
|
107 |
+
219,
|
108 |
+
278,
|
109 |
+
242,
|
110 |
+
243,
|
111 |
+
244,
|
112 |
+
245,
|
113 |
+
246,
|
114 |
+
247,
|
115 |
+
248,
|
116 |
+
271,
|
117 |
+
272,
|
118 |
+
302,
|
119 |
+
303,
|
120 |
+
308,
|
121 |
+
313,
|
122 |
+
315,
|
123 |
+
]
|
124 |
DASH_WEBM_AUDIO = [171, 172, 249, 250, 251]
|
125 |
|
126 |
+
|
127 |
def get_format_profile(itag: int) -> Dict:
|
128 |
"""Get additional format information for a given itag.
|
129 |
|
|
|
142 |
"is_3d": itag in _3D,
|
143 |
"is_hdr": itag in HDR,
|
144 |
"fps": 60 if itag in _60FPS else 30,
|
145 |
+
"is_dash": itag in DASH_MP4_VIDEO
|
146 |
+
or itag in DASH_MP4_AUDIO
|
147 |
+
or itag in DASH_WEBM_VIDEO
|
148 |
+
or itag in DASH_WEBM_AUDIO,
|
149 |
}
|
tests/contrib/test_playlist.py
CHANGED
@@ -23,6 +23,12 @@ def test_init_with_playlist_url():
|
|
23 |
|
24 |
|
25 |
def test_init_with_watch_url():
|
26 |
-
url =
|
|
|
|
|
|
|
27 |
playlist = Playlist(url)
|
28 |
-
assert
|
|
|
|
|
|
|
|
23 |
|
24 |
|
25 |
def test_init_with_watch_url():
|
26 |
+
url = (
|
27 |
+
"https://www.youtube.com/watch?v=1KeYzjILqDo&"
|
28 |
+
"list=PLynhp4cZEpTbRs_PYISQ8v_uwO0_mDg_X&index=2&t=661s"
|
29 |
+
)
|
30 |
playlist = Playlist(url)
|
31 |
+
assert (
|
32 |
+
playlist.playlist_url
|
33 |
+
== "https://www.youtube.com/playlist?list=PLynhp4cZEpTbRs_PYISQ8v_uwO0_mDg_X"
|
34 |
+
)
|