# -*- 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

import simplejson
import oxweb.wikipedia


def getMovieIdByImdbId(imdb_id):
    try:
        m = MovieId.objects.get(imdb_id=imdb_id)
    except MovieId.DoesNotExist:
        m = MovieId()
        m.imdb_id = imdb_id
        m.save()
    return m

class MovieId(models.Model):
    created = models.DateTimeField(auto_now_add=True)
    modified = models.DateTimeField(auto_now=True)

    oxdb_id = models.CharField(max_length=42, unique=True, blank=True, null=True,  default=None)
    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_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)
    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
    
    #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_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',
                        'rottentomatoes_id'):
                if key in data:
                    if data[key]:
                        setattr(self, _key.get(key, key), data[key])
            self.save()

    def json(self):
        json = {} 
        for key in ('imdb_id',
                    'amg_id',
                    'oxdb_id',
                    'wikipedia_url',
                    'impawards_url',
                    'rottentomatoes_id'):
            value = getattr(self, key)
            if value:
                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)