85 lines
3.3 KiB
Python
85 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright(C) 2013 Julien Veyssier
|
|
#
|
|
# This file is part of weboob.
|
|
#
|
|
# weboob is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# weboob is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
from weboob.tools.browser import BaseBrowser
|
|
from weboob.tools.json import json
|
|
|
|
from .pages import MoviePage, PersonPage, MovieCrewPage
|
|
|
|
|
|
__all__ = ['ImdbBrowser']
|
|
|
|
|
|
class ImdbBrowser(BaseBrowser):
|
|
DOMAIN = 'www.imdb.com'
|
|
PROTOCOL = 'http'
|
|
ENCODING = 'utf-8'
|
|
USER_AGENT = BaseBrowser.USER_AGENTS['wget']
|
|
PAGES = {
|
|
'http://www.imdb.com/title/tt[0-9]*/*': MoviePage,
|
|
'http://www.imdb.com/title/tt[0-9]*/fullcredits.*': MovieCrewPage,
|
|
'http://www.imdb.com/name/nm.*': PersonPage,
|
|
}
|
|
|
|
def iter_movies(self, pattern):
|
|
# the api leads to a json result or the html movie page if there is only one result
|
|
self.location('http://www.imdb.com/xml/find?json=1&tt=on&q=%s' % pattern.encode('utf-8'))
|
|
if self.is_on_page(MoviePage):
|
|
id = 'tt'+self.geturl().split('/tt')[1].split('/')[0]
|
|
yield self.page.get_movie(id)
|
|
else:
|
|
res = self.readurl('http://www.imdb.com/xml/find?json=1&tt=on&q=%s' % pattern.encode('utf-8'))
|
|
jres = json.loads(res)
|
|
for restype,mlist in jres.items():
|
|
for m in mlist:
|
|
yield self.get_movie(m['id'])
|
|
|
|
def iter_persons(self, pattern):
|
|
# the api leads to a json result or the html person page if there is only one result
|
|
self.location('http://www.imdb.com/xml/find?json=1&nm=on&q=%s' % pattern.encode('utf-8'))
|
|
if self.is_on_page(PersonPage):
|
|
id = 'nm'+self.geturl().split('/nm')[1].split('/')[0]
|
|
yield self.page.get_person(id)
|
|
else:
|
|
res = self.readurl('http://www.imdb.com/xml/find?json=1&nm=on&q=%s' % pattern.encode('utf-8'))
|
|
jres = json.loads(res)
|
|
for restype,plist in jres.items():
|
|
for p in plist:
|
|
yield self.get_person(p['id'])
|
|
|
|
def get_movie(self, id):
|
|
self.location('http://www.imdb.com/title/%s' % id)
|
|
assert self.is_on_page(MoviePage)
|
|
return self.page.get_movie(id)
|
|
|
|
def get_person(self, id):
|
|
self.location('http://www.imdb.com/name/%s' % id)
|
|
assert self.is_on_page(PersonPage)
|
|
return self.page.get_person(id)
|
|
|
|
def iter_movie_persons(self, movie_id):
|
|
self.location('http://www.imdb.com/title/%s' % movie_id)
|
|
assert self.is_on_page(MoviePage)
|
|
return self.page.iter_persons(movie_id)
|
|
|
|
def iter_person_movies(self, person_id):
|
|
self.location('http://www.imdb.com/name/%s' % person_id)
|
|
assert self.is_on_page(PersonPage)
|
|
return self.page.iter_movies(person_id)
|