Выполнение кода перед созданием документа в mongoengine

Question

Как я могу выполнить некоторый код только в том случае, если документ создан в mongoengine, а не при обновлении.

class Account(Document): 
    name = StringField(max_length=80, default=None) 
    username = StringField(max_length=60, required=True) 
    created_at = DateTimeField(default=datetime.now(), required=True) 
    updated_at = DateTimeField(default=datetime.now(), required=True) 

    meta = { 
     'collection': 'accounts' 
    } 

Теперь я хочу генерировать случайное имя пользователя и назначить его в поле имени пользователя до создания документа.

Любая помощь приветствуется. Благодарю.

Answer 1

Вы должны использовать один из MongoEngine's signals - pre_save() звучит как подходящий. Есть different ways присоединить обработчик события к сигналу, вот один из них:

from mongoengine import signals 

class Account(Document): 
    # ... 

    @classmethod 
    def pre_save(cls, sender, document, **kwargs): 
     document.username = "random username" 

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