Coerce pydantic model to SQLAlchemy Table **imperatively**

pydantic_to_sqlalchemy.py

from sqlalchemy.orm import declarative_base
from sqlalchemy import Text, Integer, JSON, Table, Column


Base = declarative_base()

def convert_pydantic_field_to_sqlalchemy_column(field_name, field_type):
    if field_type == str:
        return Column(field_name, Text)

    if field_type == int:
        return Column(field_name, Integer)

    if field_type.__origin__ == list:
        return Column(field_name, JSON)

    if field_type.__origin__ == typing.Union:
        nullable = types.NoneType in field_type.__args__
        return Column(field_name, JSON, nullable=nullable)

    raise ValueError(f"Value Not Supported! {field_name}:{field_type}")


def coerce_pydantic_model_to_sqlalchemy_table(table_name: str, base, model: Type[BaseModel]):

    columns = []
    fields = get_type_hints(model)

    columns.extend([convert_pydantic_field_to_sqlalchemy_column(field_name, field_type) for field_name, field_type in fields.items()])

    return Table(
        table_name,
        base.metadata,
        Column("id", Integer, primary_key=True),
        *columns
    )