running db migrations on deploy

.github/workflows/copilot_deploy.yml

@@ -76,6 +76,7 @@ jobs:
       environment: dev
       app_name: fund-application-builder
       version: sha-${{ github.sha }}
+      db_name: fs-fund-builder
 
   post_dev_deploy_tests:
     needs: dev_deploy
@@ -103,6 +104,7 @@ jobs:
       environment: test
       app_name: fund-application-builder
       version: sha-${{ github.sha }}
+      db_name: fs-fund-builder
 
   post_test_deploy_tests:
     needs: test_deploy
@@ -131,6 +133,7 @@ jobs:
       environment: uat
       app_name: fund-application-builder
       version: sha-${{ github.sha }}
+      db_name: fs-fund-builder
 
   post_uat_deploy_tests:
     needs: uat_deploy
@@ -159,3 +162,4 @@ jobs:
       environment: prod
       app_name: fund-application-builder
       version: sha-${{ github.sha }}
+      db_name: fs-fund-builder

scripts/migration-task-script.py

@@ -0,0 +1,38 @@
+#!/usr/bin/env python3
+import re
+import subprocess
+import sys
+
+command_to_run = "flask db upgrade"
+environment = sys.argv[1]
+service = sys.argv[2]
+
+try:
+    command = subprocess.run(
+        args=[
+            "copilot",
+            "task",
+            "run",
+            "--generate-cmd",
+            f"pre-award/{environment}/{service}",
+        ],
+        capture_output=True,
+        check=True,
+        text=True,
+    )
+    # Strip image argument as we want to build a new image to pick up new migrations
+    command_with_image_removed = re.sub(r"--image \S+", "", command.stderr)
+except subprocess.CalledProcessError as e:
+    print(e.stderr)
+    raise e
+
+# Remove final line break and append arguments
+try:
+    subprocess.run(
+        args=command_with_image_removed[:-1] + f" \\\n--follow \\\n--command '{command_to_run}'",
+        shell=True,
+        check=True,
+    )
+except subprocess.CalledProcessError as e:
+    # Don't want to leak the command output here so just exit with the command's return code
+    sys.exit(e.returncode)