pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implementation of DENY rules.

Browse Source
This commit is contained in:
Vincent Driessen
2010-12-05 23:43:19 -08:00
parent f3da5bc092
commit b06d794854
2 changed files with 43 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
mongoengine/document.py
View File

@@ -99,6 +99,17 @@ class Document(BaseDocument):
:param safe: check if the operation succeeded before returning
"""
# Check for DENY rules before actually deleting/nullifying any other
# references
for rule_entry in self._meta['delete_rules']:
document_cls, field_name = rule_entry
rule = self._meta['delete_rules'][rule_entry]
if rule == DENY and document_cls.objects(**{field_name: self.id}).count() > 0:
msg = u'Could not delete document (at least %s.%s refers to it)' % \
(document_cls.__name__, field_name)
logging.error(msg)
raise OperationError(msg)
for rule_entry in self._meta['delete_rules']:
document_cls, field_name = rule_entry
rule = self._meta['delete_rules'][rule_entry]
@@ -106,7 +117,8 @@ class Document(BaseDocument):
if rule == CASCADE:
document_cls.objects(**{field_name: self.id}).delete(safe=safe)
elif rule == NULLIFY:
document_cls.objects(**{field_name: self.id}).update(**{'unset__%s' % field_name: 1})
document_cls.objects(**{field_name:
self.id}).update(**{'unset__%s' % field_name: 1})
id_field = self._meta['id_field']
object_id = self._fields[id_field].to_mongo(self[id_field])