leave all SBA old stuff in tables

nsidc · Oct 14, 2024 · dcdc2dd · dcdc2dd
1 parent bb4fd66
commit dcdc2dd
Showing 1 changed file with 44 additions and 0 deletions.
diff --git a/usaon_benefit_tool/models/tables.py b/usaon_benefit_tool/models/tables.py
@@ -473,3 +473,47 @@ class SocietalBenefitArea(BaseModel):
         String(256),
         primary_key=True,
     )
+    societal_benefit_sub_areas = relationship(
+        'SocietalBenefitSubArea',
+        back_populates='societal_benefit_area',
+    )
+
+
+class SocietalBenefitSubArea(BaseModel):
+    __tablename__ = 'societal_benefit_subarea'
+    id = Column(
+        String(256),
+        primary_key=True,
+    )
+    societal_benefit_area_id = Column(
+        String(256),
+        ForeignKey('societal_benefit_area.id'),
+        nullable=False,
+    )
+
+    societal_benefit_area = relationship(
+        'SocietalBenefitArea',
+        back_populates='societal_benefit_sub_areas',
+    )
+    societal_benefit_key_objectives = relationship(
+        'SocietalBenefitKeyObjective',
+        back_populates='societal_benefit_sub_area',
+    )
+
+
+class SocietalBenefitKeyObjective(BaseModel):
+    __tablename__ = 'societal_benefit_key_objective'
+    id = Column(
+        String(256),
+        primary_key=True,
+    )
+    societal_benefit_subarea_id = Column(
+        String(256),
+        ForeignKey('societal_benefit_subarea.id'),
+        primary_key=True,
+    )
+
+    societal_benefit_sub_area = relationship(
+        'SocietalBenefitSubArea',
+        back_populates='societal_benefit_key_objectives',
+    )