如何在SQLAlchemy中将tinyint(1)自动转换为bool类型？

下面是几种简洁、Pythonic的处理方式，按推荐优先级排序：

1. 定义表时直接指定对应Boolean类型 #

这是最根本的解决方式，让SQLAlchemy自动帮你做类型转换。如果是手动创建Table对象，把tinyint(1)对应的列类型设为Boolean()：

from sqlalchemy import Table, Column, Integer, Boolean, MetaData

metadata = MetaData()
my_table = Table(
    "your_table_name",
    metadata,
    Column("id", Integer, primary_key=True),
    Column("is_enabled", Boolean),  # 对应MySQL的tinyint(1)
    # 其他列...
)

如果是反射已存在的表，可以通过type_map指定类型映射：

from sqlalchemy import create_engine, MetaData

engine = create_engine("mysql+pymysql://user:pass@host/db")
metadata = MetaData()
metadata.reflect(
    engine,
    only=["your_table_name"],
    type_map={"TINYINT": Boolean}  # 把所有TINYINT映射为Boolean
)
my_table = metadata.tables["your_table_name"]

这样查询出来的对应列直接是bool类型，无需额外处理。

2. 自定义TypeDecorator全局处理 #

如果需要更灵活的控制（比如只针对tinyint(1)转换，不影响其他tinyint），可以自定义类型装饰器：

from sqlalchemy import TypeDecorator, Integer

class TinyIntToBool(TypeDecorator):
    impl = Integer  # 对应数据库的tinyint类型
    cache_ok = True

    def process_result_value(self, value, dialect):
        # 查询结果转换为bool
        return bool(value) if value is not None else None

    def process_bind_param(self, value, dialect):
        # 写入数据库时把bool转成0/1
        return 1 if value else 0 if value is not None else None

# 定义表时使用这个类型
my_table = Table(
    "your_table_name",
    metadata,
    Column("id", Integer, primary_key=True),
    Column("is_enabled", TinyIntToBool),
)

这个方式能全局生效，查询和写入都自动处理类型转换。

3. 查询时用CAST转换列 #

如果不想修改表定义，可以在查询语句中直接把目标列转为布尔类型：

from sqlalchemy import select, cast, Boolean

query = select(
    my_table.c.id,
    cast(my_table.c.is_enabled, Boolean).label("is_enabled"),
    # 其他列...
).select_from(my_table)

result = query.all()
# 此时result中的is_enabled就是bool类型

4. 结果后处理（批量转换） #

如果以上方法都不适用，可以在查询结果返回后，批量转换指定列：

# 先执行查询
rows = select(my_table).select_from(my_table).all()

# 定义需要转换的列名列表
bool_columns = ["is_enabled", "is_verified"]

# 转换为字典并处理布尔值
processed_data = []
for row in rows:
    row_dict = dict(row)
    for col in bool_columns:
        if col in row_dict:
            row_dict[col] = bool(row_dict[col])
    processed_data.append(row_dict)

# 之后用json.dumps处理processed_data即可

内容的提问来源于stack exchange，提问作者jenny03