list music used in video if available

2 files changed, 36 insertions, 2 deletions

diff --git a/youtube/shared.css b/youtube/shared.css
index f99baa1..151111d 100644
--- a/youtube/shared.css
+++ b/youtube/shared.css
@@ -227,11 +227,15 @@ address{
         grid-column: 1 / span 2;
         grid-row: 6;
     }
+    .full-item .music-list{
+        grid-row:7;
+        grid-column: 1 / span 2;
+    }
     .full-item .comments{
-        grid-row: 7;
+        grid-row: 8;
     }
     .full-item .more-comments{
-        grid-row: 8;
+        grid-row: 9;
     }
 
 .medium-item-box{
diff --git a/youtube/watch.py b/youtube/watch.py
index c669f5e..cfe827b 100644
--- a/youtube/watch.py
+++ b/youtube/watch.py
@@ -296,6 +296,10 @@ def extract_info(downloader, *args, **kwargs):
     except YoutubeError as e:
         return str(e)
 
+music_list_table_row = Template('''<tr>
+    <td>$attribute</td>
+    <td>$value</td>
+''')
 def get_watch_page(query_string):
         id = urllib.parse.parse_qs(query_string)['v'][0]
         downloader = YoutubeDL(params={'youtube_include_dash_manifest':False})
@@ -343,7 +347,32 @@ def get_watch_page(query_string):
         else:
             related_videos_html = ''
 
+        music_list = info['music_list']
+        if len(music_list) == 0:
+            music_list_html = ''
+        else:
+            music_list_html = '''<hr>
+<table>
+    <caption>Music</caption>
+    <tr>
+        <th>Artist</th>
+        <th>Title</th>
+        <th>Album</th>
+    </tr>
+'''
+            for track in music_list:
+                music_list_html += '''<tr>\n'''
+                for attribute in ('artist', 'title', 'album'):
+                    try:
+                        value = track[attribute]
+                    except KeyError:
+                        music_list_html += '''<td></td>'''
+                    else:
+                        music_list_html += '''<td>''' + html.escape(value) + '''</td>'''
+                music_list_html += '''</tr>\n'''
+            music_list_html += '''</table>\n'''
 
+        
 
         download_options = ''
         for format in info['formats']:
@@ -371,5 +400,6 @@ def get_watch_page(query_string):
             related                 = related_videos_html,
             comments                = comments_html,
             more_comments_button    = more_comments_button,
+            music_list              = music_list_html,
         )
         return page
\ No newline at end of file