depot/third_party/nixpkgs/pkgs/development/python-modules/ansible-kernel/default.nix

{
  lib,
  buildPythonPackage,
  fetchPypi,
  writeText,
  setuptools,
  ipywidgets,
  six,
  docopt,
  tqdm,
  jupyter,
  psutil,
  pyyaml,
  ansible-runner,
  ansible,
  python,
}:

let
  kernelSpecFile = writeText "kernel.json" (
    builtins.toJSON {
      argv = [
        python.interpreter
        "-m"
        "ansible_kernel"
        "-f"
        "{connection_file}"
      ];
      codemirror_mode = "yaml";
      display_name = "Ansible";
      language = "ansible";
    }
  );
in
buildPythonPackage rec {
  pname = "ansible-kernel";
  version = "1.0.0";
  pyproject = true;

  src = fetchPypi {
    inherit pname version;
    hash = "sha256-UJjm9FpmXSznXtaIR2rVv5YJS/H83FvRkNz09vwoe0c=";
  };

  build-system = [ setuptools ];

  dependencies = [
    ipywidgets
    six
    docopt
    tqdm
    jupyter
    psutil
    pyyaml
    ansible-runner
    ansible
  ];

  postPatch = ''
    # remove when merged
    # https://github.com/ansible/ansible-jupyter-kernel/pull/82
    touch LICENSE.md

    # remove custom install
    sed -i "s/cmdclass={'install': Installer},//" setup.py
  '';

  # tests hang with launched kernel
  doCheck = false;

  # install kernel manually
  postInstall = ''
    mkdir -p $out/share/jupyter/kernels/ansible/
    ln -s ${kernelSpecFile} $out/share/jupyter/kernels/ansible/kernel.json
  '';

  meta = with lib; {
    description = "Ansible kernel for Jupyter";
    homepage = "https://github.com/ansible/ansible-jupyter-kernel";
    license = licenses.asl20;
    maintainers = [ ];
  };
}