Merge pull request #251057 from fabaff/kombu-pyro4

python311Packages.kombu: add optional-dependencies

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

@@ -2,17 +2,27 @@
 , amqp
 , azure-identity
 , azure-servicebus
+, azure-storage-queue
 , backports-zoneinfo
+, boto3
 , buildPythonPackage
 , case
+, confluent-kafka
 , fetchPypi
 , hypothesis
-, pyro4
+, kazoo
+, msgpack
+, pycurl
+, pymongo
+  #, pyro4
 , pytestCheckHook
 , pythonOlder
-, pytz
+, pyyaml
-, vine
+, redis
+, sqlalchemy
 , typing-extensions
+, urllib3
+, vine
 }:
 
 buildPythonPackage rec {
@@ -36,24 +46,65 @@ buildPythonPackage rec {
     backports-zoneinfo
   ];
 
+  passthru.optional-dependencies = {
+    msgpack = [
+      msgpack
+    ];
+    yaml = [
+      pyyaml
+    ];
+    redis = [
+      redis
+    ];
+    mongodb = [
+      pymongo
+    ];
+    sqs = [
+      boto3
+      urllib3
+      pycurl
+    ];
+    zookeeper = [
+      kazoo
+    ];
+    sqlalchemy = [
+      sqlalchemy
+    ];
+    azurestoragequeues = [
+      azure-identity
+      azure-storage-queue
+    ];
+    azureservicebus = [
+      azure-servicebus
+    ];
+    confluentkafka = [
+      confluent-kafka
+    ];
+    # pyro4 doesn't suppport Python 3.11
+    #pyro = [
+    #  pyro4
+    #];
+  };
+
   nativeCheckInputs = [
-    azure-identity
-    azure-servicebus
     case
     hypothesis
-    pyro4
     pytestCheckHook
-    pytz
+  ] ++ lib.flatten (builtins.attrValues passthru.optional-dependencies);
-  ];
 
   pythonImportsCheck = [
     "kombu"
   ];
 
+  disabledTests = [
+    # Disable pyro4 test
+    "test_driver_version"
+  ];
+
   meta = with lib; {
-    changelog = "https://github.com/celery/kombu/releases/tag/v${version}";
     description = "Messaging library for Python";
     homepage = "https://github.com/celery/kombu";
+    changelog = "https://github.com/celery/kombu/releases/tag/v${version}";
     license = licenses.bsd3;
     maintainers = with maintainers; [ fab ];
   };