fix circular imports

Author: Deepthi-Chand

api/types/type_organization.py

@@ -1,4 +1,4 @@
-from typing import Any, List, Optional
+from typing import TYPE_CHECKING, Any, List, Optional
 
 import strawberry
 import strawberry_django
@@ -7,7 +7,9 @@
 
 from api.models import Organization
 from api.types.base_type import BaseType
-from authorization.types import TypeOrganizationMembership
+
+if TYPE_CHECKING:
+    from authorization.types import TypeOrganizationMembership
 
 
 @strawberry_django.filter(Organization)
@@ -122,10 +124,11 @@ def members_count(self, info: Info) -> int:
             return 0
 
     @strawberry.field(description="Members in this organization")
-    def members(self, info: Info) -> List[TypeOrganizationMembership]:
+    def members(self, info: Info) -> List["TypeOrganizationMembership"]:
         """Get members in this organization."""
         try:
             from authorization.models import OrganizationMembership
+            from authorization.types import TypeOrganizationMembership
 
             org_id = getattr(self, "id", None)
             if not org_id:

api/types/type_user.py

@@ -1,31 +1,14 @@
-from typing import List, Optional
+from typing import TYPE_CHECKING, List, Optional
 
 import strawberry
 import strawberry_django
 from strawberry import auto
 
 from api.types.base_type import BaseType
-from api.types.type_organization import TypeOrganization
 from authorization.models import OrganizationMembership, Role, User
 
-
-@strawberry_django.type(Role, fields="__all__")
-class TypeRole(BaseType):
-    """Type for user role."""
-
-    id: auto
-    name: auto
-    description: auto
-
-
-@strawberry_django.type(OrganizationMembership)
-class TypeOrganizationMembership(BaseType):
-    """Type for organization membership."""
-
-    organization: TypeOrganization
-    role: TypeRole
-    created_at: auto
-    updated_at: auto
+if TYPE_CHECKING:
+    from authorization.types import TypeOrganizationMembership
 
 
 @strawberry_django.filter(User)
@@ -68,9 +51,11 @@ class TypeUser(BaseType):
     """Type for user."""
 
     @strawberry.field
-    def organization_memberships(self) -> List[TypeOrganizationMembership]:
+    def organization_memberships(self) -> List["TypeOrganizationMembership"]:
         """Get organization memberships for this user."""
         try:
+            from authorization.types import TypeOrganizationMembership
+
             user_id = str(getattr(self, "id", ""))
             queryset = OrganizationMembership.objects.filter(user_id=user_id)
             return TypeOrganizationMembership.from_django_list(queryset)