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
    secret_name: str = Field(..., alias="myLittleSecretAlias")  # Using alias for JSON serialization
    age: int | None = None


sqlite_file_name = "database.db"
sqlite_url = f"sqlite:///{sqlite_file_name}"

engine = create_engine(sqlite_url, echo=True)

SQLModel.metadata.create_all(engine)

with Session(engine) as session:

    hero_1 = Hero(name="Deadpond", myLittleSecretAlias="Dive Wilson")
    session.add(hero_1)


    statement = select(Hero).where(Hero.name == "Deadpond")
    hero = session.exec(statement).one_or_none()

    # without the fix the result is :
    #  {'name': 'Deadpond', 'secret_name': 'Dive Wilson', 'id': 1, 'age': None}
    # with the fix it is :
    #  {'name': 'Deadpond', 'myLittleSecretAlias': 'Dive Wilson', 'id': 1, 'age': None}
    print(hero.model_dump(by_alias=True))