Browse Source

[youku] change format_id

PeterDing 10 years ago
parent
commit
08f7db20c1
1 changed files with 24 additions and 13 deletions
  1. 24 13
      youtube_dl/extractor/youku.py

+ 24 - 13
youtube_dl/extractor/youku.py

@@ -132,26 +132,37 @@ class YoukuIE(InfoExtractor):
 
 
     def get_hd(self, fm):
     def get_hd(self, fm):
         hd_id_dict = {
         hd_id_dict = {
-            'flv': '0',
-            'mp4': '1',
-            'hd2': '2',
-            'hd3': '3',
-            '3gp': '0',
-            '3gphd': '1'
+            'flv'   : '0',
+            'mp4'   : '1',
+            'hd2'   : '2',
+            'hd3'   : '3',
+            '3gp'   : '0',
+            '3gphd' : '1'
         }
         }
         return hd_id_dict[fm]
         return hd_id_dict[fm]
 
 
     def parse_ext_l(self, fm):
     def parse_ext_l(self, fm):
         ext_dict = {
         ext_dict = {
-            'flv': 'flv',
-            'mp4': 'mp4',
-            'hd2': 'flv',
-            'hd3': 'flv',
-            '3gp': 'flv',
-            '3gphd': 'mp4',
+            'flv'   : 'flv',
+            'mp4'   : 'mp4',
+            'hd2'   : 'flv',
+            'hd3'   : 'flv',
+            '3gp'   : 'flv',
+            '3gphd' : 'mp4'
         }
         }
         return ext_dict[fm]
         return ext_dict[fm]
 
 
+    def get_format_name(self, fm):
+        _dict = {
+            '3gp'   : 'h6',
+            '3gphd' : 'h5',
+            'flv'   : 'h4',
+            'mp4'   : 'h3',
+            'hd2'   : 'h2',
+            'hd3'   : 'h1'
+        }
+        return _dict[fm]
+
     def _real_extract(self, url):
     def _real_extract(self, url):
         mobj = re.match(self._VALID_URL, url)
         mobj = re.match(self._VALID_URL, url)
         video_id = mobj.group('id')
         video_id = mobj.group('id')
@@ -191,7 +202,7 @@ class YoukuIE(InfoExtractor):
                 entries[i]['formats'].append(
                 entries[i]['formats'].append(
                     {
                     {
                         'url': video_urls[i],
                         'url': video_urls[i],
-                        'format_id': fm,
+                        'format_id': self.get_format_name(fm),
                         'ext': self.parse_ext_l(fm),
                         'ext': self.parse_ext_l(fm),
                         'filesize': int(data1['segs'][fm][i]['size'])
                         'filesize': int(data1['segs'][fm][i]['size'])
                     }
                     }