fix: Add association table relationship

lablup · Nov 13, 2024 · b2abda9 · b2abda9
1 parent c1c349c
commit b2abda9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/ai/backend/manager/models/container_registry.py b/src/ai/backend/manager/models/container_registry.py
@@ -12,7 +12,7 @@
 import yarl
 from graphql import Undefined, UndefinedType
 from sqlalchemy.ext.asyncio import AsyncSession
-from sqlalchemy.orm import load_only
+from sqlalchemy.orm import load_only, relationship
 from sqlalchemy.orm.exc import NoResultFound
 
 from ai.backend.common.exception import UnknownImageRegistry
@@ -82,6 +82,12 @@ class ContainerRegistryRow(Base):
     )
     extra = sa.Column("extra", sa.JSON, nullable=True, default=None)
 
+    association_container_registries_groups_rows = relationship(
+        "AssociationContainerRegistriesGroupsRow",
+        back_populates="container_registry_row",
+        primaryjoin="ContainerRegistryRow.id == foreign(AssociationContainerRegistriesGroupsRow.registry_id)",
+    )
+
     @classmethod
     async def get(
         cls,

diff --git a/src/ai/backend/manager/models/group.py b/src/ai/backend/manager/models/group.py
@@ -210,6 +210,11 @@ class GroupRow(Base):
         back_populates="group_row",
         primaryjoin="GroupRow.id == foreign(VFolderRow.group)",
     )
+    association_container_registries_groups_rows = relationship(
+        "AssociationContainerRegistriesGroupsRow",
+        back_populates="group_row",
+        primaryjoin="GroupRow.id == foreign(AssociationContainerRegistriesGroupsRow.group_id)",
+    )
 
 
 @dataclass