some wikipedia

This commit is contained in:
j 2008-04-29 15:54:08 +02:00
parent 3bcdd5e169
commit f7947fb72a

37
ox/wikipedia.py Normal file
View file

@ -0,0 +1,37 @@
# -*- Mode: Python; -*-
# -*- coding: utf-8 -*-
# vi:si:et:sw=2:sts=2:ts=2
from urllib import urlencode
import simplejson
from oxutils.cache import getUrl
def getMovieId(title, director='', year=''):
query = '"%s" film %s %s' % (title, director, year)
result = find(query, 1)
if result:
return result[0][1]
return ''
def getUrlByImdb(imdbId):
if len(imdbId) != 7: return ''
result = find(imdbId)
if result:
url = result[0][1]
return url
return ''
def find(query, max_results=10):
query = {'action': 'query', 'list':'search', 'format': 'json',
'srlimit': max_results, 'srwhat': 'text', 'srsearch': query.encode('utf-8')}
url = "http://en.wikipedia.org/w/api.php?" + urlencode(query)
data = getUrl(url)
result = simplejson.loads(data)
results = []
for r in result['query']['search']:
title = r['title']
url = "http://en.wikipedia.org/wiki/%s" % title.replace(' ', '_')
results.append((title, url, ''))
return results