from fastapi import Depends, FastAPI, HTTPException, Query

from sqlmodel import Field, Session, SQLModel, create_engine, select

class Hero(SQLModel, table=True):

id: int | None = Field(default=None, primary_key=True)

name: str = Field(index=True)

age: int | None = Field(default=None, index=True)

secret_name: str

sqlite_file_name = "database.db"

sqlite_url = f"sqlite:///{sqlite_file_name}"

connect_args = {"check_same_thread": False}

engine = create_engine(sqlite_url, connect_args=connect_args)

def create_db_and_tables():

SQLModel.metadata.create_all(engine)

def get_session():

with Session(engine) as session:

yield session

app = FastAPI()

@app.on_event("startup")

def on_startup():

create_db_and_tables()

@app.post("/heroes/")

def create_hero(hero: Hero, session: Session = Depends(get_session)) -> Hero:

session.add(hero)

session.commit()

session.refresh(hero)

return hero

@app.get("/heroes/")

def read_heroes(

session: Session = Depends(get_session),

offset: int = 0,

limit: int = Query(default=100, le=100),

) -> list[Hero]:

heroes = session.exec(select(Hero).offset(offset).limit(limit)).all()

return heroes

@app.get("/heroes/{hero_id}")

def read_hero(hero_id: int, session: Session = Depends(get_session)) -> Hero:

hero = session.get(Hero, hero_id)

if not hero:

raise HTTPException(status_code=404, detail="Hero not found")

return hero

@app.delete("/heroes/{hero_id}")

def delete_hero(hero_id: int, session: Session = Depends(get_session)):

hero = session.get(Hero, hero_id)

if not hero:

raise HTTPException(status_code=404, detail="Hero not found")

session.delete(hero)

session.commit()

return {"ok": True}