From 116f7cf41432ffb394d1506afa9c581a68116e51 Mon Sep 17 00:00:00 2001
From: Lars Noack <lars.noack@outlook.de>
Date: Thu, 20 Oct 2022 16:20:48 +0200
Subject: [PATCH] added youtube as music src

---
 src/__pycache__/youtube_music.cpython-310.pyc | Bin 1648 -> 1755 bytes
 src/download.py                               |  15 ++++++++++++---
 src/youtube_music.py                          |   5 +++++
 3 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/src/__pycache__/youtube_music.cpython-310.pyc b/src/__pycache__/youtube_music.cpython-310.pyc
index a8840259fb17fec518e72c23a6727da827cd837b..b706e4ebeb026a5e22ea1b26264f8c2929204327 100644
GIT binary patch
delta 239
zcmeysbDNhhpO=@50SIKn1C!0zH}bWxXr%zTH4G_?k_@$sK$;0iGo>)6u%xi2u%)oi
zVNPdE;izFs;he)Vc>{~Cuy#2kBLfIV0&xmMFoP!7<PR(|@wbHB+;R(2(_Je{iV~Ac
z97|I&^KUWd78u`RHncR-WWL3momzQ|ub?P3Ew!j9H6=MeB{lh$0GwY~nwXPWQt772
zQp5su5m04*aY;c@esXGYaeh(pWD(YQDJh@?2O|d~69_OeF|z$-Vg-^vc(^CGvfkqr
PWMKlyaxn5Rvit%7x@kS#

delta 132
zcmcc3`+<iqpO=@50SNNl0+VO3ZRBfVk&6IwYZy`(B^hcNfix44W=dgBVX0wCVV%P~
zc@K-OuxdFYBLfIV0&xmMFoP!B<UcGj9B!J-MJzz=lgnA-g@k|%4n__}CMG6ECPucu
XJZzIcu-@b4VPOJ^axn5Rvit%7(O?)6

diff --git a/src/download.py b/src/download.py
index 89986be..f7af8df 100644
--- a/src/download.py
+++ b/src/download.py
@@ -1,7 +1,9 @@
 import requests
 import os.path
 import pandas as pd
+import mutagen
 from mutagen.easyid3 import EasyID3
+import mutagen.mp3
 import json
 import logging
 
@@ -90,6 +92,7 @@ class Download:
         for idx, row in self.dataframe.iterrows():
             row['artist'] = json.loads(row['artist'].replace("'", '"'))
             if self.path_stuff(row['path'], row['file']):
+                self.write_metadata(row, row['file'])
                 continue
 
             src = row['src']
@@ -119,18 +122,24 @@ class Download:
             mp3_file.write(r.content)
         logging.info("finished")
 
-    def write_metadata(self, row, file):
-        audiofile = EasyID3(file)
+    def write_metadata(self, row, filePath):
+        try:
+            audiofile = EasyID3(filePath)
+        except mutagen.id3.ID3NoHeaderError:
+            audiofile = mutagen.mp3.EasyMP3(filePath)
+            audiofile.add_tags()
         
         valid_keys = list(EasyID3.valid_keys.keys())
 
         for key in list(row.keys()):
             if key in valid_keys and row[key] is not None and not pd.isna(row[key]):
+                # print(key)
                 if type(row[key]) == int or type(row[key]) == float:
                     row[key] = str(row[key])
                 audiofile[key] = row[key]
 
-        audiofile.save()
+        print("saving")
+        audiofile.save(filePath, v1=2)
 
 
 if __name__ == "__main__":
diff --git a/src/youtube_music.py b/src/youtube_music.py
index 78d2c41..6f8eaf7 100644
--- a/src/youtube_music.py
+++ b/src/youtube_music.py
@@ -40,6 +40,11 @@ def download(row):
     file_ = row['file']
     options = {
         'format': 'bestaudio/best',
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'mp3',
+            'preferredquality': '192',
+        }],
         'keepvideo': False,
         'outtmpl': file_
     }