rename

lookup/models.py

@@ -9,16 +9,16 @@ import simplejson
 import oxweb.wikipedia
 
 
-def getMappingByImdb(imdb_id):
+def getMovieIdByImdbId(imdb_id):
     try:
-        m = IdMapping.objects.get(imdb_id=imdb_id)
-    except IdMapping.DoesNotExist:
-        m = IdMapping()
+        m = MovieId.objects.get(imdb_id=imdb_id)
+    except MovieId.DoesNotExist:
+        m = MovieId()
         m.imdb_id = imdb_id
         m.save()
     return m
 
-class IdMapping(models.Model):
+class MovieId(models.Model):
     created = models.DateTimeField(auto_now_add=True)
     modified = models.DateTimeField(auto_now=True)
 
@@ -26,22 +26,23 @@ class IdMapping(models.Model):
     imdb_id = models.CharField(max_length=7, unique=True, blank=True, null=True,  default=None)
 
     amg_id = models.IntegerField(unique=True, blank=True, null=True,  default=None)
-    wikipedia_url = models.CharField(unique=True, max_length=255, blank=True, null=True,  default=None)
+    wikipedia_id = models.CharField(unique=True, max_length=255, blank=True, null=True,  default=None)
     criterion_id = models.IntegerField(unique=True, blank=True, null=True,  default=None)
-    movieposterdb_url = models.CharField(max_length=255, unique=True, blank=True, null=True,  default=None)
-    impawards_url = models.CharField(max_length=255, unique=True, blank=True, null=True,  default=None)
+    impawards_id = models.CharField(max_length=255, unique=True, blank=True, null=True,  default=None)
     rottentomatoes_id = models.CharField(max_length=255, unique=True, blank=True, null=True,  default=None)
 
     #FIXME: look into other ids
     #what about tv.com ids/urls for tv episodes
-    karagarga_id = models.IntegerField(unique=True, blank=True, null=True,  default=None)
+    
+    #movieposterdb can be mapped via imdbid and does not need to be saved here
 
     def __unicode__(self):
         return self.imdb_id
 
     def updateFromWikipedia(self):
-        if self.wikipedia_url:
-            data = oxweb.wikipedia.getMovieData(self.wikipedia_url)
+        if self.wikipedia_id:
+            wikipedia_url = "http://en.wikipedia.org/wiki/%s" % self.wikipedia_id
+            data = oxweb.wikipedia.getMovieData(wikipedia_url)
             _key = {}
             for key in ('imdb_id',
                         'amg_id',
@@ -57,7 +58,6 @@ class IdMapping(models.Model):
                     'amg_id',
                     'oxdb_id',
                     'wikipedia_url',
-                    'movieposterdb_id',
                     'impawards_url',
                     'rottentomatoes_id'):
             value = getattr(self, key)
@@ -65,3 +65,11 @@ class IdMapping(models.Model):
                 json[key] = value
         return simplejson.dumps(json, indent=4)
 
+def getMappingByKaragargaId(id):
+    kg = Karagarga.objects.get(imdb_id=imdb_id)
+    return kg.mapping
+
+class Karagarga(models.Model):
+    mapping = models.ForeignKey(MovieId, related_name='karagarga_ids', default=None)
+    karagarga_id = models.IntegerField(unique=True)
+

lookup/urls.py

@@ -0,0 +1,7 @@
+from django.conf.urls.defaults import *
+
+urlpatterns = patterns('oxdata.lookup.views',
+    (r'^$', 'ids'),
+    (r'^urls$', 'urls'),
+)
+

lookup/views.py

@@ -1 +1,17 @@
-# Create your views here.
+# -*- coding: utf-8 -*-
+# vi:si:et:sw=4:sts=4:ts=4
+import os.path
+from django.db import models
+from django.db.models import Q
+from django.contrib.auth.models import User
+
+from oxdata.utils.shortcuts import render_to_json_response
+
+def ids(request):
+    json = {}
+    return render_to_json_response(json)
+
+def urls(request):
+    json = {}
+    return render_to_json_response(json)
+