11 Commits

Author SHA1 Message Date
e73e27b954 Remove old code.
Remove the old config setup code.
2024-10-16 11:22:04 -04:00
d7fb3f78f9 Simplify setup of gaa_conf.
Now done on one line in the main file.
2024-10-16 11:21:18 -04:00
811979cea7 Move configparser code to seperate module
This code is now in its own Class. This is to make it easier to do reporting and updating.
2024-10-16 11:20:14 -04:00
5856541a22 Display version number
The program will now output a version number with the use of either the 'version' or 'help' options.
2024-10-15 15:06:54 -04:00
26dfe543a1 Reimpliment delay to only trigger on 503 response
The code has been refactored to only trigger a one second delay when one of the APIs have returned a 503. This will allow the script to finish faster.
2024-10-15 08:52:22 -04:00
36a1f0ffab Reenable basic ratelimit.
A one second delay between requests to the fanart.tv servers. This is to prevent a 503 response that is given when requests are over the number per second allowed.
2024-10-14 23:33:33 -04:00
cbde52eb59 Merge branch 'master' into ratelimiting 2024-10-14 11:17:37 -04:00
883f594f4f Fix the output path of Artist Background.
The artist background had been saving to the folder the script was run from. This has now been resolved.
2024-10-14 11:08:17 -04:00
c5ffadd1f0 Get the artist background.
Adds the artist background as a possible source of artist image.
2024-10-14 10:34:43 -04:00
485b565a3f Replace working code in empty file.
api_calls.py had been emptied for some reason. Probably an acidential deletion. The old code has now been placed in the file.
2024-10-13 17:38:03 -04:00
386f17779b Changes the internal ratelimiting scheme.
Rather than ratelimit to one request per-second, attempt to detect the
MusicBrainz API ratelimiting. The API will return HTTP 503 errors once
ratelimiting has begun. Once this has been detected, the script will wait 1
second before continuing processing.
2024-10-03 14:53:52 -04:00
5 changed files with 146 additions and 139 deletions

71
api_calls.py Normal file
View File

@@ -0,0 +1,71 @@
import requests
import os
from time import sleep
def get_mb_id(artist_name, mb_confidence):
artist_name = artist_name.strip('_')
mb_url = f'https://musicbrainz.org/ws/2/artist?query=artist:"{artist_name}"&fmt=json'
header = {'User-Agent': 'get_artist_art.py/1.0'}
response = requests.get(mb_url, headers=header)
if response.status_code == 200:
mb_data = response.json()
if mb_data['count'] > 0:
if mb_data['artists'][0]['score'] > mb_confidence:
return True, mb_data['artists'][0]['id'], True
else:
print("No artist found of hight enough confidance.")
return False, "", True
else:
print("No artist found.")
return False, "", True
elif (response.status_code == 503):
sleep(1)
return False, "", False
else:
print(f"MB Error: {response.status_code}")
return False, "", True
def get_image(mb_id, ftv_api_key, artist_path):
ftv_api_url = f'https://webservice.fanart.tv/v3/music/{mb_id}?api_key={ftv_api_key}'
response = requests.get(ftv_api_url)
ftv_data =response.json()
if not ('status' in ftv_data):
if ('artistthumb' in ftv_data):
art_url = ftv_data['artistthumb'][0]['url']
print(art_url)
response = requests.get(art_url)
if response.status_code == 200:
with open(os.path.join(artist_path, 'artist.jpg'), 'wb') as f:
f.write(response.content)
return True
elif ('artistbackground' in ftv_data):
art_url = ftv_data['artistbackground'][0]['url']
response = requests.get(art_url)
if (response.status_code == 200):
with open(os.path.join(artist_path, 'artist.jpg'),'wb') as f:
f.write(response.content)
return True
elif ('hdmusiclogo' in ftv_data):
art_url = ftv_data['hdmusiclogo'][0]['url']
response = requests.get(art_url)
if response.status_code == 200:
with open(os.path.join(artist_path, 'artist.png'), 'wb') as f:
f.write(response.content)
return True
else:
print("Error downloading: ", response.status_code)
return True
else:
print("Thumb not found.")
return True
elif (response.status_code == 503):
sleep(1)
return False
else:
error_msg = ftv_data['error message']
if (error_msg == "503"):
sleep(1)
return False
else:
print(f"FTV Error: {error_msg}")
return True

10
dir_activities.py Normal file
View File

@@ -0,0 +1,10 @@
import os
def get_all(path):
return [name for name in os.listdir(path) if os.path.isdir(os.path.join(path, name))]
def has_artist_art(path):
if(os.path.exists(os.path.join(path, "artist.jpg")))or (os.path.exists(os.path.join(path, "artist.png"))):
return True
else:
return False

View File

