44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Depends
|
|
from sqlmodel import SQLModel
|
|
|
|
from learn_sql_model.api.user import oauth2_scheme
|
|
from learn_sql_model.config import Config, get_config
|
|
from learn_sql_model.models.hero import Hero
|
|
|
|
hero_router = APIRouter()
|
|
|
|
|
|
@hero_router.on_event("startup")
|
|
def on_startup(config: Config = Depends(get_config)) -> None:
|
|
SQLModel.metadata.create_all(config.database.engine)
|
|
|
|
|
|
@hero_router.get("/items/")
|
|
async def read_items(token: Annotated[str, Depends(oauth2_scheme)]):
|
|
return {"token": token}
|
|
|
|
|
|
@hero_router.get("/hero/{id}")
|
|
def get_hero(id: int, config: Config = Depends(get_config)) -> Hero:
|
|
"get one hero"
|
|
return Hero().get(id=id, config=config)
|
|
|
|
|
|
@hero_router.post("/hero/")
|
|
def post_hero(hero: Hero, config: Config = Depends(get_config)) -> Hero:
|
|
"read all the heros"
|
|
hero.post(config=config)
|
|
return hero
|
|
|
|
|
|
@hero_router.get("/heros/")
|
|
def get_heros(config: Config = Depends(get_config)) -> list[Hero]:
|
|
"get all heros"
|
|
return Hero().get(config=config)
|
|
# Alternatively
|
|
# with get_config().database.session as session:
|
|
# statement = select(Hero)
|
|
# results = session.exec(statement).all()
|
|
# return results
|