Mongoengine 信号监听所有模型

Question

我已经使用 mongoengine 设置了 django 项目，以便将 mongodb 与 django 一起使用。我已经创建了 2 个模型并且它们工作正常，但是当我对一个模型使用信号监听器时它也会监听另一个模型，那么我怎样才能将信号绑定到它们的模型？

这是我的模型用户代码：

from mongoengine import *
from mongoengine import signals
from datetime import datetime


class User(Document):
    uid = StringField(max_length=60, required=True)
    platform = StringField(max_length=20, required=True)
    index = StringField(max_length=80)
    last_updated = DateTimeField(required=True, default=datetime.now())

    meta = {
        'collection': 'social_users'
    }


def before_save(sender, document, **kwargs):
    if document.platform and document.uid:
        document.index = document.platform+'/'+document.uid

signals.pre_save.connect(before_save)

这是另一个模型Error

from mongoengine import *
from datetime import datetime


class Error(Document):
    call = DictField(required=True)
    response = DictField(required=True)
    date = DateTimeField(default=datetime.now(), required=True)

    meta = {
        'collection': 'errors'
    }

这是我用来测试代码的文件：

from src.social.models.error import Error
from src.social.models.user import User

error = Error.objects.first()

print(error.to_json())

但它不起作用，抛出以下错误：

AttributeError: 'Error' object has no attribute 'platform'

请帮我解决这个问题，谢谢。

Answer 1

我找到了一种为特定模型绑定信号的方法，下面是我的代码：

from mongoengine import *
from mongoengine import signals
from datetime import datetime


class User(Document):
    uid = StringField(max_length=60, required=True)
    platform = StringField(max_length=20, required=True)
    index = StringField(max_length=80)
    last_updated = DateTimeField(required=True, default=datetime.now())

    meta = {
        'collection': 'social_users'
    }

    @classmethod
    def pre_save(cls, sender, document, **kwargs):
        if document.platform and document.uid:
            document.index = document.platform+'/'+document.uid

signals.pre_save.connect(User.pre_save, sender=User)

希望对遇到同样问题的人有所帮助。

Mongoengine 信号监听所有模型

Mongoengine signals listens for all models

python

django

mongodb

mongoengine