vevo: fixing naming when there are featured artists

rscnt · Nov 24, 2016 · 8b27d83 · 8b27d83
1 parent 8eb7b5c
commit 8b27d83
Showing 1 changed file with 27 additions and 4 deletions.
diff --git a/youtube_dl/extractor/vevo.py b/youtube_dl/extractor/vevo.py
@@ -91,14 +91,30 @@ class VevoIE(VevoBaseIE):
         'info_dict': {
             'id': 'USUV71503000',
             'ext': 'mp4',
-            'title': 'K Camp - Till I Die',
+            'title': 'K Camp ft. T.I. - Till I Die',
             'age_limit': 18,
             'timestamp': 1449468000,
             'upload_date': '20151207',
             'uploader': 'K Camp',
             'track': 'Till I Die',
             'artist': 'K Camp',
-            'genre': 'Rap/Hip-Hop',
+            'genre': 'Hip-Hop',
+        },
+    }, {
+        'note': 'Featured test',
+        'url': 'https://www.vevo.com/watch/lemaitre/Wait/USUV71402190',
+        'md5': 'd28675e5e8805035d949dc5cf161071d',
+        'info_dict': {
+            'id': 'USUV71402190',
+            'ext': 'mp4',
+            'title': 'Lemaitre ft. LoLo - Wait',
+            'age_limit': 0,
+            'timestamp': 1413432000,
+            'upload_date': '20141016',
+            'uploader': 'Lemaitre',
+            'track': 'Wait',
+            'artist': 'Lemaitre',
+            'genre': 'Electronic',
         },
     }, {
         'note': 'Only available via webpage',
@@ -242,8 +258,15 @@ def _real_extract(self, url):
 
             timestamp = parse_iso8601(video_info.get('releaseDate'))
             artists = video_info.get('artists')
-            if artists:
-                artist = uploader = artists[0]['name']
+            for curr_artist in artists:
+                if 'role' in curr_artist:
+                    if curr_artist['role'] == 'Featured':
+                        featured_artist = curr_artist['name']
+                    elif curr_artist['role'] == 'Main':
+                        artist = uploader = curr_artist['name']
+                else:
+                    artist = uploader = curr_artist['name']
+                    break
             view_count = int_or_none(video_info.get('views', {}).get('total'))
 
             for video_version in video_versions: