From f2f5a5b67a7e8eccacf1a8b2d77f03a9441ea26b Mon Sep 17 00:00:00 2001
From: "Vinicius D. Cerutti" <51954708+viniciusdc@users.noreply.github.com>
Date: Mon, 29 Jul 2024 17:24:55 -0300
Subject: [PATCH] Update upgrade command for 2024.7.1 (#2594)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
---
 src/_nebari/upgrade.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/src/_nebari/upgrade.py b/src/_nebari/upgrade.py
index 7e45dfa807..5fc18b1af5 100644
--- a/src/_nebari/upgrade.py
+++ b/src/_nebari/upgrade.py
@@ -1182,6 +1182,29 @@ def _version_specific_upgrade(
         return config
 
 
+class Upgrade_2024_7_1(UpgradeStep):
+    """
+    Upgrade step for Nebari version 2024.7.1
+
+    Note:
+        - Digital Ocean deprecation warning.
+    """
+
+    version = "2024.7.1"
+
+    @override
+    def _version_specific_upgrade(
+        self, config, start_version, config_filename: Path, *args, **kwargs
+    ):
+        if config.get("provider", "") == ProviderEnum.do.value:
+            rich.print("\n ⚠️  Deprecation Warning ⚠️")
+            rich.print(
+                "-> Digital Ocean support is currently being deprecated and will be removed in a future release.",
+            )
+            rich.print("")
+        return config
+
+
 __rounded_version__ = str(rounded_ver_parse(__version__))
 
 # Manually-added upgrade steps must go above this line