diff --git a/django_scaffold/settings.py b/django_scaffold/settings.py index 94346a286..892d88c30 100644 --- a/django_scaffold/settings.py +++ b/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" 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..c12cd0213 --- /dev/null +++ b/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()