crud.py

from sqlalchemy.orm import Session
from schemas import Motorbike
import models


def get_motorbikes(db: Session):
    return db.query(models.DBMotorbike).all()

def create_motorbike(db: Session, motorbike: Motorbike):
    db_motorbike = models.DBMotorbike(
        brand=motorbike.brand,
        model=motorbike.model,
        engine_capacity=motorbike.engine_capacity,
    )
    db.add(db_motorbike)
    db.commit()
    db.refresh(db_motorbike)
    return db_motorbike

def get_motorbike(
        db: Session,
        motorbike_id: int
):
    return db.query(models.DBMotorbike).filter(models.DBMotorbike.id == motorbike_id).first()

def update_motorbike(
        db: Session,
        db_motorbike: models.DBMotorbike,
        motorbike: Motorbike
):
    db_motorbike.brand = motorbike.brand
    db_motorbike.model = motorbike.model
    db_motorbike.engine_capacity = motorbike.engine_capacity
    db.commit()
    db.refresh(db_motorbike)
    return db_motorbike

def delete_motorbike(
        db: Session,
        db_motorbike: models.DBMotorbike
):
    db.delete(db_motorbike)
    db.commit()
    return db_motorbike