build: include migrating timeseries

django_scaffold/settings.py

@@ -36,6 +36,7 @@
     "shared.django_apps.reports",
     "shared.django_apps.staticanalysis",
     "shared.django_apps.ta_timeseries",
+    "shared.django_apps.timeseries",
 ]
 
 TELEMETRY_VANILLA_DB = "default"

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

migrate_timeseries.py

@@ -0,0 +1,49 @@
+import logging
+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
+
+logger = logging.getLogger(__name__)
+
+
+def run_migrate_commands():
+    try:
+        if settings.TIMESERIES_ENABLED:
+            logger.info("Running timeseries migrations")
+
+            call_command(
+                "migrate",
+                database="timeseries",
+                app_label="timeseries",
+                settings="django_scaffold.settings",
+                verbosity=1,
+            )
+        else:
+            logger.info("Skipping timeseries migrations")
+
+        if settings.TA_TIMESERIES_ENABLED:
+            logger.info("Running ta_timeseries migrations")
+
+            call_command(
+                "migrate",
+                database="ta_timeseries",
+                app_label="ta_timeseries",
+                settings="django_scaffold.settings",
+                verbosity=1,
+            )
+        else:
+            logger.info("Skipping ta_timeseries migrations")
+
+    except Exception as e:
+        logger.error(f"An error occurred: {e}")
+
+
+if __name__ == "__main__":
+    run_migrate_commands()