mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-06-05 03:29:12 +00:00
Code Issues 32 Releases Wiki Activity

magic-removal: fixed RelatedManager by making it work as a custom manager, and 24 related tests

Browse Source 
git-svn-id: http://code.djangoproject.com/svn/django/branches/magic-removal@2207 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Luke Plant 2006-01-31 22:26:53 +00:00
parent a37d862ded
commit cef5e5421d
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
django/db/models/fields/related.py
View File

@ -109,14 +109,19 @@ class ManyRelatedObjectsDescriptor(object):
# model's default manager. # model's default manager.
manager = types.ClassType('RelatedManager', (self.related.model._default_manager.__class__,), {})() manager = types.ClassType('RelatedManager', (self.related.model._default_manager.__class__,), {})()
# Override get_query_set on the RelatedManager
def get_query_set(self):
return super(RelatedManager, self).filter(**core_filters)
manager.get_query_set = get_query_set
# Set core_filters on the new manager to limit it to the # Set core_filters on the new manager to limit it to the
# foreign-key relationship. # foreign-key relationship.
rel_field = self.related.field rel_field = self.related.field
if self.rel_type == 'o2m': if self.rel_type == 'o2m':
manager.core_filters.update({'%s__%s__exact' % (rel_field.name, rel_field.rel.to._meta.pk.name): getattr(instance, rel_field.rel.get_related_field().attname)}) manager.core_filters = {'%s__%s__exact' % (rel_field.name, rel_field.rel.to._meta.pk.name): getattr(instance, rel_field.rel.get_related_field().attname)}
else: else:
manager.core_filters.update({'%s__%s__exact' % (rel_field.name, instance_type._meta.pk.name): instance._get_pk_val()}) manager.core_filters = {'%s__%s__exact' % (rel_field.name, instance_type._meta.pk.name): instance._get_pk_val()}
manager.core_values = {rel_field.name: instance} manager.core_values = {rel_field.name: instance}
# Prepare the manager. # Prepare the manager.