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

{ lib
, buildPythonPackage
, fetchPypi
, isPyPy

# build-system
, cython
, setuptools
, setuptools-scm
, packaging
, cffi

# dependencies

, py
, pytestCheckHook
, python
, pythonOlder
, tornado
, zeromq
, pytest-asyncio
}:

buildPythonPackage rec {
  pname = "pyzmq";
  version = "25.1.2";
  pyproject = true;

  disabled = pythonOlder "3.6";

  src = fetchPypi {
    inherit pname version;
    hash = "sha256-k/GqMR6LuRLjTwBM8YZAek6Q7sTw7MDv0mBWv37aAiY=";
  };

  nativeBuildInputs = [
    setuptools
    setuptools-scm
    packaging
  ] ++ (if isPyPy then [
    cffi
  ] else [
    cython
  ]);

  buildInputs = [
    zeromq
  ];

  propagatedBuildInputs = lib.optionals isPyPy [
    cffi
  ];

  nativeCheckInputs = [
    pytestCheckHook
    tornado
    pytest-asyncio
  ];

  pythonImportsCheck = [
    "zmq"
  ];

  pytestFlagsArray = [
    "$out/${python.sitePackages}/zmq/tests/" # Folder with tests
    # pytest.ini is missing in pypi's sdist
    # https://github.com/zeromq/pyzmq/issues/1853#issuecomment-1592731986
    "--asyncio-mode auto"
    "--ignore=$out/lib/python3.12/site-packages/zmq/tests/test_mypy.py"
  ];

  disabledTests = [
    # Tests hang
    "test_socket"
    "test_monitor"
    # https://github.com/zeromq/pyzmq/issues/1272
    "test_cython"
    # Test fails
    "test_mockable"
    # Issues with the sandbox
    "TestFutureSocket"
    "TestIOLoop"
    "TestPubLog"
  ];

  # Some of the tests use localhost networking.
  __darwinAllowLocalNetworking = true;

  meta = with lib; {
    description = "Python bindings for ØMQ";
    homepage = "https://pyzmq.readthedocs.io/";
    license = with licenses; [ bsd3 /* or */ lgpl3Only ];
    maintainers = with maintainers; [ ];
  };
}