tvl-depot/users/Profpatsch/writers/default.nix
Profpatsch b046c77493 feat(Profpatsch/writers): add yants types to python3 writer
Uses the new restrict type to make sure flake errors start with an E.

Change-Id: I30369ade28e1ef612c91a368de2d5b128e6cf2a9
Reviewed-on: https://cl.tvl.fyi/c/depot/+/2313
Tested-by: BuildkiteCI
Reviewed-by: Profpatsch <mail@profpatsch.de>
2021-01-03 16:12:33 +00:00

61 lines
1.5 KiB
Nix

{ depot, pkgs, ... }:
let
bins = depot.nix.getBins pkgs.coreutils ["printf" "mkdir" "cat"];
inherit (depot.nix.yants) defun struct restrict attrs list string drv any;
FlakeError = restrict "flake error" (s: builtins.substring 0 1 s == "E") string;
Libraries = defun [ (attrs any) (list drv) ];
python3 = name: {
libraries ? (_: []),
flakeIgnore ? []
}: pkgs.writers.writePython3 name {
libraries = Libraries libraries pkgs.python3Packages;
flakeIgnore = list FlakeError flakeIgnore;
};
# TODO: add the same flake check as the pyhon3 writer
python3Lib = { name, libraries ? (_: []) }: moduleString:
let srcTree = depot.nix.runExecline.local name { stdin = moduleString; } [
"importas" "out" "out"
"if" [ bins.mkdir "-p" "\${out}/${name}" ]
"if" [
"redirfd" "-w" "1" "\${out}/setup.py"
bins.printf ''
from distutils.core import setup
setup(
name='%s',
packages=['%s']
)
'' name name
]
"if" [
# redirect stdin to the init py
"redirfd" "-w" "1" "\${out}/${name}/__init__.py"
bins.cat
]
];
in pkgs.python3Packages.buildPythonPackage {
inherit name;
src = srcTree;
propagatedBuildInputs = libraries pkgs.python3Packages;
doCheck = false;
};
tests = import ./tests.nix {
inherit
depot
pkgs
python3
python3Lib
;
};
in {
inherit
python3
python3Lib
tests
;
}