init_spotify.py

import os
import json
import logging
import yaml

import spotipy.util as util
from langchain_community.utilities import TextRequestsWrapper

from utils import init_spotify

logger = logging.getLogger()


def main():
    config = yaml.load(open("config.yaml", "r"), Loader=yaml.FullLoader)
    os.environ["OPENAI_API_KEY"] = config["openai_api_key"]
    os.environ["SPOTIPY_CLIENT_ID"] = config["spotipy_client_id"]
    os.environ["SPOTIPY_CLIENT_SECRET"] = config["spotipy_client_secret"]
    os.environ["SPOTIPY_REDIRECT_URI"] = config["spotipy_redirect_uri"]

    with open("specs/spotify_oas.json") as f:
        raw_api_spec = json.load(f)

    scopes = list(
        raw_api_spec["components"]["securitySchemes"]["oauth_2_0"]["flows"][
            "authorizationCode"
        ]["scopes"].keys()
    )
    access_token = util.prompt_for_user_token(scope=",".join(scopes))
    headers = {"Authorization": f"Bearer {access_token}"}

    requests_wrapper = TextRequestsWrapper(headers=headers)

    init_spotify(requests_wrapper)


if __name__ == "__main__":
    main()