8000 ibis/flake.nix at main · goldmedal/ibis · GitHub
[go: up one dir, main page]
Skip to content

flake.nix

Latest commit

 

History

History
222 lines (197 loc) · 5.46 KB

flake.nix

File metadata and controls

222 lines (197 loc) · 5.46 KB
Raw
{
description = "Expressive Python analytics at any scale.";
inputs = {
flake-utils.url = "github:numtide/flake-utils";
gitignore = {
url = "github:hercules-ci/gitignore.nix";
inputs.nixpkgs.follows = "nixpkgs";
};
nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable-small";
pyproject-nix = {
url = "github:nix-community/pyproject.nix";
inputs.nixpkgs.follows = "nixpkgs";
};
uv2nix = {
url = "github:adisbladis/uv2nix";
inputs = {
pyproject-nix.follows = "pyproject-nix";
nixpkgs.follows = "nixpkgs";
};
};
pyproject-build-systems = {
url = "github:pyproject-nix/build-system-pkgs";
inputs = {
pyproject-nix.follows = "pyproject-nix";
uv2nix.follows = "uv2nix";
nixpkgs.follows = "nixpkgs";
};
};
};
outputs =
{
self,
flake-utils,
gitignore,
nixpkgs,
pyproject-nix,
uv2nix,
pyproject-build-systems,
...
}:
{
overlays.default = nixpkgs.lib.composeManyExtensions [
gitignore.overlay
(import ./nix/overlay.nix { inherit uv2nix pyproject-nix pyproject-build-systems; })
];
}
// flake-utils.lib.eachDefaultSystem (
localSystem:
let
pkgs = import nixpkgs {
inherit localSystem;
overlays = [ self.overlays.default ];
};
backendDevDeps = with pkgs; [
# impala UDFs
clang_15
cmake
ninja
# snowflake
openssl
# backend test suite
docker-compose
# visualization
graphviz-nox
# duckdb
duckdb
# mysql
mariadb-client
# pyodbc setup debugging
# in particular: odbcinst -j
unixODBC
# pyspark
openjdk17_headless
# postgres client
libpq.pg_config
postgresql
# sqlite with readline
sqlite-interactive
# mysqlclient build
libmysqlclient
pkg-config
];
shellHook = ''
rm -f "$PWD/ci/ibis-testing-data"
ln -s "${pkgs.ibisTestingData}" "$PWD/ci/ibis-testing-data"
# Undo dependency propagation by nixpkgs.
unset PYTHONPATH
# Get repository root using git. This is expanded at runtime by the editable `.pth` machinery.
export REPO_ROOT=$(git rev-parse --show-toplevel)
'';
preCommitDeps = with pkgs; [
actionlint
codespell
deadnix
git
just
nixfmt-rfc-style
nodejs.pkgs.prettier
shellcheck
shfmt
statix
taplo-cli
];
mkDevShell =
env:
pkgs.mkShell {
inherit (env) name;
packages =
[
# python dev environment
env
]
++ (with pkgs; [
# uv executable
uv
# rendering release notes
changelog
glow
# used in the justfile
jq
yj
# commit linting
commitlint
# link checking
lychee
# release automation
nodejs
# used in notebooks to download data
curl
# docs
quarto
])
++ preCommitDeps
++ backendDevDeps;
inherit shellHook;
PYSPARK_PYTHON = "${env}/bin/python";
AWS_PROFILE = "ibis-testing";
AWS_REGION = "us-east-2";
# needed for mssql+pyodbc
ODBCSYSINI = pkgs.writeTextDir "odbcinst.ini" ''
[FreeTDS]
Driver = ${pkgs.lib.makeLibraryPath [ pkgs.freetds ]}/libtdsodbc.so
'';
GDAL_DATA = "${pkgs.gdal}/share/gdal";
PROJ_DATA = "${pkgs.proj}/share/proj";
__darwinAllowLocalNetworking = true;
};
in
rec {
packages = {
default = packages.ibis313;
inherit (pkgs)
ibis310
ibis311
ibis312
ibis313
check-release-notes-spelling
;
};
checks = {
ibis310-pytest = pkgs.ibis310.passthru.tests.pytest;
ibis311-pytest = pkgs.ibis311.passthru.tests.pytest;
ibis312-pytest = pkgs.ibis312.passthru.tests.pytest;
ibis313-pytest = pkgs.ibis313.passthru.tests.pytest;
};
devShells = rec {
ibis310 = mkDevShell pkgs.ibisDevEnv310;
ibis311 = mkDevShell pkgs.ibisDevEnv311;
ibis312 = mkDevShell pkgs.ibisDevEnv312;
ibis313 = mkDevShell pkgs.ibisDevEnv313;
default = ibis313;
preCommit = pkgs.mkShell {
name = "preCommit";
packages = preCommitDeps ++ [ pkgs.ibisSmallDevEnv ];
};
links = pkgs.mkShell {
name = "links";
packages = with pkgs; [
just
lychee
];
};
release = pkgs.mkShell {
name = "release";
packages = with pkgs; [
git
uv
nodejs
unzip
gnugrep
(python3.withPackages (p: [ p.packaging ]))
];
};
};
}
);
}
0