linting

Author: wtripp180901

coral_credits/api/db_utils.py

@@ -72,12 +72,10 @@ def get_all_active_reservations(resource_provider_account):
 def get_credit_allocation(id):
     now = timezone.now()
 
-    credit_allocation = models.CreditAllocation.objects.filter(
-            id=id
-        ).first()
+    credit_allocation = models.CreditAllocation.objects.filter(id=id).first()
     if credit_allocation == None:
         raise db_exceptions.NoCreditAllocation("Invalid allocation_id")
-    
+
     return credit_allocation
 
 

coral_credits/api/serializers.py

@@ -78,7 +78,13 @@ class Consumer(serializers.ModelSerializer):
 
     class Meta:
         model = models.Consumer
-        fields = ["consumer_ref", "resource_provider_account", "start", "end", "resources"]
+        fields = [
+            "consumer_ref",
+            "resource_provider_account",
+            "start",
+            "end",
+            "resources",
+        ]
 
 
 class ResourceRequestSerializer(serializers.Serializer):

coral_credits/api/views.py

@@ -27,7 +27,9 @@ class CreditAllocationResourceViewSet(viewsets.ModelViewSet):
     permission_classes = [permissions.IsAuthenticated]
 
     def get_queryset(self):
-        return models.CreditAllocationResource.objects.filter(allocation__pk=self.kwargs["allocation_pk"])
+        return models.CreditAllocationResource.objects.filter(
+            allocation__pk=self.kwargs["allocation_pk"]
+        )
 
     def _create_update_credit_allocations(self, request, allocation_pk):
         """Allocate credits to a dictionary of resource classes.
@@ -65,15 +67,15 @@ def _create_update_credit_allocations(self, request, allocation_pk):
         # When creating with multiple resources a list of multiple entries is returned,
         # each with their own unique ID
         if len(serializer.data) == 1:
-          return Response(serializer.data[0], status=status.HTTP_201_CREATED)
+            return Response(serializer.data[0], status=status.HTTP_201_CREATED)
         else:
-          return Response(serializer.data, status=status.HTTP_201_CREATED)
+            return Response(serializer.data, status=status.HTTP_201_CREATED)
 
     def create(self, request, allocation_pk=None):
-        return self._create_update_credit_allocations(request,allocation_pk)
-    
+        return self._create_update_credit_allocations(request, allocation_pk)
+
     def update(self, request, allocation_pk=None, pk=None):
-        return self._create_update_credit_allocations(request,allocation_pk)
+        return self._create_update_credit_allocations(request, allocation_pk)
 
     def _validate_request(self, request):
 
@@ -129,19 +131,25 @@ def retrieve(self, request, pk=None):
         )
 
         # TODO(johngarbutt) look for any during the above allocations
-        consumers_query = models.Consumer.objects.filter(resource_provider_account__account__pk=pk)
+        consumers_query = models.Consumer.objects.filter(
+            resource_provider_account__account__pk=pk
+        )
         consumers = serializers.Consumer(
             consumers_query, many=True, context={"request": request}
         )
 
         account_summary["allocations"] = allocations.data
         account_summary["consumers"] = consumers.data
 
-        all_allocation_resources_query = models.CreditAllocationResource.objects.filter(allocation__account__pk=pk)
+        all_allocation_resources_query = models.CreditAllocationResource.objects.filter(
+            allocation__account__pk=pk
+        )
         # add resource_hours_remaining... must be a better way!
         # TODO(johngarbut) we don't check the dates line up!!
         for allocation in account_summary["allocations"]:
-            resources_for_allocation_query = all_allocation_resources_query.filter(allocation__id=allocation["id"])
+            resources_for_allocation_query = all_allocation_resources_query.filter(
+                allocation__id=allocation["id"]
+            )
             resources_for_allocation = serializers.CreditAllocationResourceSerializer(
                 resources_for_allocation_query, many=True, context={"request": request}
             )

coral_credits/urls.py

@@ -28,13 +28,25 @@
 #
 router = routers.DefaultRouter(trailing_slash=False)
 router.register(r"resource_class", views.ResourceClassViewSet)
-router.register(r"resource_class/", views.ResourceClassViewSet, basename="resourceclassslash")
+router.register(
+    r"resource_class/", views.ResourceClassViewSet, basename="resourceclassslash"
+)
 router.register(r"resource_provider", views.ResourceProviderViewSet)
-router.register(r"resource_provider/", views.ResourceProviderViewSet, basename="resourceproviderslash")
+router.register(
+    r"resource_provider/",
+    views.ResourceProviderViewSet,
+    basename="resourceproviderslash",
+)
 router.register(r"resource_provider_account", views.ResourceProviderAccountViewSet)
-router.register(r"resource_provider_account/", views.ResourceProviderAccountViewSet, basename="resourceprovideraccountslash")
+router.register(
+    r"resource_provider_account/",
+    views.ResourceProviderAccountViewSet,
+    basename="resourceprovideraccountslash",
+)
 router.register(r"allocation", views.CreditAllocationViewSet)
-router.register(r"allocation/", views.CreditAllocationViewSet, basename="allocationslash")
+router.register(
+    r"allocation/", views.CreditAllocationViewSet, basename="allocationslash"
+)
 router.register(r"account", views.AccountViewSet, basename="creditaccount")
 router.register(r"account/", views.AccountViewSet, basename="creditaccountslash")
 router.register(r"consumer", views.ConsumerViewSet, basename="resource-request")
@@ -43,8 +55,14 @@
 allocation_router = routers.NestedSimpleRouter(
     router, r"allocation", lookup="allocation", trailing_slash=False
 )
-allocation_router.register(r"resources", views.CreditAllocationResourceViewSet, basename="allocation-resource")
-allocation_router.register(r"resources/", views.CreditAllocationResourceViewSet, basename="allocation-resource-slash")
+allocation_router.register(
+    r"resources", views.CreditAllocationResourceViewSet, basename="allocation-resource"
+)
+allocation_router.register(
+    r"resources/",
+    views.CreditAllocationResourceViewSet,
+    basename="allocation-resource-slash",
+)
 
 
 def prometheus_metrics(request):