Support assignment in ClassLookupDict

encode · Jan 23, 2015 · 889a07f · 889a07f
1 parent 5bb3486
commit 889a07f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/rest_framework/utils/field_mapping.py b/rest_framework/utils/field_mapping.py
@@ -38,6 +38,9 @@ def __getitem__(self, key):
                 return self.mapping[cls]
         raise KeyError('Class %s not found in lookup.', cls.__name__)
 
+    def __setitem__(self, key, value):
+        self.mapping[key] = value
+
 
 def needs_label(model_field, field_name):
     """