from sqlalchemy.orm import Session

from . import dto, models


class TrailerService:
    @staticmethod
    def get_trailer_by_number(db: Session, trailer_number: str):
        return (
            db.query(models.Trailer)
            .filter(models.Trailer.trailer_number == trailer_number)
            .first()
        )

    @staticmethod
    def create_trailer(db: Session, trailer_data: dto.TrailerCreateDTO):
        new_trailer = models.Trailer(**trailer_data.dict())
        db.add(new_trailer)
        db.commit()
        db.refresh(new_trailer)
        return new_trailer

    @staticmethod
    def delete_trailer(db: Session, trailer_number: str):
        trailer = TrailerService.get_trailer_by_number(db, trailer_number)
        if trailer:
            db.delete(trailer)
            db.commit()
        return trailer