143 lines
3.8 KiB
Python
143 lines
3.8 KiB
Python
import sys
|
|
import os
|
|
from os.path import basename
|
|
import acoustid
|
|
import discogs_client as discogs
|
|
from string import digits
|
|
import musicbrainzngs as m
|
|
from mutagen.id3 import ID3, APIC, _util
|
|
from mutagen.mp3 import EasyMP3
|
|
from bs4 import BeautifulSoup
|
|
|
|
|
|
def improve_song_names(songs):
|
|
text_file = open("blacklist.txt", "r")
|
|
blacklistWords = text_file.read().splitlines()
|
|
blacklistWords = [word.lower() for word in blacklistWords]
|
|
|
|
improved_names = []
|
|
for song in songs:
|
|
searchText = song[0:-4]
|
|
# remove_digits = searchText.maketrans('', '', digits)
|
|
# searchText = searchText.translate(remove_digits)
|
|
for word in song[0:-4].split(' '):
|
|
if word.lower() in blacklistWords:
|
|
searchText = searchText.replace(word, '')
|
|
|
|
improved_names.append(searchText.strip())
|
|
|
|
return improved_names
|
|
|
|
|
|
def get_metadata_acoustid(song):
|
|
API_KEY = 'GkQibmzT1u'
|
|
try:
|
|
results = acoustid.match(API_KEY, song)
|
|
except acoustid.NoBackendError:
|
|
print("chromaprint library/tool not found", file=sys.stderr)
|
|
sys.exit(1)
|
|
except acoustid.FingerprintGenerationError:
|
|
print("fingerprint could not be calculated", file=sys.stderr)
|
|
sys.exit(1)
|
|
except acoustid.WebServiceError as exc:
|
|
print("web service request failed:", exc.message, file=sys.stderr)
|
|
sys.exit(1)
|
|
except acoustid.FingerprintSubmissionError:
|
|
print("Error")
|
|
|
|
first = True
|
|
for result in results:
|
|
if first:
|
|
first = False
|
|
else:
|
|
print(result)
|
|
score, rid, title, artist = result[0], result[1], result[2], result[3]
|
|
valid = True
|
|
|
|
if first is True:
|
|
print("No Data found for " + song + " .... Skipping it")
|
|
valid = False
|
|
score, rid, title, artist = None, None, None, None
|
|
|
|
return valid, score, rid, title, artist
|
|
|
|
|
|
def get_metadata_musicbrainz(rid, artist, title):
|
|
d = m.set_useragent(
|
|
"MetadataRepair",
|
|
"0.1",
|
|
"https://github.com/yash2696/python-musicbrainzngs/",
|
|
)
|
|
|
|
result = m.search_recordings(
|
|
artist=artist, recording=title, rid=rid, limit=5)
|
|
|
|
print(result)
|
|
|
|
|
|
def add_metadata(file_name, rid, title, artist):
|
|
|
|
return
|
|
|
|
|
|
def list_files():
|
|
files = []
|
|
return [f for f in os.listdir('.') if f.endswith('.mp3')]
|
|
|
|
|
|
def get_metadata_discogs(song_name):
|
|
d = discogs.Client('ExampleApplication/0.1',
|
|
user_token="mbNSnSrqdyJOKorCDBXjzzeYNhgulysWNJbwgBmk")
|
|
|
|
results = d.search(song_name, type="release")
|
|
first_result = results.page(1)[0]
|
|
|
|
# print(first_result)
|
|
title = first_result.title
|
|
|
|
# print(dir(first_result))
|
|
print(title)
|
|
print(first_result.artists[0].name)
|
|
print(first_result.images[0])
|
|
|
|
|
|
def main():
|
|
files = list_files()
|
|
|
|
# title, artist =
|
|
# get_metadata_discogs(
|
|
# "Jonas Blue - By Your Side (Feat Raye)")
|
|
|
|
# get_metadata_discogs(
|
|
# "Maroon 5-Animals")
|
|
# valid, score, rid, title, artist = get_metadata_acoustid(
|
|
# "18. Justin Timberlake - CAN'T STOP THE FEELING!.mp3")
|
|
# print(valid)
|
|
# print(score)
|
|
# print(files)
|
|
# for file_name in files:
|
|
# print("-------------" + file_name + "------------------")
|
|
# print()
|
|
# valid, score, rid, title, artist = get_metadata_acoustid(file_name)
|
|
# if valid == True:
|
|
# get_metadata_musicbrainz(rid, artist, title)
|
|
# add_metadata(file_name, rid, title, artist)
|
|
# print()
|
|
# break
|
|
|
|
files = improve_song_names(files)
|
|
# print(files)
|
|
count = 0
|
|
for file_name in files:
|
|
print("-------------" + file_name + "------------------")
|
|
print()
|
|
get_metadata_discogs(file_name)
|
|
count += 1
|
|
if count > 5:
|
|
break
|
|
print()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|