Skip to content

Commit

Permalink
add missing convandtag.py file for previous commit
Browse files Browse the repository at this point in the history
  • Loading branch information
rickyrorton committed Jun 21, 2023
1 parent 8bb977a commit e5ba295
Showing 1 changed file with 123 additions and 0 deletions.
123 changes: 123 additions & 0 deletions convandtag.py
@@ -0,0 +1,123 @@
from mutagen.flac import FLAC, Picture
from mutagen.id3 import APIC, ID3, TALB, TIT2, TPE1, TPE2, TRCK, TYER
from mutagen.mp4 import MP4, MP4Cover
from mutagen.wave import WAVE
from pydub import AudioSegment
from os import path as ospath
from os import rename,remove
from urllib import request

def remove_sus_characters(name: str):
converted = "".join(
i
for i in name
if i
not in ("/", "\\", "?", "%", "*", ":", "|", '"', "<", ">", ".", ",", ";", "=")
)
return converted

def m4atagger(input_file, m4a, song, path, bitrate, icon_url, conv):
if not conv:
rename(input_file, m4a)
elif conv:
convert = AudioSegment.from_file(input_file)
convert.export(m4a, format="mp4", bitrate=bitrate)
remove(input_file)
iconname = ospath.join(
path,
remove_sus_characters(song["artists"][0]["name"] + "-" + song["name"]) + ".jpg",
)
request.urlretrieve(icon_url, iconname)
tags = MP4(m4a)
if not tags.tags:
tags.add_tags()
tags["\xa9nam"] = song["name"]
tags["\xa9alb"] = song["album"]["name"]
tags["\xa9ART"] = ", ".join([i["name"] for i in song["artists"]])
tags["aART"] = ", ".join([i["name"] for i in song["album"]["artists"]])
tags["\xa9day"] = song["album"]["release_date"][0:4]
tags["trkn"] = ((int(song["track_number"]), int(song["album"]["total_tracks"])),)
with open(iconname, "rb") as f:
tags["covr"] = [MP4Cover(f.read(), imageformat=MP4Cover.FORMAT_JPEG)]
tags.save()
remove(iconname)


def mp3convtagger(webm, mp3, song, path, bitrate, icon_url):
iconname = ospath.join(
path,
remove_sus_characters(song["artists"][0]["name"] + "-" + song["name"]) + ".jpg",
)
request.urlretrieve(icon_url, iconname)
convert = AudioSegment.from_file(webm)
convert.export(mp3, format="mp3", bitrate=bitrate)
tags = ID3(mp3)
tags.add(TIT2(encoding=3, text=[song["name"]]))
tags.add(TALB(encoding=3, text=[song["album"]["name"]]))
tags.add(TPE1(encoding=3, text=", ".join([i["name"] for i in song["artists"]])))
tags.add(
TPE2(encoding=3, text=", ".join([i["name"] for i in song["album"]["artists"]]))
)
tags.add(TYER(encoding=3, text=[song["album"]["release_date"][0:4]]))
tags.add(TRCK(encoding=3, text=[song["track_number"]]))
with open(iconname, "rb") as f:
tags.add(
APIC(encoding=3, mime="image/jpeg", type=3, desc="Cover", data=f.read())
)
tags.save(v2_version=3)
remove(webm)
remove(iconname)


def wavconvtagger(webm, wav, song, path, bitrate, icon_url):
iconname = ospath.join(
path,
remove_sus_characters(song["artists"][0]["name"] + "-" + song["name"]) + ".jpg",
)
request.urlretrieve(icon_url, iconname)
convert = AudioSegment.from_file(webm)
convert.export(wav, format="wav", bitrate=bitrate)
tags = WAVE(wav)
tags.add_tags()
tags = tags.tags
tags.add(TIT2(encoding=3, text=[song["name"]]))
tags.add(TALB(encoding=3, text=[song["album"]["name"]]))
tags.add(TPE1(encoding=3, text=", ".join([i["name"] for i in song["artists"]])))
tags.add(
TPE2(encoding=3, text=", ".join([i["name"] for i in song["album"]["artists"]]))
)
tags.add(TYER(encoding=3, text=[song["album"]["release_date"][0:4]]))
tags.add(TRCK(encoding=3, text=[song["track_number"]]))
with open(iconname, "rb") as f:
tags.add(
APIC(encoding=3, mime="image/jpeg", type=3, desc="Cover", data=f.read())
)
tags.save(wav, v2_version=3)
remove(webm)
remove(iconname)


def flacconvtagger(webm, flac, song, path, bitrate, icon_url):
iconname = ospath.join(
path,
remove_sus_characters(song["artists"][0]["name"] + "-" + song["name"]) + ".jpg",
)
request.urlretrieve(icon_url, iconname)
convert = AudioSegment.from_file(webm)
convert.export(flac, format="flac", bitrate=bitrate)
tags = FLAC(flac)
tags["TITLE"] = song["name"]
tags["ARTIST"] = ", ".join([i["name"] for i in song["artists"]])
tags["ALBUMARTIST"] = ", ".join([i["name"] for i in song["album"]["artists"]])
tags["ALBUM"] = song["album"]["name"]
tags["DATE"] = song["album"]["release_date"][0:4]
tags["TRACKNUMBER"] = str(song["track_number"])
image = Picture()
image.type = 3
image.desc = "front cover"
with open(iconname, "rb") as f:
image.data = f.read()
tags.add_picture(image)
tags.save()
remove(webm)
remove(iconname)

0 comments on commit e5ba295

Please sign in to comment.