Added functionality to also grab search terms

This commit is contained in:
Leonetienne
2022-01-30 21:02:18 +01:00
parent b37105afa9
commit 8daf6b5245
7 changed files with 240 additions and 53 deletions

View File

@@ -1,13 +1,26 @@
import yaml
import os
with open("artists.yaml", "r") as yamlfile:
with open("to-grab.yaml", "r") as yamlfile:
try:
config = yaml.safe_load(yamlfile)
for artist in config:
print(f"\033[92mGrabbing artist '{artist}'")
os.system(f"python3 grab.py '{artist}'")
# Grab artists
if "artists" in config:
for artist in config["artists"]:
print(f"\033[92mGrabbing artist '{artist}'")
os.system(f"python3 grab-artist.py '{artist}'")
# Grab search results
if "searches" in config:
for search in config["searches"]:
print(f"\033[92mGrabbing search results for '{search['terms']}'")
max_results = ""
if "max" in search:
max_results = search["max"]
os.system("python3 grab-search.py '" + search['terms'] + "' " + str(max_results))
except yaml.YAMLError as exc:
print("You fucked up the yaml format.")