From c6d8c7e18e4bc9557007b32c26106e505fbf0a06 Mon Sep 17 00:00:00 2001
From: Lars Noack <lars.noack@outlook.de>
Date: Fri, 21 Oct 2022 10:56:22 +0200
Subject: [PATCH] fixed encoding bug with mp3 files

---
 src/download.py | 43 +++++++++++++++++++++++++++++++++++++------
 1 file changed, 37 insertions(+), 6 deletions(-)

diff --git a/src/download.py b/src/download.py
index f7af8df..41db5d2 100644
--- a/src/download.py
+++ b/src/download.py
@@ -3,7 +3,7 @@ import os.path
 import pandas as pd
 import mutagen
 from mutagen.easyid3 import EasyID3
-import mutagen.mp3
+from pydub import AudioSegment
 import json
 import logging
 
@@ -77,6 +77,38 @@ dict_keys(
         ])
 """
 
+ID3_TAGS = {
+    "TALB": "album",
+    "TBPM": "bpm",
+    "TCMP": "compilation",  # iTunes extension
+    "TCOM": "composer",
+    "TCOP": "copyright",
+    "TENC": "encodedby",
+    "TEXT": "lyricist",
+    "TLEN": "length",
+    "TMED": "media",
+    "TMOO": "mood",
+    "TIT1": "grouping",
+    "TIT2": "title",
+    "TIT3": "version",
+    "TPE1": "artist",
+    "TPE2": "albumartist",
+    "TPE3": "conductor",
+    "TPE4": "arranger",
+    "TPOS": "discnumber",
+    "TPUB": "organization",
+    "TRCK": "tracknumber",
+    "TOLY": "author",
+    "TSO2": "albumartistsort",  # iTunes extension
+    "TSOA": "albumsort",
+    "TSOC": "composersort",  # iTunes extension
+    "TSOP": "artistsort",
+    "TSOT": "titlesort",
+    "TSRC": "isrc",
+    "TSST": "discsubtitle",
+    "TLAN": "language",
+}
+
 class Download:
     def __init__(self, session: requests.Session = requests.Session(), file: str = ".cache3.csv", temp: str = "temp"):
         self.session = session
@@ -123,11 +155,9 @@ class Download:
         logging.info("finished")
 
     def write_metadata(self, row, filePath):
-        try:
-            audiofile = EasyID3(filePath)
-        except mutagen.id3.ID3NoHeaderError:
-            audiofile = mutagen.mp3.EasyMP3(filePath)
-            audiofile.add_tags()
+        AudioSegment.from_file(filePath).export(filePath, format="mp3")
+        
+        audiofile = EasyID3(filePath)
         
         valid_keys = list(EasyID3.valid_keys.keys())
 
@@ -142,6 +172,7 @@ class Download:
         audiofile.save(filePath, v1=2)
 
 
+
 if __name__ == "__main__":
     logging.basicConfig(level=logging.DEBUG)
     s = requests.Session()