def credential_provider():
config = Config(
host=os.getenv('DATABRICKS_HOST'),
client_id=os.getenv("DATABRICKS_CLIENT_ID"),
client_secret=os.getenv("DATABRICKS_CLIENT_SECRET"))
return oauth_service_principal(config)
def db_connector():
connection = sql.connect(
server_hostname=os.getenv('DATABRICKS_HOST'),
http_path=os.getenv('WAREHOUSE_HTTP_PATH'),
credentials_provider=credential_provider)
return connection
and in the api i am using it as
with db_connector() as connection:
with connection.cursor() as cursor:
results = cursor.execute(query).fetchall()