From e437520217e20d500d641b95482d49484b1f24a7 Mon Sep 17 00:00:00 2001
From: Timo Tuominen <tituomin@gmail.com>
Date: Mon, 1 Sep 2014 17:02:48 +0300
Subject: [PATCH] Generator implementation of class mapping.

---
 rest_framework/serializers.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/rest_framework/serializers.py b/rest_framework/serializers.py
index 5c33300c4..b3db35823 100644
--- a/rest_framework/serializers.py
+++ b/rest_framework/serializers.py
@@ -633,11 +633,10 @@ def _get_class_mapping(mapping, obj):
     from the dictionary or None.
 
     """
-    for baseclass in inspect.getmro(obj.__class__):
-        val = mapping.get(baseclass)
-        if val:
-            return val
-    return None
+    return next(
+        (mapping[cls] for cls in inspect.getmro(obj.__class__) if cls in mapping),
+        None
+    )
 
 
 class ModelSerializer(Serializer):