From 1f72fcddb10185a4187c3137c8dedb30047dafa7 Mon Sep 17 00:00:00 2001
From: Eduardo Blancas <github@blancas.io>
Date: Tue, 5 Dec 2023 18:26:27 -0600
Subject: [PATCH] show ploomber cloud sign up link every 2 days

---
 src/ploomber_core/telemetry/telemetry.py | 30 ++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/src/ploomber_core/telemetry/telemetry.py b/src/ploomber_core/telemetry/telemetry.py
index d3fde65..dba793e 100644
--- a/src/ploomber_core/telemetry/telemetry.py
+++ b/src/ploomber_core/telemetry/telemetry.py
@@ -103,6 +103,7 @@ class Internal(Config):
     """
 
     last_version_check: datetime.datetime = None
+    last_cloud_check: datetime.datetime = None
     uid: str
     first_time: bool = True
 
@@ -307,6 +308,34 @@ def check_version(package_name, version):
     internal.last_version_check = now
 
 
+def check_cloud():
+    """Displays a message to ask the user to sign up for Ploomber Cloud"""
+    settings = UserSettings()
+
+    if not settings.version_check_enabled:
+        return
+
+    # this feature is not documented. we added it to prevent the doctests
+    # from failing
+    if "PLOOMBER_VERSION_CHECK_DISABLED" in os.environ:
+        return
+
+    now = datetime.datetime.now()
+
+    # Check if we already notified in the last 2 days
+    if internal.last_cloud_check and (now - internal.last_cloud_check).days < 2:
+        return
+
+    click.secho(
+        "Deploy AI and data apps for free on Ploomber Cloud! "
+        "Sign up here: https://www.platform.ploomber.io/register",
+        fg="green",
+    )
+
+    # Update latest check date
+    internal.last_cloud_check = now
+
+
 def _get_telemetry_info(package_name, version):
     """
     The function checks for the local config and uid files, returns the right
@@ -318,6 +347,7 @@ def _get_telemetry_info(package_name, version):
 
     # Check latest version
     check_version(package_name, version)
+    check_cloud()
 
     if telemetry_enabled:
         # Check first time install