fix: Fixed migration issue with get_feature_status causing migration …

…to fail as FeatureFlags wasn't yet added. (#238)

infrastructure/aws/handler.py

@@ -1,4 +1,5 @@
 import logging
+import sys
 
 import boto3
 from botocore.config import Config
@@ -16,17 +17,18 @@
 
 AWS_SCHEDULES_ENABLED = get_var("AWS_SCHEDULES_ACCESS_KEY_ID") and get_var("AWS_SCHEDULES_SECRET_ACCESS_KEY")
 
-if not AWS_SCHEDULES_ENABLED and get_feature_status("isInvoiceSchedulingEnabled", should_use_cache=False):
-    raise ValueError(
-        "If using schedules, the variables MUST be set. If you are not going to use schedules, "
-        "set the isInvoiceSchedulingEnabled feature flag to False"
-    )
+if sys.argv[1] != "migrate":
+    if not AWS_SCHEDULES_ENABLED and get_feature_status("isInvoiceSchedulingEnabled", should_use_cache=False):
+        raise ValueError(
+            "If using schedules, the variables MUST be set. If you are not going to use schedules, "
+            "set the isInvoiceSchedulingEnabled feature flag to False"
+        )
 
-if not AWS_SCHEDULES_ENABLED:
-    try:
-        FeatureFlags.objects.get(name="isInvoiceSchedulingEnabled").value = False
-    except FeatureFlags.DoesNotExist:
-        ...
+    if not AWS_SCHEDULES_ENABLED:
+        try:
+            FeatureFlags.objects.get(name="isInvoiceSchedulingEnabled").value = False
+        except FeatureFlags.DoesNotExist:
+            ...
 
 if AWS_SCHEDULES_ENABLED:
     Boto3HandlerSession = boto3.session.Session(