Fixed the model for Django<1.6

encode · Jul 25, 2015 · 6b54703 · 6b54703
1 parent 16508cf
commit 6b54703
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/models.py b/tests/models.py
@@ -49,7 +49,10 @@ class ForeignKeyTarget(RESTFrameworkModel):
     name = models.CharField(max_length=100)
 
     def get_first_source(self):
-        return self.sources.first()
+        try:
+            return self.sources.all()[0]
+        except IndexError:
+            return None
 
 
 class ForeignKeySource(RESTFrameworkModel):