13

私は一度に2つの新しいことを試みているので、単純化と明確化の両方の支援に感謝します。

from sqlalchemy.ext.declarative import declared_attr
from sqlalchemy import Column, Float, event

class TimeStampMixin(object):

    @declared_attr
    def __tablename__(cls):
        return cls.__name__.lower()

    created = Column(Float)
    modified = Column(Float)
    def __init__(self, created = None,
                       modified = None):
        self.created = created
        self.modified = modified

def create_time(mapper, connection, target):
    target.created = time()

#def modified_time(mapper, connection, target):
#    target.modified = time()

event.listen(TimeStampMixin, 'before_insert', create_time)
#event.listen(TimeStampMixin, 'before_update', modified_time)

したがって、どのクラスにも適用できるミックスインを作成したいと思います。

class MyClass(TimeStampMixin, Base):
    etc, etc, etc

このクラスは、作成時にタイムスタンプを作成し、更新時にタイムスタンプを作成/変更する機能を継承します。

インポート時にこのエラーが発生します：

raise exc.UnmappedClassError(class_)
sqlalchemy.orm.exc.UnmappedClassError: Class 'db.database.TimeStampMixin' is not mapped

aaaそして私はこの時点で困惑しています。

4

4 に答える 4

38

クラスメソッド内にリスナーをアタッチすると、イベントが子クラスにアタッチされます。

class TimeStampMixin(object):
    @staticmethod
    def create_time(mapper, connection, target):
        target.created = time()

    @classmethod
    def __declare_last__(cls):
        # get called after mappings are completed
        # http://docs.sqlalchemy.org/en/rel_0_7/orm/extensions/declarative.html#declare-last
        event.listen(cls, 'before_insert', cls.create_time)
于 2012-11-29T01:52:39.647 に答える
16

before_insertイベントをリッスンするために私が行うことは次のとおりです。現在のクラスを登録し、作成時間の設定を処理するをに追加classmethodします。TimeStampMixin

例えば

class TimeStampMixin(object):

    # other class methods

    @staticmethod
    def create_time(mapper, connection, target):
        target.created = time()

    @classmethod
    def register(cls):
        sqlalchemy.event.listen(cls, 'before_insert', cls.create_time)

そうすれば、次のことができます。

  1. 聞くものと登録するものを簡単に拡張および変更できます。
  2. 特定のクラスのcreate_timeメソッドをオーバーライドします
  3. どのメソッドにタイムスタンプを設定する必要があるかを明確にします。

あなたはそれを簡単に使うことができます：

class MyMappedClass(TimeStampMixin, Base):
    pass

MyMappedClass.register()

シンプルで、非常に明確で、魔法はありませんが、それでもあなたが望むようにカプセル化します。

于 2012-10-05T21:09:24.707 に答える
4

最新のSqlAlchemyでの最良の方法は、@listens_forデコレータをで使用することpropagate=Trueです。

from datetime import datetime
from sqlalchemy import Column, Float
from sqlalchemy.ext.declarative import declared_attr
from sqlalchemy.event import listens_for

class TimestampMixin():
    @declared_attr
    def created(cls):
        return Column(DateTime(timezone=True))

@listens_for(TimeStampMixin, "init", propagate=True)
def timestamp_init(target, args, kwargs):
    kwargs["created"] = datetime.utcnow()
于 2017-05-22T19:40:30.697 に答える
3

次のようにすることもできます。

from sqlalchemy.orm.interfaces import MapperExtension

class BaseExtension(MapperExtension):
    """Base entension class for all entities """

    def before_insert(self, mapper, connection, instance):
        """ set the created_at  """
        instance.created = datetime.datetime.now()

    def before_update(self, mapper, connection, instance):
        """ set the updated_at  """
        instance.modified = datetime.datetime.now()


class TimeStampMixin(object):
    id = Column(Integer, primary_key=True, autoincrement=True)
    created = Column(DateTime())
    modified = Column(DateTime())

    __table_args__ = {
        'mysql_engine': 'InnoDB',
        'mysql_charset': 'utf8'
    }
    __mapper_args__ = { 'extension': BaseExtension() }

次のようにクラスを定義します。

class User(TimeStampMixin, Base):
于 2012-10-06T17:04:59.723 に答える