Add models and routes for costs, consumption, equipment, maintenance, and production; implement CRUD operations and unit tests

models/capex.py

@@ -0,0 +1,17 @@
+from sqlalchemy import Column, Integer, Float, String, ForeignKey
+from sqlalchemy.orm import relationship
+from config.database import Base
+
+
+class Capex(Base):
+    __tablename__ = "capex"
+
+    id = Column(Integer, primary_key=True, index=True)
+    scenario_id = Column(Integer, ForeignKey("scenario.id"), nullable=False)
+    amount = Column(Float, nullable=False)
+    description = Column(String, nullable=True)
+
+    scenario = relationship("Scenario", back_populates="capex_items")
+
+    def __repr__(self):
+        return f"<Capex id={self.id} scenario_id={self.scenario_id} amount={self.amount}>"

models/consumption.py

@@ -0,0 +1,17 @@
+from sqlalchemy import Column, Integer, Float, String, ForeignKey
+from sqlalchemy.orm import relationship
+from config.database import Base
+
+
+class Consumption(Base):
+    __tablename__ = "consumption"
+
+    id = Column(Integer, primary_key=True, index=True)
+    scenario_id = Column(Integer, ForeignKey("scenario.id"), nullable=False)
+    amount = Column(Float, nullable=False)
+    description = Column(String, nullable=True)
+
+    scenario = relationship("Scenario", back_populates="consumption_items")
+
+    def __repr__(self):
+        return f"<Consumption id={self.id} scenario_id={self.scenario_id} amount={self.amount}>"

models/equipment.py

@@ -0,0 +1,17 @@
+from sqlalchemy import Column, Integer, String, ForeignKey
+from sqlalchemy.orm import relationship
+from config.database import Base
+
+
+class Equipment(Base):
+    __tablename__ = "equipment"
+
+    id = Column(Integer, primary_key=True, index=True)
+    scenario_id = Column(Integer, ForeignKey("scenario.id"), nullable=False)
+    name = Column(String, nullable=False)
+    description = Column(String, nullable=True)
+
+    scenario = relationship("Scenario", back_populates="equipment_items")
+
+    def __repr__(self):
+        return f"<Equipment id={self.id} scenario_id={self.scenario_id} name={self.name}>"

models/maintenance.py

@@ -0,0 +1,17 @@
+from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, func
+from sqlalchemy.orm import relationship
+from config.database import Base
+
+
+class Maintenance(Base):
+    __tablename__ = "maintenance"
+
+    id = Column(Integer, primary_key=True, index=True)
+    scenario_id = Column(Integer, ForeignKey("scenario.id"), nullable=False)
+    performed_at = Column(DateTime(timezone=True), server_default=func.now())
+    details = Column(String, nullable=True)
+
+    scenario = relationship("Scenario", back_populates="maintenance_items")
+
+    def __repr__(self):
+        return f"<Maintenance id={self.id} scenario_id={self.scenario_id} performed_at={self.performed_at}>"

models/opex.py

@@ -0,0 +1,17 @@
+from sqlalchemy import Column, Integer, Float, String, ForeignKey
+from sqlalchemy.orm import relationship
+from config.database import Base
+
+
+class Opex(Base):
+    __tablename__ = "opex"
+
+    id = Column(Integer, primary_key=True, index=True)
+    scenario_id = Column(Integer, ForeignKey("scenario.id"), nullable=False)
+    amount = Column(Float, nullable=False)
+    description = Column(String, nullable=True)
+
+    scenario = relationship("Scenario", back_populates="opex_items")
+
+    def __repr__(self):
+        return f"<Opex id={self.id} scenario_id={self.scenario_id} amount={self.amount}>"

models/production_output.py

@@ -0,0 +1,18 @@
+from sqlalchemy import Column, Integer, Float, String, ForeignKey
+from sqlalchemy.orm import relationship
+from config.database import Base
+
+
+class ProductionOutput(Base):
+    __tablename__ = "production_output"
+
+    id = Column(Integer, primary_key=True, index=True)
+    scenario_id = Column(Integer, ForeignKey("scenario.id"), nullable=False)
+    amount = Column(Float, nullable=False)
+    description = Column(String, nullable=True)
+
+    scenario = relationship(
+        "Scenario", back_populates="production_output_items")
+
+    def __repr__(self):
+        return f"<ProductionOutput id={self.id} scenario_id={self.scenario_id} amount={self.amount}>"

models/scenario.py

@@ -1,5 +1,12 @@
 from sqlalchemy import Column, Integer, String, DateTime, func
 from sqlalchemy.orm import relationship
+from models.simulation_result import SimulationResult
+from models.capex import Capex
+from models.opex import Opex
+from models.consumption import Consumption
+from models.production_output import ProductionOutput
+from models.equipment import Equipment
+from models.maintenance import Maintenance
 from config.database import Base
 
 
@@ -13,7 +20,19 @@ class Scenario(Base):
     updated_at = Column(DateTime(timezone=True), onupdate=func.now())
     parameters = relationship("Parameter", back_populates="scenario")
     simulation_results = relationship(
-        "SimulationResult", back_populates="scenario")
+        SimulationResult, back_populates="scenario")
+    capex_items = relationship(
+        Capex, back_populates="scenario")
+    opex_items = relationship(
+        Opex, back_populates="scenario")
+    consumption_items = relationship(
+        Consumption, back_populates="scenario")
+    production_output_items = relationship(
+        ProductionOutput, back_populates="scenario")
+    equipment_items = relationship(
+        Equipment, back_populates="scenario")
+    maintenance_items = relationship(
+        Maintenance, back_populates="scenario")
 
     # relationships can be defined later
     def __repr__(self):