python310Packages.papermill: add pythonImportsCheck

- disable on older Python releases
- add extra requirements

pkgs/development/python-modules/papermill/default.nix

@@ -1,54 +1,69 @@
 { lib
-, buildPythonPackage
-, fetchPypi
 , ansiwrap
+, azure-datalake-store
+, azure-storage-blob
+, boto3
+, buildPythonPackage
 , click
-, future
+, entrypoints
-, pyyaml
+, fetchPypi
-, nbformat
+, gcsfs
-, nbconvert
 , nbclient
-, six
+, nbformat
-, tqdm
+, pyarrow
-, jupyter-client
+, PyGithub
+, pytest-mock
+, pytestCheckHook
+, pythonOlder
+, pyyaml
 , requests
-, entrypoints
 , tenacity
-, futures ? null
+, tqdm
-, backports_tempfile
-, isPy27
-, pytestCheckHook
-, pytest-mock
 }:
 
 buildPythonPackage rec {
   pname = "papermill";
   version = "2.4.0";
+  format = "setuptools";
+
+  disabled = pythonOlder "3.7";
 
   src = fetchPypi {
     inherit pname version;
-    sha256 = "sha256-b4+KmwazlnfyB8CRAMjThrz1kvDLvdqfD1DoFEVpdic=";
+    hash = "sha256-b4+KmwazlnfyB8CRAMjThrz1kvDLvdqfD1DoFEVpdic=";
   };
 
   propagatedBuildInputs = [
     ansiwrap
     click
-    future
     pyyaml
     nbformat
-    nbconvert
     nbclient
-    six
     tqdm
-    jupyter-client
     requests
     entrypoints
     tenacity
-  ] ++ lib.optionals isPy27 [
-    futures
-    backports_tempfile
   ];
 
+  passthru.optional-dependencies = {
+    azure = [
+      azure-datalake-store
+      azure-storage-blob
+    ];
+    gcs = [
+      gcsfs
+    ];
+    github = [
+      PyGithub
+    ];
+    hdfs = [
+      pyarrow
+    ];
+    s3 = [
+      boto3
+    ];
+  };
+
   checkInputs = [
     pytestCheckHook
     pytest-mock
@@ -58,13 +73,17 @@ buildPythonPackage rec {
     export HOME=$(mktemp -d)
   '';
 
-  # the test suite depends on cloud resources azure/aws
+  # The test suite depends on cloud resources azure/aws
   doCheck = false;
 
+  pythonImportsCheck = [
+    "papermill"
+  ];
+
   meta = with lib; {
-    description = "Parametrize and run Jupyter and nteract Notebooks";
+    description = "Parametrize and run Jupyter and interact with notebooks";
     homepage = "https://github.com/nteract/papermill";
     license = licenses.bsd3;
-    maintainers = [ maintainers.costrouc ];
+    maintainers = with maintainers; [ costrouc ];
   };
 }