from fastapi import FastAPI, Depends, HTTPException

from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine

from sqlalchemy.orm import sessionmaker

from sqlmodel import SQLModel, Field, create_engine

​

# PostgreSQL connection URL

DATABASE_URL = "postgresql+asyncpg://user:password@localhost/dbname"

​

# Create the async engine

async_engine = create_async_engine(DATABASE_URL)

​

# Create async session class

async_session = sessionmaker(

    autocommit=False, autoflush=False, bind=async_engine, class_=AsyncSession

)

​

# Dependency to get async session

async def get_session():

    async with async_session() as session:

        yield session

​

# Example model

class Item(SQLModel, table=True):

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

    name: str

    description: str

​

app = FastAPI()

​

# Route to create a new item

@app.post("/items/")

async def create_item(item: Item, session: AsyncSession = Depends(get_session)):

    async with session.begin():

        session.add(item)

        await session.commit()

        await session.refresh(item)

        return item

​