depot/third_party/nixpkgs/pkgs/development/python-modules/duckdb/default.nix
Default email e7ec2969af Project import generated by Copybara.
GitOrigin-RevId: 9b19f5e77dd906cb52dade0b7bd280339d2a1f3d
2024-01-13 09:15:51 +01:00

80 lines
1.5 KiB
Nix

{ lib
, buildPythonPackage
, duckdb
, fsspec
, google-cloud-storage
, numpy
, openssl
, pandas
, psutil
, pybind11
, setuptools-scm
, pytestCheckHook
}:
buildPythonPackage rec {
inherit (duckdb) pname version src;
format = "setuptools";
# 1. let nix control build cores
# 2. default to extension autoload & autoinstall disabled
# 3. unconstrain setuptools_scm version
patches = (duckdb.patches or []) ++ [ ./setup.patch ];
postPatch = (duckdb.postPatch or "") + ''
# we can't use sourceRoot otherwise patches don't apply, because the patches apply to the C++ library
cd tools/pythonpkg
substituteInPlace setup.py --subst-var NIX_BUILD_CORES
# avoid dependency on mypy
rm tests/stubs/test_stubs.py
'';
BUILD_HTTPFS = 1;
nativeBuildInputs = [
pybind11
setuptools-scm
];
buildInputs = [ openssl ];
propagatedBuildInputs = [
numpy
pandas
];
nativeCheckInputs = [
fsspec
google-cloud-storage
psutil
pytestCheckHook
];
disabledTests = [
# tries to make http request
"test_install_non_existent_extension"
# test is racy and interrupt can be delivered before or after target point
"test_connection_interrupt"
];
preCheck = ''
export HOME="$(mktemp -d)"
'';
setupPyBuildFlags = [
"--inplace"
];
pythonImportsCheck = [
"duckdb"
];
meta = with lib; {
description = "Python binding for DuckDB";
homepage = "https://duckdb.org/";
license = licenses.mit;
maintainers = with maintainers; [ cpcloud ];
};
}