summaryrefslogtreecommitdiff
path: root/django
diff options
context:
space:
mode:
authorMalcolm Tredinnick <malcolm.tredinnick@gmail.com>2009-03-20 03:57:12 +0000
committerMalcolm Tredinnick <malcolm.tredinnick@gmail.com>2009-03-20 03:57:12 +0000
commitfa89fdcd6b4a4ab4887740c0d0f4a73297a5b060 (patch)
treed32b42ab5121a4014b420e70f86c26c4d2222457 /django
parent44b0f68f44c4e5ca6f2b9da294dab1db531abc6e (diff)
Fixed #2698 -- Fixed deleting in the presence of custom managers.
A custom manager on a related object that filtered away objects would prevent those objects being deleted via the relation. This is now fixed. git-svn-id: http://code.djangoproject.com/svn/django/trunk@10104 bcc190cf-cafb-0310-a4f2-bffc1f526a37
Diffstat (limited to 'django')
-rw-r--r--django/db/models/base.py12
-rw-r--r--django/db/models/fields/related.py44
2 files changed, 39 insertions, 17 deletions
diff --git a/django/db/models/base.py b/django/db/models/base.py
index cf5903d289..013dc96e92 100644
--- a/django/db/models/base.py
+++ b/django/db/models/base.py
@@ -519,7 +519,17 @@ class Model(object):
else:
sub_obj._collect_sub_objects(seen_objs, self.__class__, related.field.null)
else:
- for sub_obj in getattr(self, rel_opts_name).all():
+ # To make sure we can access all elements, we can't use the
+ # normal manager on the related object. So we work directly
+ # with the descriptor object.
+ for cls in self.__class__.mro():
+ if rel_opts_name in cls.__dict__:
+ rel_descriptor = cls.__dict__[rel_opts_name]
+ break
+ else:
+ raise AssertionError("Should never get here.")
+ delete_qs = rel_descriptor.delete_manager(self).all()
+ for sub_obj in delete_qs:
sub_obj._collect_sub_objects(seen_objs, self.__class__, related.field.null)
# Handle any ancestors (for the model-inheritance case). We do this by
diff --git a/django/db/models/fields/related.py b/django/db/models/fields/related.py
index 9aa2421f03..1545c73f58 100644
--- a/django/db/models/fields/related.py
+++ b/django/db/models/fields/related.py
@@ -188,7 +188,7 @@ class SingleRelatedObjectDescriptor(object):
return getattr(instance, self.cache_name)
except AttributeError:
params = {'%s__pk' % self.related.field.name: instance._get_pk_val()}
- rel_obj = self.related.model._default_manager.get(**params)
+ rel_obj = self.related.model._base_manager.get(**params)
setattr(instance, self.cache_name, rel_obj)
return rel_obj
@@ -296,13 +296,36 @@ class ForeignRelatedObjectsDescriptor(object):
if instance is None:
return self
+ return self.create_manager(instance,
+ self.related.model._default_manager.__class__)
+
+ def __set__(self, instance, value):
+ if instance is None:
+ raise AttributeError, "Manager must be accessed via instance"
+
+ manager = self.__get__(instance)
+ # If the foreign key can support nulls, then completely clear the related set.
+ # Otherwise, just move the named objects into the set.
+ if self.related.field.null:
+ manager.clear()
+ manager.add(*value)
+
+ def delete_manager(self, instance):
+ """
+ Returns a queryset based on the related model's base manager (rather
+ than the default manager, as returned by __get__). Used by
+ Model.delete().
+ """
+ return self.create_manager(instance,
+ self.related.model._base_manager.__class__)
+
+ def create_manager(self, instance, superclass):
+ """
+ Creates the managers used by other methods (__get__() and delete()).
+ """
rel_field = self.related.field
rel_model = self.related.model
- # Dynamically create a class that subclasses the related
- # model's default manager.
- superclass = self.related.model._default_manager.__class__
-
class RelatedManager(superclass):
def get_query_set(self):
return superclass.get_query_set(self).filter(**(self.core_filters))
@@ -352,17 +375,6 @@ class ForeignRelatedObjectsDescriptor(object):
return manager
- def __set__(self, instance, value):
- if instance is None:
- raise AttributeError, "Manager must be accessed via instance"
-
- manager = self.__get__(instance)
- # If the foreign key can support nulls, then completely clear the related set.
- # Otherwise, just move the named objects into the set.
- if self.related.field.null:
- manager.clear()
- manager.add(*value)
-
def create_many_related_manager(superclass, through=False):
"""Creates a manager that subclasses 'superclass' (which is a Manager)
and adds behavior for many-to-many related objects."""