from sqlalchemy.orm import Session
from app.models.calculation import Calculation
from app.schemas.calculation import CalculationCreate


def create_calculation(
    db: Session, calc: CalculationCreate, result: float
) -> Calculation:
    db_calculation = Calculation(
        operation=calc.operation,
        first_number=calc.first_number,
        second_number=calc.second_number,
        result=result,
    )
    db.add(db_calculation)
    db.commit()
    db.refresh(db_calculation)
    return db_calculation


def get_calculations(db: Session, skip: int = 0, limit: int = 100):
    return (
        db.query(Calculation)
        .order_by(Calculation.created_at.desc())
        .offset(skip)
        .limit(limit)
        .all()
    )


def get_calculation(db: Session, calculation_id: int):
    return db.query(Calculation).filter(Calculation.id == calculation_id).first()