SQLAlchemy：如何将 hybrid_property 变更为 declared_attr？

Question

我希望能够用不同的参数来调用 Duration.create_duration_field()，这样我的类就能创建多个 hybrid_property。它们之间唯一的区别就是每个属性会减去不同的时间戳。

当然，使用 declarative_attr 并不是必须的，但我需要这些属性是 hybrid_property。

import datetime
from sqlalchemy import create_engine, MetaData
from sqlalchemy import Column, String, DateTime, Integer
from sqlalchemy.ext.declarative import declarative_base, declared_attr
from sqlalchemy.ext.hybrid import hybrid_property
from sqlalchemy.orm import sessionmaker


metadata = MetaData()
Base = declarative_base(metadata=metadata)


class Duration(Base):
    __tablename__ = "duration"

    pk = Column(Integer, primary_key=True)
    name = Column(String, nullable=False)
    notes = Column(String)
    timestamp_initiated = Column(DateTime,
                                default=datetime.datetime.now(),
                                nullable=False)
    timestamp_done = Column(DateTime)

    def __unicode__(self):
        return self.name

    @declared_attr
    def duration(cls):
        return cls.create_duration_field("initiated", "done")

    @classmethod
    def create_duration_field(cls, start, end):
        @hybrid_property
        def duration(obj):
            getattr(obj, "timestamp_%s" % end) - getattr(obj, "timestamp_%s" % start)
        @duration.expression
        def duration(cls):
            return getattr(cls, "timestamp_%s" % end) - getattr(cls, "timestamp_%s" % start)
        return duration



engine = create_engine('sqlite:///:memory:', echo=True)
metadata.create_all(engine)
Session = sessionmaker(bind=engine)
session = Session()

duration = Duration(name="Test", timestamp_done=datetime.datetime.now() + datetime.timedelta(seconds=25))
session.add(duration)
session.commit()

assert isinstance(duration.duration, datetime.timedelta)

目前 duration.duration 是一个指向 hybrid_property 的引用。

<sqlalchemy.ext.hybrid.hybrid_property object at 0x1d02190>

Answer 1

我运行了这个脚本，没问题（版本0.9.6）。你只需要把代码从：

@hybrid_property
def duration(obj):
    getattr(obj, "timestamp_%s" % end) - getattr(obj, "timestamp_%s" % start)

改成：

@hybrid_property
def duration(obj):
    return getattr(obj, "timestamp_%s" % end) - getattr(obj, "timestamp_%s" % start)

结果：

duration.duration = 0:00:25.017861 datetime.timedelta(0, 25, 17861) <type 'datetime.timedelta'>

Answer 2

我现在还在0.6版本，所以不能试你那些高级的属性。 :)

不过，除了

class Duration(Base):
    # [...] 
    @declared_attr
    def duration(cls):
        return cls.create_duration_field("initiated", "done")

你可以试试这样的：

class Duration(Base):

    duration = declared_attr(create_duration_field("initiated", "done"))

把你的 *create_duration_field* 这个类方法移出那个类。这样你就可以在不同的类中重复使用它。也许declared_attr也应该放到 *create_duration_field* 里面，你可以自己试试看。

一个例子就是我一直在用的这个小东西：

def constructor(**kw):                                                          
    """Creates a class constructor class method"""
    def load(cls):
        query = cls.query
        return query.filter_by(**kw).one()
    return classmethod(load)

你可以这样使用它：

class SomeModel(Base):

    SomeInstance = constructor(primary_key=17)

instance = SomeModel.SomeInstance()

希望你能明白这个意思。