Source code for app_api.models.models

"""ORM models for the application.

Defines the Operation table to store mathematical operations with results.
"""

from sqlalchemy import Column, Float, Integer, String

from app_api.modules.connect import Base  # importer Base depuis connect




class Operation(Base):
    """SQLAlchemy ORM model for a mathematical operation.

    Attributes:
        id (int): Primary key.
        operation (str): Operation type ("add", "sub", "square").
        a (float): First operand.
        b (float | None): Second operand (optional for square).
        result (float | None): Computed result of the operation.

    """

    __tablename__ = "operations"
    __table_args__ = {"extend_existing": True}  # Evite l'erreur "table already defined"

    id = Column(Integer, primary_key=True, index=True)
    operation = Column(String, nullable=False)
    a = Column(Float, nullable=False)
    b = Column(Float, nullable=True)
    result = Column(Float, nullable=True)