From faa5ecad9da89e42fe7f8376e29326a5892c7953 Mon Sep 17 00:00:00 2001
From: Fabian Affolter <mail@fabian-affolter.ch>
Date: Mon, 27 Feb 2023 10:25:33 +0100
Subject: [PATCH] dbx: 0.7.6 -> 0.8.8

Diff: https://github.com/databrickslabs/dbx/compare/refs/tags/v0.7.6...v0.8.8

Changelog: https://github.com/databrickslabs/dbx/blob/v0.8.8/CHANGELOG.md
---
 pkgs/applications/misc/dbx/default.nix | 29 +++++++++++++++++++-------
 1 file changed, 21 insertions(+), 8 deletions(-)

diff --git a/pkgs/applications/misc/dbx/default.nix b/pkgs/applications/misc/dbx/default.nix
index 76081e90c9a..8e627af0287 100644
--- a/pkgs/applications/misc/dbx/default.nix
+++ b/pkgs/applications/misc/dbx/default.nix
@@ -6,16 +6,22 @@
 
 python3.pkgs.buildPythonApplication rec {
   pname = "dbx";
-  version = "0.7.6";
+  version = "0.8.8";
   format = "setuptools";
 
   src = fetchFromGitHub {
     owner = "databrickslabs";
     repo = "dbx";
     rev = "refs/tags/v${version}";
-    hash = "sha256-P/cniy0xYaDoUbKdvV7KCubCpmOAhYp3cg2VBRA+a6I=";
+    hash = "sha256-nx6fz+atlnB/KxdznnZArHpyv41cuBDQauG0irq1Zyc=";
   };
 
+  postPatch = ''
+    substituteInPlace setup.py \
+      --replace "mlflow-skinny>=1.28.0,<3.0.0" "mlflow" \
+      --replace "rich==12.6.0" "rich"
+  '';
+
   propagatedBuildInputs = with python3.pkgs; [
     aiohttp
     click
@@ -34,6 +40,19 @@ python3.pkgs.buildPythonApplication rec {
     watchdog
   ] ++ typer.optional-dependencies.all;
 
+  passthru.optional-dependencies = with python3.pkgs; {
+    aws = [
+      boto3
+    ];
+    azure = [
+      azure-storage-blob
+      azure-identity
+    ];
+    gcp = [
+      google-cloud-storage
+    ];
+  };
+
   nativeCheckInputs = [
     git
   ] ++ (with python3.pkgs; [
@@ -43,12 +62,6 @@ python3.pkgs.buildPythonApplication rec {
     pytestCheckHook
   ]);
 
-  postPatch = ''
-    substituteInPlace setup.py \
-      --replace "mlflow-skinny>=1.28.0,<=2.0.0" "mlflow" \
-      --replace "rich==12.5.1" "rich"
-  '';
-
   preCheck = ''
     export HOME=$(mktemp -d)
     export PATH="$PATH:$out/bin"