src/servala/core/models/__init__.py

@@ -6,18 +6,18 @@ from .organization import (
     OrganizationRole,
 )
 from .service import (
+    CloudProvider,
     ControlPlane,
     Plan,
     Service,
     ServiceCategory,
     ServiceOffering,
-    ServiceOfferingPlan,
-    ServiceProvider,
 )
 from .user import User
 
 __all__ = [
     "BillingEntity",
+    "CloudProvider",
     "ControlPlane",
     "Organization",
     "OrganizationMembership",
@@ -27,7 +27,5 @@ __all__ = [
     "Service",
     "ServiceCategory",
     "ServiceOffering",
-    "ServiceOfferingPlan",
-    "ServiceProvider",
     "User",
 ]

src/servala/core/models/organization.py

@@ -1,4 +1,3 @@
-from django.contrib.auth import get_user_model
 from django.db import models
 from django.utils.translation import gettext_lazy as _
 
@@ -20,7 +19,7 @@ class Organization(models.Model):
     )
 
     members = models.ManyToManyField(
-        to=get_user_model(),
+        to="core.User",
         through="OrganizationMembership",
         related_name="organizations",
         verbose_name=_("Members"),
@@ -70,7 +69,7 @@ class OrganizationMembership(models.Model):
     """Through-model for the many-to-many relationship between organizations and users."""
 
     user = models.ForeignKey(
-        to=get_user_model(),
+        to="core.User",
         on_delete=models.CASCADE,
         related_name="organization_memberships",
         verbose_name=_("User"),