from sqlalchemy.orm import Session
from ..models.Reagendamento import Reagendamento as ModelReagendamento
from ..schemas.Schemas import ReagendamentoBase
from fastapi import HTTPException

def get_reagendamentos(visita_comercial_id: int, session: Session):
    return session.query(ModelReagendamento).filter_by(visita_comercial_id=visita_comercial_id).all()

def add_reagendamento(reagendamento: ReagendamentoBase, session: Session):
    novo_reagendamento = ModelReagendamento(
        visita_comercial_id=reagendamento.visita_comercial_id,
        data_da_visita=reagendamento.data_da_visita
    )
    session.add(novo_reagendamento)
    session.commit()
    session.refresh(novo_reagendamento)
    return novo_reagendamento