# Copyright (C) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See LICENSE in project root for information.
import sys
if sys.version >= '3':
basestring = str
import pyspark
from pyspark import SparkContext
from pyspark import sql
from pyspark.ml.param.shared import *
from pyspark.sql import DataFrame
[docs]def streamToPowerBI(df, url, options=dict()):
jvm = SparkContext.getOrCreate()._jvm
writer = jvm.com.microsoft.ml.spark.io.powerbi.PowerBIWriter
return writer.stream(df.drop("label")._jdf, url, options)
setattr(pyspark.sql.DataFrame, 'streamToPowerBI', streamToPowerBI)
[docs]def writeToPowerBI(df, url, options=dict()):
jvm = SparkContext.getOrCreate()._jvm
writer = jvm.com.microsoft.ml.spark.io.powerbi.PowerBIWriter
writer.write(df._jdf, url, options)
setattr(pyspark.sql.DataFrame, 'writeToPowerBI', writeToPowerBI)