musicpom/utils/get_reorganize_vars.py
2024-09-04 15:01:46 -04:00

28 lines
703 B
Python

from mutagen.id3 import ID3
def get_reorganize_vars(filepath: str) -> tuple[str, str]:
"""
Takes in a path to an audio file
returns the (artisr, album) as a tuple of strings
if no artist or album or ID3 tags at all are found,
function will return ("Unknown Artist", "Unknown Album")
"""
audio = ID3(filepath)
try:
artist = str(audio["TPE1"].text[0])
if artist == "":
artist = "Unknown Artist"
except KeyError:
artist = "Unknown Artist"
try:
album = str(audio["TALB"].text[0])
if album == "":
album = "Unknown Album"
except KeyError:
album = "Unknown Album"
return artist, album