import asyncio

from sqlalchemy import Column, Integer, Float
from sqlalchemy.orm import declarative_base
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine

Base = declarative_base()

async_engine = create_async_engine(
    "postgresql+asyncpg://user:password@localhost:5450/postgres",
    echo=True,
)

async_session = sessionmaker(async_engine, expire_on_commit=False, class_=AsyncSession)

class Test(Base):
    __tablename__ = "test"
    id = Column(Integer, primary_key=True)
    data = Column(Float)

async def async_main():
    async with async_engine.begin() as conn:
        await conn.run_sync(Base.metadata.drop_all)
        await conn.run_sync(Base.metadata.create_all)

    async with async_session() as session:
        test = Test(data=float("inf"))
        session.add(test)
        await session.commit()


if __name__ == "__main__":
    asyncio.run(async_main())