main.py

from userTaste.user_taste_spotify import SpotifyTaste
from videoRetriever.retriever_recent import RecentUploads
from videoWrapper.video_pytube_handler import PyTubeHandler
from metricAnalyzer.metric_analyzer_bpm import RegexBPM
from metricAnalyzer.metric_analyzer_key import LibrosaKey
from converter.converter_ffmpeg import FfmpegWrapper
import os

#callback function for using pytube - has restricted parameters
def postProcess(stream, filePath):
    global converter
    global paths

    directory, filename = os.path.split(filePath)
    name, extension = os.path.splitext(filename)
        
    spacedFilename = name.replace(' ','_')
    filteredName = filter(lambda x: x.isalnum() or (x=="_"), spacedFilename)
    parseableName = "".join(filteredName)

    parseablePath = f"{directory}/{parseableName}{extension}"

    os.rename(filePath, parseablePath)
    
    fileName = converter.convert(parseablePath)
    paths.append(fileName)

client_id = '9bf2211d3826492aa72471ec394689e1' #swap for different user
client_secret = input("Spotify client secret:")

spotify = SpotifyTaste(client_id, client_secret)
spotify.triggerCallback()
authUrl = input("Auth URL:")
spotify.authenticate(authUrl)

artists = spotify.getSearchTerms()

retriever = RecentUploads()
videoLinks = retriever.search(artists)

bpmProcessor = RegexBPM()
keyProcessor = LibrosaKey()
converter = FfmpegWrapper()

paths = []

for link in videoLinks:
    try:
        videoWrapper = PyTubeHandler(link, bpmProcessor, postProcess)
        videoWrapper.download('C:/Users/m-sus/Desktop/sound/anewtypeoflove/proceeds/')
    except:
        print(f"error w {link}")

for path in paths:
    try:
        fullPath = path + ".mp3"
        key = keyProcessor.analyse(fullPath)
        converter.rename(fullPath, key)
    except:
        print(f"error on {path}")