from sqlalchemy import Identity
from sqlalchemy.orm import Mapped, mapped_column

from db.models.base import Base


class ChExperimentDBExperimentData(Base):
    __tablename__ = 'ch_experimentdb_experiment_data'

    id: Mapped[int] = mapped_column(Identity(start=11, cycle=True),
                                    primary_key=True)
    volume: Mapped[float]
    nitrogen_oxide_emission: Mapped[float]
    temperature: Mapped[float]
    co_fraction: Mapped[float]
    co2_fraction: Mapped[float]
    x: Mapped[float]
    y: Mapped[float]
    z: Mapped[float]
    file_id: Mapped[str]

    def __repr__(self):
        return f"<ChExperimentDBExperimentData>"