main.py

import uvicorn
from fastapi import FastAPI, HTTPException, Depends
from schemas import Motorbike, MotorbikeResponse
import models
from database import engine, SessionLocal
models.Base.metadata.create_all(bind=engine)
from sqlalchemy.orm import Session
import crud
from typing import List


def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()
app = FastAPI()


@app.get("/")
async def root():
    return {"message": "Hello World"}


@app.get("/status")
async def status():
    return {"status": "operational", "version": "1.0.0"}


@app.post("/motorbikes")
async def create_motorbike(motorbike: Motorbike, db: Session = Depends(get_db)):
    return crud.create_motorbike(db, motorbike)


@app.get("/motorbikes", response_model=List[MotorbikeResponse])
async def get_motorbikes(db: Session = Depends(get_db)):
    return crud.get_motorbikes(db)


@app.get("/motorbikes/{motorbike_id}", response_model=MotorbikeResponse)
async def get_motorbike(motorbike_id: int, db: Session = Depends(get_db)):
    if not crud.get_motorbike(db, motorbike_id):
        raise HTTPException(status_code=404, detail="Motorbike not found")
    return crud.get_motorbike(db, motorbike_id)


@app.put("/motorbikes/{motorbike_id}")
async def put_motorbike(
        motorbike: Motorbike,
        motorbike_id: int,
        db: Session = Depends(get_db)
):
    db_motorbike = crud.get_motorbike(db, motorbike_id)
    if not db_motorbike:
        raise HTTPException(status_code=404, detail="Motorbike not found")
    return crud.update_motorbike(db, db_motorbike, motorbike)


@app.delete("/motorbikes/{motorbike_id}")
async def delete_motorbike(motorbike_id: int, db: Session = Depends(get_db)):
    db_motorbike = crud.get_motorbike(db, motorbike_id)
    if not db_motorbike:
        raise HTTPException(status_code=404, detail="Motorbike not found")
    crud.delete_motorbike(db, db_motorbike)
    return {"message": "Motorbike deleted"}

if __name__ == "__main__":
    import uvicorn
    uvicorn.run("main:app", host="localhost", port=8000)