sqlmodel/docs_src/tutorial/fastapi/update/tutorial002.py at patch-1 · Highfire1/sqlmodel · GitHub

Repository navigation

History

100

101

from typing import List, Optional

from fastapi import FastAPI, HTTPException, Query

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

class HeroBase(SQLModel):

name: str = Field(index=True)

secret_name: str

age: Optional[int] = Field(default=None, index=True)

class Hero(HeroBase, table=True):

id: Optional[int] = Field(default=None, primary_key=True)

hashed_password: str = Field()

class HeroCreate(HeroBase):

password: str

class HeroPublic(HeroBase):

id: int

class HeroUpdate(SQLModel):

name: Optional[str] = None

secret_name: Optional[str] = None

age: Optional[int] = None

password: Optional[str] = None

sqlite_file_name = "database.db"

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

connect_args = {"check_same_thread": False}

engine = create_engine(sqlite_url, echo=True, connect_args=connect_args)

def create_db_and_tables():

SQLModel.metadata.create_all(engine)

def hash_password(password: str) -> str:

# Use something like passlib here

return f"not really hashed {password} hehehe"

app = FastAPI()

@app.on_event("startup")

def on_startup():

create_db_and_tables()

@app.post("/heroes/", response_model=HeroPublic)

def create_hero(hero: HeroCreate):

hashed_password = hash_password(hero.password)

with Session(engine) as session:

extra_data = {"hashed_password": hashed_password}

db_hero = Hero.model_validate(hero, update=extra_data)

session.add(db_hero)

session.commit()

session.refresh(db_hero)

return db_hero

@app.get("/heroes/", response_model=List[HeroPublic])

def read_heroes(offset: int = 0, limit: int = Query(default=100, le=100)):

with Session(engine) as session:

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

return heroes

@app.get("/heroes/{hero_id}", response_model=HeroPublic)

def read_hero(hero_id: int):

with Session(engine) as session:

hero = session.get(Hero, hero_id)

if not hero:

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

return hero

@app.patch("/heroes/{hero_id}", response_model=HeroPublic)

def update_hero(hero_id: int, hero: HeroUpdate):

with Session(engine) as session:

db_hero = session.get(Hero, hero_id)

if not db_hero:

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

hero_data = hero.model_dump(exclude_unset=True)

extra_data = {}

if "password" in hero_data:

password = hero_data["password"]

hashed_password = hash_password(password)

extra_data["hashed_password"] = hashed_password

db_hero.sqlmodel_update(hero_data, update=extra_data)

session.add(db_hero)

session.commit()

session.refresh(db_hero)

return db_hero

sqlmodel/docs_src/tutorial/fastapi/update/tutorial002.py at patch-1 · Highfire1/sqlmodel · GitHub

Repository navigation

Expand file tree

Breadcrumbs

tutorial002.py

Latest commit

History

Breadcrumbs

tutorial002.py

File metadata and controls