1{ lib, jdk8, buildPythonPackage, fetchPypi, six, py4j }:
2
3buildPythonPackage rec {
4 pname = "databricks-connect";
5 version = "9.1.2";
6
7 src = fetchPypi {
8 inherit pname version;
9 sha256 = "9672aae60b299de58a527f320df45769cadf436398e21f4ce73424a25badb7a7";
10 };
11
12 sourceRoot = ".";
13
14 propagatedBuildInputs = [ py4j six jdk8 ];
15
16 # requires network access
17 doCheck = false;
18
19 prePatch = ''
20 substituteInPlace setup.py \
21 --replace "py4j==0.10.9" "py4j"
22 '';
23
24 preFixup = ''
25 substituteInPlace "$out/bin/find-spark-home" \
26 --replace find_spark_home.py .find_spark_home.py-wrapped
27 '';
28
29 pythonImportsCheck = [ "pyspark" "six" "py4j" ];
30
31 meta = with lib; {
32 description = "Client for connecting to remote Databricks clusters";
33 homepage = "https://pypi.org/project/databricks-connect";
34 license = licenses.databricks;
35 maintainers = with maintainers; [ kfollesdal ];
36 };
37}