FastAPI + SQLModel Postgres DB연동

FastAPI + SQLModel Postgres DB연동 검색해도 잘 안나와서 포스팅 합니다.

다음 공식 사이트는 sqlite로 되어 있습니다.

https://sqlmodel.tiangolo.com/

 SQLModelSQLModel, SQL databases in Python, designed for simplicity, compatibility, and robustness.sqlmodel.tiangolo.com

postgres연동으로 교체한 후 실행

from typing import Optional

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


class Hero(SQLModel, table=True):
    id: Optional[int] = Field(default=None, primary_key=True)
    name: str
    secret_name: str
    age: Optional[int] = None


hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson")
hero_2 = Hero(name="Spider-Boy", secret_name="Pedro Parqueador")
hero_3 = Hero(name="Rusty-Man", secret_name="Tommy Sharp", age=48)

# DB 접속 정보
db_host = "your_db_host" # ex) ec2-123-123-123-123.ap-northeast-2.compute.amazonaws.com
user = "your_db_user"
password = "your_db_password"

engine = create_engine(f'postgresql://{user}:{password}@{db_host}/your_db_name')


SQLModel.metadata.create_all(engine)

with Session(engine) as session:
    session.add(hero_1)
    session.add(hero_2)
    session.add(hero_3)
    session.commit()

with Session(engine) as session:
    statement = select(Hero).where(Hero.name == "Spider-Boy")
    hero = session.exec(statement).first()
    print(hero)

결과

DB에 생성

콘솔 출력