mirror of https://github.com/databricks/cli.git
37 lines
902 B
Python
37 lines
902 B
Python
#This file is autogenerated by the Databricks Extension for VS Code
|
|
import runpy
|
|
import sys
|
|
import os
|
|
|
|
python_file = {{.SourceFile}}
|
|
project_root = {{.ProjectRoot}}
|
|
#remove databricks args from argv
|
|
sys.argv = sys.argv[1:]
|
|
|
|
# change working directory
|
|
os.chdir(os.path.dirname(python_file))
|
|
|
|
# update python path
|
|
sys.path.insert(0, project_root)
|
|
|
|
# provide spark globals
|
|
user_ns = {
|
|
"display": display,
|
|
"displayHTML": displayHTML,
|
|
"dbutils": dbutils,
|
|
"table": table,
|
|
"sql": sql,
|
|
"udf": udf,
|
|
"getArgument": getArgument,
|
|
"sc": sc,
|
|
"spark": spark,
|
|
"sqlContext": sqlContext,
|
|
}
|
|
|
|
# Set log level to "ERROR". See https://kb.databricks.com/notebooks/cmd-c-on-object-id-p0.html
|
|
import logging; logger = spark._jvm.org.apache.log4j;
|
|
logging.getLogger("py4j.java_gateway").setLevel(logging.ERROR)
|
|
|
|
runpy.run_path(python_file, run_name="__main__", init_globals=user_ns)
|
|
None
|