23 lines
807 B
Python
23 lines
807 B
Python
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)
|
|
unit_name = Column(String(64), nullable=True)
|
|
unit_symbol = Column(String(16), nullable=True)
|
|
|
|
scenario = relationship("Scenario", back_populates="consumption_items")
|
|
|
|
def __repr__(self):
|
|
return (
|
|
f"<Consumption id={self.id} scenario_id={self.scenario_id} "
|
|
f"amount={self.amount} unit={self.unit_symbol or self.unit_name}>"
|
|
)
|