@@ -1,139 +0,0 @@
#include <curl/curl.h>
#include <libconfig.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#define conf_file "config.ini"
#define mb_url \
"https://musicbrainz.org/ws/2/artist?query=artist:\"Slayer\"&fmt=json"
const char *get_conf_str(char set_key[]) {
const char *prog_conf = malloc(10 * sizeof(char));
config_t cfg;
config_setting_t *setting;
config_init(&cfg);
if (!config_read_file(&cfg, conf_file)) {
fprintf(stderr, "%s:%d - %s\n", config_error_file(&cfg),
config_error_line(&cfg), config_error_text(&cfg));
config_destroy(&cfg);
return "";
}
setting = config_lookup(&cfg, set_key);
prog_conf = config_setting_get_string(setting);
config_destroy(&cfg);
return prog_conf;
}
int get_conf_int(char set_key[]) {
int set_int;
config_t cfg;
config_setting_t *setting;
config_init(&cfg);
if (!config_read_file(&cfg, conf_file)) {
fprintf(stderr, "%s:%d - %s\n", config_error_file(&cfg),
config_error_line(&cfg), config_error_text(&cfg));
config_destroy(&cfg);
return 1;
}
setting = config_lookup(&cfg, set_key);
set_int = config_setting_get_int(setting);
config_destroy(&cfg);
return set_int;
}
void print_conf(const char m_dir[], const int mb_conf,
const char ftv_api_key[32]) {
if (m_dir != NULL) {
printf("The music directory is: %s\n", m_dir);
}
if (mb_conf != 1) {
printf("The lower limit for MB confidence is: %d\n", mb_conf);
}
if (ftv_api_key != NULL) {
printf("Your API key is: %s\n", ftv_api_key);
}
}
const char *get_mb_id() {
int response_code;
const char *mb_id;
CURL *curl;
CURLcode res;
char *buffer;
long res_len;
curl_global_init(CURL_GLOBAL_DEFAULT);
curl = curl_easy_init();
if (curl) {
curl_easy_setopt(curl, CURLOPT_URL, mb_url);
curl_easy_setopt(curl, CURLOPT_USERAGENT, "get_artist_art.py/1.0");
res = curl_easy_perform(curl);
if (res != CURLE_OK) {
fprintf(stderr, "curl_easy_perform() failed with error: %s\n",
curl_easy_strerror(res));
return "1";
}
curl_easy_getinfo(curl, CURLINFO_CONTENT_LENGTH_DOWNLOAD_T, &res_len);
buffer = malloc(res_len);
res = curl_easy_recv(curl, buffer, res_len, 0);
if (res != CURLE_OK) {
fprintf(stderr, "curl_easy_recv() failed with error: %s\n",
curl_easy_strerror(res));
return "1";
}
printf("Response body: %s\n", buffer);
free(buffer);
curl_easy_cleanup(curl);
}
return mb_id;
}
int main(int argc, char **argv) {
int mb_conf, opt;
const char *m_dir, *ftv_api_key;
m_dir = get_conf_str("dir");
mb_conf = get_conf_int("confidence");
ftv_api_key = get_conf_str("api_key");
while ((opt = getopt(argc, argv, "ps:")) != -1) {
switch (opt) {
case 'p':
print_conf(m_dir, mb_conf, ftv_api_key);
break;
case 's':
printf("This doesn't do anything yet but you searched for %s\n", optarg);
break;
default:
printf("So long and thanks for all the fish.\n");
break;
}
}
get_mb_id();
return EXIT_SUCCESS;
}

49
get_artist_art.py Executable file
View File

@@ -0,0 +1,49 @@
#!/usr/bin/env python3
import argparse
import os
from time import sleep
import dir_activities
import api_calls
from prog_conf import gaa_conf
gaa_version = "2024.10.15.2"
option_set = argparse.ArgumentParser(description=f"An automatic downloader of artist art using MusicBrainz and Fanart.TV.\nVersion: {gaa_version}")
jls_extract_var = f'%(prog)s {gaa_version}'
option_set.add_argument('--version', '-v', action='version', version=jls_extract_var)
cmd_options = option_set.parse_args()
config = gaa_conf().conf
music_path = config['music']['dir']
ftv_api_key = config['fanart_tv']['api_key']
mb_confidence = int(config['musicbrainz']['confidence'])
count = 1
dir_list = dir_activities.get_all(music_path)
dir_list.sort()
for artist in dir_list:
artist_path = os.path.join(music_path, artist)
if (not(dir_activities.has_artist_art(artist_path))):
# print(dir_activities.has_artist_art(artist_path))
print(str(count) + ": " + artist.strip('_'))
try:
ftv_response = False
mb_exit = False
while not mb_exit:
found_status, mb_id, mb_exit = api_calls.get_mb_id(artist, mb_confidence)
# print("Getting ", artist_image)
if found_status:
while not ftv_response:
ftv_response = api_calls.get_image(mb_id, ftv_api_key, artist_path)
print(ftv_response)
else:
print(f"{artist} returned no results.")
# api_requests.get_art(artist_image, artist, music_path)
except Exception as e:
print("Artist or art not found.")
print(e)
print("---------")
count += 1
#sleep(1)

16
prog_conf.py Normal file
View File

@@ -0,0 +1,16 @@
import configparser
from os.path import exists,join,expanduser
class gaa_conf:
def __init__(self):
self.conf = configparser.ConfigParser()
if (exists('config.ini')):
self.conf_path = 'config.ini'
else:
self.conf_path = join(expanduser(
"~"), ".local/share/get_artist_art/config.ini")
self.conf.read(self.conf_path)
def get():
return gaa_conf