add party subtype relationships

casact · Jul 2, 2020 · 9a8de03 · 9a8de03
1 parent 9c53c80
commit 9a8de03
Showing 1 changed file with 42 additions and 0 deletions.
diff --git a/pcdm/party.py b/pcdm/party.py
@@ -184,6 +184,30 @@ class Organization(Base):
         back_populates='organization'
     )
 
+    organization_unit = relationship(
+        'OrganizationUnit',
+        primaryjoin='Organization.organization_id == OrganizationUnit.organization_id',
+        back_populates='organization'
+    )
+
+    for_profit_organization = relationship(
+        'ForProfitOrganization',
+        primaryjoin='Organization.organization_id == ForProfitOrganization.organization_id',
+        back_populates='organization'
+    )
+
+    government_organization = relationship(
+        'GovernmentOrganization',
+        primaryjoin='Organization.organization_id == GovernmentOrganization.organization_id',
+        back_populates='organization'
+    )
+
+    not_for_profit_organization = relationship(
+        'NotForProfitOrganization',
+        primaryjoin='Organization.organization_id == NotForProfitOrganization.organization_id',
+        back_populates='organization'
+    )
+
     def __repr__(self):
         return "<Organization(" \
                "party_id='%s', " \
@@ -453,6 +477,24 @@ class Grouping(Base):
         back_populates='grouping'
     )
 
+    professional_group = relationship(
+        'ProfessionalGroup',
+        primaryjoin='Grouping.grouping_id == ProfessionalGroup.grouping_id',
+        back_populates='grouping'
+    )
+
+    project = relationship(
+        'Project',
+        primaryjoin='Grouping.grouping_id == Project.grouping_id',
+        back_populates='grouping'
+    )
+
+    team = relationship(
+        'Team',
+        primaryjoin='Grouping.grouping_id == Team.grouping_id',
+        back_populates='grouping'
+    )
+
     def __repr__(self):
         return "<Grouping(" \
                "party_id='%s', " \