import json
import os
from src.helpers.api_data_provider import ApiDataProvider


class DataLoader:
    config = {}

    def __init__(self, path):
        self.config = self.load_test_data(path)

    def get(self, attr):
        return self.config.get(attr)

    @staticmethod
    def load_test_data(path):
        full_path = f"src/test_data/{path}"
        with open(full_path) as json_data:
            return json.load(json_data)

    @staticmethod
    def get_artist():
        api_data = ApiDataProvider()
        default_artist = api_data.get_artist_and_track_from_top_chart(vendor='spotify', market='global')[0]
        return os.environ.get("ARTIST", default_artist)