Merge branch 'main' into unleash-disable-resource-matching

project-koku · Dec 2, 2024 · 5d64f4b · 5d64f4b
2 parents fc6c87a + 5f35779
commit 5d64f4b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/koku/cost_models/cost_model_manager.py b/koku/cost_models/cost_model_manager.py
@@ -17,7 +17,6 @@
 from cost_models.models import CostModelMap
 from masu.processor.tasks import update_cost_model_costs
 
-
 LOG = logging.getLogger(__name__)
 
 
@@ -34,8 +33,11 @@ def __init__(self, cost_model_uuid=None):
         self._cost_model_uuid = None
 
         if cost_model_uuid:
-            self._model = CostModel.objects.get(uuid=cost_model_uuid)
-            self._cost_model_uuid = cost_model_uuid
+            try:
+                self._model = CostModel.objects.get(uuid=cost_model_uuid)
+                self._cost_model_uuid = cost_model_uuid
+            except CostModel.DoesNotExist:
+                LOG.warning(f"CostModel with UUID {cost_model_uuid} does not exist.")
 
     @property
     def instance(self):