nficano commited on
Commit
4241e78
·
2 Parent(s): 938b4f8 172f2df

Merge branch 'master' of github.com:nficano/pytube

Browse files

* 'master' of github.com:nficano/pytube:
HDR & vp9.2 support.

Files changed (1) hide show
  1. pytube/itags.py +11 -1
pytube/itags.py CHANGED
@@ -69,6 +69,14 @@ ITAGS = {
69
  308: ('1440p', None),
70
  313: ('2160p', None),
71
  315: ('2160p', None),
 
 
 
 
 
 
 
 
72
 
73
  # DASH Audio
74
  139: (None, '48kbps'),
@@ -85,7 +93,8 @@ ITAGS = {
85
  328: (None, None),
86
  }
87
 
88
- _60FPS = [298, 299, 302, 303, 308, 315]
 
89
  _3D = [82, 83, 84, 85, 100, 101, 102]
90
  LIVE = [91, 92, 93, 94, 95, 96, 132, 151]
91
 
@@ -106,5 +115,6 @@ def get_format_profile(itag):
106
  'abr': bitrate,
107
  'is_live': itag in LIVE,
108
  'is_3d': itag in _3D,
 
109
  'fps': 60 if itag in _60FPS else 30,
110
  }
 
69
  308: ('1440p', None),
70
  313: ('2160p', None),
71
  315: ('2160p', None),
72
+ 330: ('144p', None),
73
+ 331: ('240p', None),
74
+ 332: ('360p', None),
75
+ 333: ('480p', None),
76
+ 334: ('720p', None),
77
+ 335: ('1080p', None),
78
+ 336: ('1440p', None),
79
+ 337: ('2160p', None),
80
 
81
  # DASH Audio
82
  139: (None, '48kbps'),
 
93
  328: (None, None),
94
  }
95
 
96
+ HDR = [330, 331, 332, 333, 334, 335, 336, 337]
97
+ _60FPS = [298, 299, 302, 303, 308, 315] + HDR
98
  _3D = [82, 83, 84, 85, 100, 101, 102]
99
  LIVE = [91, 92, 93, 94, 95, 96, 132, 151]
100
 
 
115
  'abr': bitrate,
116
  'is_live': itag in LIVE,
117
  'is_3d': itag in _3D,
118
+ 'is_hdr': itag in HDR,
119
  'fps': 60 if itag in _60FPS else 30,
120
  }