some wikipedia
This commit is contained in:
parent
3bcdd5e169
commit
f7947fb72a
1 changed files with 37 additions and 0 deletions
37
ox/wikipedia.py
Normal file
37
ox/wikipedia.py
Normal 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
|
||||||
|
|
Loading…
Reference in a new issue