[COST-5012] add cost models to internal sources endpoint (#5084)

Author: maskarb

koku/masu/api/sources/serializers.py

@@ -48,7 +48,7 @@ class Meta:
 
 
 class ProviderBillingSourceSerializer(serializers.ModelSerializer):
-    """Serializer for the Provider Authentication model."""
+    """Serializer for the Provider Billing Source model."""
 
     data_source = serializers.JSONField(allow_null=False, required=True)
 

koku/masu/api/sources/views.py

@@ -9,6 +9,7 @@
 from django_filters import FilterSet
 from django_filters import UUIDFilter
 from django_filters.rest_framework import DjangoFilterBackend
+from django_tenants.utils import schema_context
 from rest_framework import mixins
 from rest_framework import viewsets
 from rest_framework.permissions import AllowAny
@@ -17,6 +18,7 @@
 
 from api.common.filters import CharListFilter
 from api.provider.models import Sources
+from cost_models.models import CostModelMap
 from masu.api.sources.serializers import SourceSerializer
 
 MIXIN_LIST = [mixins.ListModelMixin, mixins.RetrieveModelMixin, mixins.UpdateModelMixin, viewsets.GenericViewSet]
@@ -97,3 +99,24 @@ def get_object(self):
             raise Http404
 
         return obj
+
+    def list(self, request, *args, **kwargs):
+        """Obtain the list of sources."""
+        response = super().list(request=request, args=args, kwargs=kwargs)
+        for obj in response.data["data"]:
+            obj["cost_models"] = self.get_cost_models(obj)
+        return response
+
+    def retrieve(self, request, *args, **kwargs):
+        """Get a source."""
+        response = super().retrieve(request=request, args=args, kwargs=kwargs)
+        response.data["cost_models"] = self.get_cost_models(response.data)
+        return response
+
+    def get_cost_models(self, obj):
+        """Get the cost models associated with this provider."""
+        if not (schema := obj.get("provider", {}).get("customer", {}).get("schema_name")):
+            return []
+        with schema_context(schema):
+            cost_models_map = CostModelMap.objects.filter(provider_uuid=obj["source_uuid"])
+            return [{"name": m.cost_model.name, "uuid": m.cost_model.uuid} for m in cost_models_map]