getMappingByImdb

lookup/models.py

@@ -9,6 +9,15 @@ import simplejson
 import oxweb.wikipedia
 
 
+def getMappingByImdb(imdb_id):
+    try:
+        m = IdMapping.objects.get(imdb_id=imdb_id)
+    except IdMapping.DoesNotExist:
+        m = IdMapping()
+        m.imdb_id = imdb_id
+        m.save()
+    return m
+
 class IdMapping(models.Model):
     created = models.DateTimeField(auto_now_add=True)
     modified = models.DateTimeField(auto_now=True)