mirror of https://github.com/databricks/cli.git
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
# This cell is autogenerated by the Databricks Extension for VS Code
|
|
def databricks_preamble():
|
|
from IPython import get_ipython
|
|
from typing import List
|
|
from shlex import quote
|
|
import os
|
|
|
|
src_file_dir = None
|
|
project_root_dir = None
|
|
|
|
src_file = "{{.SourceFile}}"
|
|
src_file_dir = os.path.dirname(src_file)
|
|
os.chdir(src_file_dir)
|
|
|
|
project_root_dir = "{{.ProjectRoot}}"
|
|
sys.path.insert(0, project_root_dir)
|
|
|
|
def parse_databricks_magic_lines(lines: List[str]):
|
|
if len(lines) == 0 or src_file_dir is None:
|
|
return lines
|
|
|
|
first = ""
|
|
for line in lines:
|
|
if len(line.strip()) != 0:
|
|
first = line
|
|
break
|
|
|
|
if first.startswith("%"):
|
|
magic = first.split(" ")[0].strip().strip("%")
|
|
rest = ' '.join(first.split(" ")[1:])
|
|
|
|
if magic == "sh":
|
|
return [
|
|
"%sh\n",
|
|
f"cd {quote(src_file_dir)}\n",
|
|
rest.strip() + "\n",
|
|
*lines[1:]
|
|
]
|
|
|
|
return lines
|
|
|
|
ip = get_ipython()
|
|
ip.input_transformers_cleanup.append(parse_databricks_magic_lines)
|
|
|
|
|
|
try:
|
|
databricks_preamble()
|
|
del databricks_preamble
|
|
except Exception as e:
|
|
print("Error in databricks_preamble: " + str(e))
|