23 lines
779 B
Python
23 lines
779 B
Python
from sqlalchemy.orm import Session
|
|
from . import models, dto
|
|
|
|
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 |