我应该如何处理SQLalchemy& SQLite的

时间:2012-04-27 18:12:46

标签: sqlite sqlalchemy decimal type-conversion

当我使用带有SQLite数据库引擎的数值列时,SQLalchemy会给出以下警告。

  

SAWarning:方言sqlite + pysqlite 原生支持十进制对象

我正在尝试找出在SQLalchemy中使用SQLite时pkgPrice = Column(Numeric(12,2))的最佳方法。

这个问题[1] How to convert Python decimal to SQLite numeric?显示了一种使用sqlite3.register_adapter(D, adapt_decimal)让SQLite接收并返回Decimal但是存储字符串的方法,但我不知道如何深入研究SQLAlchemy核心这个呢。类型装饰器看起来是正确的方法,但我还没有弄清楚它们。

有没有人在SQLAlchemy模型中有一个SQLAlchemy Type装饰器配方,它会有数字或十进制数字,但是在SQLite中将它们存储为字符串?

3 个答案:

答案 0 :(得分:17)

由于看起来你正在使用小数作为货币值,我建议你做一些安全的事情并将货币的价值存储在其最低面额中,例如: 1610美分而不是16.10美元。然后你可以使用Integer列类型。

这可能不是您期望的答案,但它解决了您的问题,通常被认为是理智的设计。

答案 1 :(得分:16)

from decimal import Decimal as D
import sqlalchemy.types as types

class SqliteNumeric(types.TypeDecorator):
    impl = types.String
    def load_dialect_impl(self, dialect):
        return dialect.type_descriptor(types.VARCHAR(100))
    def process_bind_param(self, value, dialect):
        return str(value)
    def process_result_value(self, value, dialect):
        return D(value)

# can overwrite the imported type name
# @note: the TypeDecorator does not guarantie the scale and precision.
# you can do this with separate checks
Numeric = SqliteNumeric
class T(Base):
    __tablename__ = 't'
    id = Column(Integer, primary_key=True, nullable=False, unique=True)
    value = Column(Numeric(12, 2), nullable=False)
    #value = Column(SqliteNumeric(12, 2), nullable=False)

    def __init__(self, value):
        self.value = value

答案 2 :(得分:2)

这是一个受@van和@JosefAssad启发的解决方案。

class SqliteDecimal(TypeDecorator):
    # This TypeDecorator use Sqlalchemy Integer as impl. It converts Decimals
    # from Python to Integers which is later stored in Sqlite database.
    impl = Integer

    def __init__(self, scale):
        # It takes a 'scale' parameter, which specifies the number of digits
        # to the right of the decimal point of the number in the column.
        TypeDecorator.__init__(self)
        self.scale = scale
        self.multiplier_int = 10 ** self.scale

    def process_bind_param(self, value, dialect):
        # e.g. value = Column(SqliteDecimal(2)) means a value such as
        # Decimal('12.34') will be converted to 1234 in Sqlite
        if value is not None:
            value = int(Decimal(value) * self.multiplier_int)
        return value

    def process_result_value(self, value, dialect):
        # e.g. Integer 1234 in Sqlite will be converted to Decimal('12.34'),
        # when query takes place.
        if value is not None:
            value = Decimal(value) / self.multiplier_int
        return value

就像@Jinghui牛提到的那样,当十进制作为字符串存储在sqlite中时,某些查询将无法始终按预期运行,例如session.query(T).filter(T.value> 100)或sqlalchemy之类的东西.sql.expression.func.min,甚至是order_by,因为SQL会比较字符串(例如,字符串中的“ 9.2”>“ 19.2”)而不是我们在这些情况下期望的数值。