depot/third_party/nixpkgs/pkgs/development/python-modules/flake8-future-import/default.nix

{
  lib,
  buildPythonPackage,
  fetchFromGitHub,
  isPy27,
  isPy38,
  isPy39,
  pythonAtLeast,
  setuptools,
  flake8,
  six,
  python,
}:

buildPythonPackage rec {
  pname = "flake8-future-import";
  version = "0.4.7";
  pyproject = true;

  # PyPI tarball doesn't include the test suite
  src = fetchFromGitHub {
    owner = "xZise";
    repo = "flake8-future-import";
    rev = "refs/tags/${version}";
    hash = "sha256-2EcCOx3+PCk9LYpQjHCFNpQVI2Pdi+lWL8R6bNadFe0=";
  };

  patches =
    lib.optionals (pythonAtLeast "3.10") [ ./fix-annotations-version-11.patch ]
    ++ lib.optionals (isPy38 || isPy39) [ ./fix-annotations-version-10.patch ]
    ++ lib.optionals isPy27 [
      # Upstream disables this test case naturally on python 3, but it also fails
      # inside NixPkgs for python 2. Since it's going to be deleted, we just skip it
      # on py2 as well.
      ./skip-test.patch
    ];

  postPatch = ''
    substituteInPlace "test_flake8_future_import.py" \
      --replace-fail "'flake8'" "'${lib.getExe flake8}'"
  '';

  build-system = [ setuptools ];

  dependencies = [ flake8 ];

  nativeCheckInputs = [ six ];

  checkPhase = ''
    runHook preCheck

    ${python.interpreter} -m test_flake8_future_import

    runHook postCheck
  '';

  meta = with lib; {
    description = "Flake8 extension to check for the imported __future__ modules to make it easier to have a consistent code base";
    homepage = "https://github.com/xZise/flake8-future-import";
    license = licenses.mit;
  };
}