85 lines
2.2 KiB
Python
85 lines
2.2 KiB
Python
from fastapi import APIRouter, Depends, HTTPException
|
|
from sqlmodel import SQLModel, Session
|
|
|
|
from learn_sql_model.api.websocket_connection_manager import manager
|
|
from learn_sql_model.config import get_config, get_session
|
|
from learn_sql_model.models.hero import Hero, HeroCreate, HeroRead, HeroUpdate
|
|
|
|
hero_router = APIRouter()
|
|
|
|
|
|
@hero_router.on_event("startup")
|
|
def on_startup() -> None:
|
|
SQLModel.metadata.create_all(get_config().database.engine)
|
|
|
|
|
|
@hero_router.get("/hero/{hero_id}")
|
|
async def get_hero(
|
|
*,
|
|
session: Session = Depends(get_session),
|
|
hero_id: int,
|
|
) -> HeroRead:
|
|
"get one hero"
|
|
hero = session.get(Hero, hero_id)
|
|
if not hero:
|
|
raise HTTPException(status_code=404, detail="Hero not found")
|
|
return hero
|
|
|
|
|
|
@hero_router.post("/hero/")
|
|
async def post_hero(
|
|
*,
|
|
session: Session = Depends(get_session),
|
|
hero: HeroCreate,
|
|
) -> HeroRead:
|
|
"read all the heros"
|
|
db_hero = Hero.from_orm(hero)
|
|
session.add(db_hero)
|
|
session.commit()
|
|
session.refresh(db_hero)
|
|
await manager.broadcast({hero.json()}, id=1)
|
|
return db_hero
|
|
|
|
|
|
@hero_router.patch("/hero/")
|
|
async def patch_hero(
|
|
*,
|
|
session: Session = Depends(get_session),
|
|
hero: HeroUpdate,
|
|
) -> HeroRead:
|
|
"read all the heros"
|
|
db_hero = session.get(Hero, hero.id)
|
|
if not db_hero:
|
|
raise HTTPException(status_code=404, detail="Hero not found")
|
|
for key, value in hero.dict(exclude_unset=True).items():
|
|
setattr(db_hero, key, value)
|
|
session.add(db_hero)
|
|
session.commit()
|
|
session.refresh(db_hero)
|
|
await manager.broadcast({hero.json()}, id=1)
|
|
return db_hero
|
|
|
|
|
|
@hero_router.delete("/hero/{hero_id}")
|
|
async def delete_hero(
|
|
*,
|
|
session: Session = Depends(get_session),
|
|
hero_id: int,
|
|
):
|
|
"read all the heros"
|
|
hero = session.get(Hero, hero_id)
|
|
if not hero:
|
|
raise HTTPException(status_code=404, detail="Hero not found")
|
|
session.delete(hero)
|
|
session.commit()
|
|
await manager.broadcast(f"deleted hero {hero_id}", id=1)
|
|
return {"ok": True}
|
|
|
|
|
|
@hero_router.get("/heros/")
|
|
async def get_heros(
|
|
*,
|
|
session: Session = Depends(get_session),
|
|
) -> list[Hero]:
|
|
"get all heros"
|
|
return HeroRead.list(session=session)
|