diff options
| author | Tom Christie | 2012-11-05 12:51:04 +0000 |
|---|---|---|
| committer | Tom Christie | 2012-11-05 12:51:04 +0000 |
| commit | 33be4b43b95273da117f6896adc7814aaedf43fa (patch) | |
| tree | ac203801b75a0c4681d0b44d1626d95c87c5bb47 /rest_framework/fields.py | |
| parent | ccd5f29510bd39e85ea8267836e42545370fff1e (diff) | |
| download | django-rest-framework-33be4b43b95273da117f6896adc7814aaedf43fa.tar.bz2 | |
`queryset` argument is now optional on writable model fields.
Diffstat (limited to 'rest_framework/fields.py')
| -rw-r--r-- | rest_framework/fields.py | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/rest_framework/fields.py b/rest_framework/fields.py index eb4c5e1f..dc726a32 100644 --- a/rest_framework/fields.py +++ b/rest_framework/fields.py @@ -40,7 +40,7 @@ class Field(object): self.source = source - def initialize(self, parent): + def initialize(self, parent, field_name): """ Called to set up a field prior to field_to_native or field_from_native. @@ -248,7 +248,22 @@ class RelatedField(WritableField): def __init__(self, *args, **kwargs): self.queryset = kwargs.pop('queryset', None) super(RelatedField, self).__init__(*args, **kwargs) - self.read_only = self.default_read_only + self.read_only = kwargs.pop('read_only', self.default_read_only) + + def initialize(self, parent, field_name): + super(RelatedField, self).initialize(parent, field_name) + if self.queryset is None and not self.read_only: + try: + manager = getattr(self.parent.opts.model, self.source or field_name) + if hasattr(manager, 'related'): # Forward + self.queryset = manager.related.model._default_manager.all() + else: # Reverse + self.queryset = manager.field.rel.to._default_manager.all() + except: + raise + msg = ('Serializer related fields must include a `queryset`' + + ' argument or set `read_only=True') + raise Exception(msg) ### We need this stuff to make form choices work... |
