From dd3da7a6b0112834e100ba29d3b54026ab63188f Mon Sep 17 00:00:00 2001
From: Joseph Sawaya <joseph.sawaya@sentry.io>
Date: Tue, 11 Feb 2025 16:47:34 -0500
Subject: [PATCH 1/2] fix: update migrate-timeseries.sh

---
 migrate-timeseries.sh |  3 +--
 migrate_timeseries.py | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 33 insertions(+), 2 deletions(-)
 create mode 100644 migrate_timeseries.py

diff --git a/migrate-timeseries.sh b/migrate-timeseries.sh
index 3d47e86fa..1f8ebbfa8 100644
--- a/migrate-timeseries.sh
+++ b/migrate-timeseries.sh
@@ -6,5 +6,4 @@ if [ -f "/usr/local/bin/berglas" ]; then
   prefix="berglas exec --"
 fi
 
-$prefix python manage.py migrate --database timeseries rollouts
-$prefix python manage.py migrate --database timeseries pg_telemetry
+$prefix python migrate_timeseries.py
diff --git a/migrate_timeseries.py b/migrate_timeseries.py
new file mode 100644
index 000000000..d135cf0ac
--- /dev/null
+++ b/migrate_timeseries.py
@@ -0,0 +1,32 @@
+import os
+
+import django
+from django.core.management import call_command
+
+# Setup Django environment
+os.environ.setdefault("DJANGO_SETTINGS_MODULE", "django_scaffold.settings")
+django.setup()
+
+from django.conf import settings  # noqa: E402
+
+
+def run_migrate_commands():
+    try:
+        if settings.TA_TIMESERIES_ENABLED:
+            print("Running ta_timeseries migrations")
+            call_command(
+                "migrate",
+                database="ta_timeseries",
+                app_label="ta_timeseries",
+                settings="django_scaffold.settings",
+                verbosity=1,
+            )
+        else:
+            print("Skipping ta_timeseries migrations")
+
+    except Exception as e:
+        print(f"An error occurred: {e}")
+
+
+if __name__ == "__main__":
+    run_migrate_commands()

From 22d1b6536fd2ff711846a3c62ddd8b87bd714c2d Mon Sep 17 00:00:00 2001
From: Joseph Sawaya <joseph.sawaya@sentry.io>
Date: Fri, 21 Mar 2025 16:33:30 -0400
Subject: [PATCH 2/2] use logging instead of print

---
 migrate_timeseries.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/migrate_timeseries.py b/migrate_timeseries.py
index d135cf0ac..b40fbe33a 100644
--- a/migrate_timeseries.py
+++ b/migrate_timeseries.py
@@ -1,3 +1,4 @@
+import logging
 import os
 
 import django
@@ -9,11 +10,13 @@
 
 from django.conf import settings  # noqa: E402
 
+logger = logging.getLogger(__name__)
+
 
 def run_migrate_commands():
     try:
         if settings.TA_TIMESERIES_ENABLED:
-            print("Running ta_timeseries migrations")
+            logger.info("Running ta_timeseries migrations")
             call_command(
                 "migrate",
                 database="ta_timeseries",
@@ -22,10 +25,10 @@ def run_migrate_commands():
                 verbosity=1,
             )
         else:
-            print("Skipping ta_timeseries migrations")
+            logger.info("Skipping ta_timeseries migrations")
 
     except Exception as e:
-        print(f"An error occurred: {e}")
+        logger.error(f"An error occurred: {e}")
 
 
 if __name__ == "__main__":