当前位置：首页 > news >正文

SQLAlchemy系列教程：基本数据类型及自定义类型

news 来源：原创 2025/6/21 22:08:33

在SQLAlchemy、Python SQL工具包和ORM中定义模型时，理解基本数据类型至关重要。本教程提供了在SQLAlchemy模型中有效使用内置基本类型的指南。

SQLAlchemy中的基本类型

SQLAlchemy支持一组与SQL数据库类型一致的基本数据类型。SQLAlchemy中的每种类型都为各种SQL类型提供了Python等价。
在这里插入图片描述

Integer

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

engine = create_engine('sqlite:///:memory:')
metadata = MetaData()
table = Table('example', metadata,
               Column('id', Integer, primary_key=True)
)
metadata.create_all(engine)

String

from sqlalchemy import String

table.append_column(Column('name', String(50)))  # Appending new column to the 'table'
metadata.create_all(engine)

Numeric

from sqlalchemy import Numeric, Float

table.append_column(Column('price', Numeric(10, 2)))
table.append_column(Column('quantity', Float))
metadata.create_all(engine)

用TypeDecorator自定义类型

SQLAlchemy中的类型装饰器允许扩展和定制现有类型。让我们看看如何使用类型装饰器创建一个自定义的json编码类型。

from sqlalchemy.types import TypeDecorator, VARCHAR
import json

class JSONEncodedDict(TypeDecorator):
    """Enables JSON storage by encoding and decoding on the fly."""
    impl = VARCHAR

    def process_bind_param(self, value, dialect):
        if value is not None:
            value = json.dumps(value)
        return value

    def process_result_value(self, value, dialect):
        if value is not None:
            value = json.loads(value)
        return value

table.append_column(Column('attributes', JSONEncodedDict(255)))
metadata.create_all(engine)

高级数据类型

SQLAlchemy的高级数据类型包括枚举、数组等。这里我们将深入研究在SQLAlchemy中定义Enum和Array类型。

Enum类型

import enum
from sqlalchemy import Enum

class MyEnum(enum.Enum):
    foo = 1
    bar = 2
    baz = 3

table.append_column(Column('status', Enum(MyEnum)))
metadata.create_all(engine)

Array类型

# PostgreSQL specific example
from sqlalchemy.dialects.postgresql import ARRAY

table.append_column(Column('data_points', ARRAY(Integer)))
metadata.create_all(engine)