rename

criterion/cache.py

@@ -8,7 +8,7 @@ from oxlib import findRe
 import oxlib.net
 import oxweb.criterion
 
-from oxdata.lookup.models import IdMapping, getMappingByImdb
+from oxdata.lookup.models import MovieId, getMovieIdByImdbId
 
 
 def getPoster(id, url=None):
@@ -27,15 +27,15 @@ def archivePosters(init=False):
         data = oxweb.criterion.getData(criterionId)
         imdbId = data['imdbId']
         try:
-            m = IdMapping.objects.get(criterion_id=criterionId)
-            if imdbId and imdbId != m.imdb_id:
+            m = MovieId.objects.get(criterion_id=criterionId)
+            if imdbId and not m.imdb_id:
                 m.imdb_id = imdbId
                 m.save()
-        except IdMapping.DoesNotExist:
+        except MovieId.DoesNotExist:
             if imdbId:
-                m = getMappingByImdb(imdbId)
+                m = getMovieIdByImdbId(imdbId)
             else:
-                m = IdMapping()
+                m = MovieId()
             m.criterion_id = criterionId
             m.save()
             url = data['posterUrl']