From 16e1f72e657895e6491f4111e621c510784f596a Mon Sep 17 00:00:00 2001 From: Vincent Driessen Date: Tue, 14 Dec 2010 03:39:14 -0800 Subject: [PATCH] Avoid confusing semantics when comparing delete rules. --- mongoengine/base.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/mongoengine/base.py b/mongoengine/base.py index 42db460f..405f642c 100644 --- a/mongoengine/base.py +++ b/mongoengine/base.py @@ -191,9 +191,10 @@ class DocumentMetaclass(type): new_class = super_new(cls, name, bases, attrs) for field in new_class._fields.values(): field.owner_document = new_class - if hasattr(field, 'delete_rule') and field.delete_rule > DO_NOTHING: + delete_rule = getattr(field, 'delete_rule', DO_NOTHING) + if delete_rule != DO_NOTHING: field.document_type.register_delete_rule(new_class, field.name, - field.delete_rule) + delete_rule) module = attrs.get('__module__')