Put back license endpoint billing manager

PostHog · Oct 11, 2024 · bbf0813 · bbf0813
1 parent 4278277
commit bbf0813
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/ee/api/billing.py b/ee/api/billing.py
@@ -319,10 +319,9 @@ def license(self, request: Request, *args: Any, **kwargs: Any) -> HttpResponse:
             return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
 
         license = License(key=serializer.validated_data["license"])
-        billing_manager = self.get_billing_manager()
         res = requests.get(
             f"{BILLING_SERVICE_URL}/api/billing",
-            headers=billing_manager.get_auth_headers(organization),
+            headers=BillingManager(license).get_auth_headers(organization),
         )
 
         if res.status_code != 200:
@@ -332,7 +331,7 @@ def license(self, request: Request, *args: Any, **kwargs: Any) -> HttpResponse:
                 }
             )
         data = res.json()
-        billing_manager.update_license_details(data)
+        BillingManager(license).update_license_details(data)
         return Response({"success": True})
 
     def _get_org(self) -> Optional[Organization]: