nixpkgs/pkgs/development/python-modules/google_cloud_spanner/default.nix

{ stdenv
, buildPythonPackage
, fetchPypi
, grpc_google_iam_v1
, google_cloud_core
, google_cloud_testutils
, libcst
, mock
, proto-plus
, pytestCheckHook
, pytest-asyncio
, sqlparse
}:

buildPythonPackage rec {
  pname = "google-cloud-spanner";
  version = "2.1.0";

  src = fetchPypi {
    inherit pname version;
    sha256 = "0mkkx6l3cbwfwng12zpisbv6m919fkhdb48xk24ayc19193bi86n";
  };

  postPatch = ''
    substituteInPlace setup.py \
      --replace '"proto-plus == 1.11.0"' '"proto-plus"'
  '';

  propagatedBuildInputs = [ google_cloud_core grpc_google_iam_v1 libcst proto-plus sqlparse ];

  checkInputs = [ google_cloud_testutils mock pytestCheckHook pytest-asyncio ];

  preCheck = ''
    # prevent google directory from shadowing google imports
    rm -r google
    # disable tests which require credentials
    rm tests/system/test_{system,system_dbapi}.py
    rm tests/unit/spanner_dbapi/test_{connect,connection,cursor}.py
  '';

  pythonImportsCheck = [
    "google.cloud.spanner_admin_database_v1"
    "google.cloud.spanner_admin_instance_v1"
    "google.cloud.spanner_dbapi"
    "google.cloud.spanner_v1"
  ];

  meta = with stdenv.lib; {
    description = "Cloud Spanner API client library";
    homepage = "https://github.com/googleapis/python-spanner";
    license = licenses.asl20;
    maintainers = with maintainers; [ SuperSandro2000 ];
  };
}