Example Client Setup

'''
Setting up the Rapidata client based on environment variables
'''
import dotenv
import os
from src.rapidata.rapidata_client.rapidata_client import RapidataClient


def setup_client():

    dotenv.load_dotenv()

    CLIENT_ID = os.getenv("CLIENT_ID")
    CLIENT_SECRET = os.getenv("CLIENT_SECRET")
    if not CLIENT_ID:
        raise Exception("CLIENT_ID not found in environment variables")

    if not CLIENT_SECRET:
        raise Exception("CLIENT_SECRET not found in environment variables")

    return RapidataClient(
        client_id=CLIENT_ID, client_secret=CLIENT_SECRET
    )