43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
||
|
|
import json
|
||
|
|
|
||
|
|
from ox.cache import readUrlUnicode
|
||
|
|
from ox import findRe
|
||
|
|
|
||
|
|
class Imdb(dict):
|
||
|
|
def __init__(self, id, timeout=-1):
|
||
|
|
url = "http://ids.freebaseapps.com/get_ids?id=/authority/imdb/title/tt%s" % id
|
||
|
|
'''
|
||
|
|
"http://graph.freebase.com/imdb.title.tt%s" % id
|
||
|
|
might also be of interest at some point, right now not much info
|
||
|
|
'''
|
||
|
|
data = readUrlUnicode(url)
|
||
|
|
try:
|
||
|
|
data = json.loads(data)
|
||
|
|
except ValueError:
|
||
|
|
return
|
||
|
|
'''
|
||
|
|
for key in data:
|
||
|
|
self[key] = data[key]
|
||
|
|
'''
|
||
|
|
for key in ('id', 'guid', 'name'):
|
||
|
|
self[key] = data[key]
|
||
|
|
keys = {
|
||
|
|
'wikipedia': '/wikipedia/en',
|
||
|
|
'netflix': '/authority/netflix/movie',
|
||
|
|
'nytimes': '/source/nytimes/movie',
|
||
|
|
'metacritic': '/source/metacritic/movie',
|
||
|
|
}
|
||
|
|
for key in keys:
|
||
|
|
links = filter(lambda x: x['namespace'] == keys[key],data['ids'])
|
||
|
|
if links:
|
||
|
|
self[key] = links[0]['uri']
|
||
|
|
|
||
|
|
if 'nytimes' in self:
|
||
|
|
self['nytimes'] = self['nytimes'].replace('_/overview', '%s/overview' % self['name'].replace(' ', '-'))
|
||
|
|
self['amgId'] = findRe(self['nytimes'], 'movie/(\d+)/')
|
||
|
|
|
||
|
|
|
||
|
|
|