SQLAlchemy: Как сделать гибридное_производство объявленным_аттром? - PullRequest
Новая
       35

SQLAlchemy: Как сделать гибридное_производство объявленным_аттром?

2 голосов
/ 13 июля 2011

Я хотел бы иметь возможность вызывать 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>

Ответы [ 2 ]

0 голосов
/ 04 июля 2014

Я запустил скрипт и всё в порядке (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'>
0 голосов
/ 13 июля 2011

Я все еще на 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 * classmethod из этого класса.Таким образом, вы также можете использовать его в разных классах.Возможно, Объявленный_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()

Надеюсь, вы поняли идею.

...