44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
# -*- coding: UTF-8 -*-
|
|
# vi:si:et:sw=4:sts=4:ts=4
|
|
|
|
import re
|
|
|
|
from ox.cache import read_url
|
|
from ox import find_re
|
|
|
|
def get_data(id):
|
|
'''
|
|
>>> get_data('0060304')['posters'][0]
|
|
u'http://www.movieposterdb.com/posters/06_03/1967/0060304/l_99688_0060304_639fdd1e.jpg'
|
|
>>> get_data('0123456')['posters']
|
|
[]
|
|
'''
|
|
data = {
|
|
"url": get_url(id)
|
|
}
|
|
data["posters"] = get_posters(data["url"])
|
|
return data
|
|
|
|
def get_id(url):
|
|
return url.split("/")[-2]
|
|
|
|
def get_posters(url, group=True, timeout=-1):
|
|
posters = []
|
|
html = read_url(url, timeout=timeout, unicode=True)
|
|
if url in html:
|
|
if group:
|
|
results = re.compile('<a href="(http://www.movieposterdb.com/group/.+?)\??">', re.DOTALL).findall(html)
|
|
for result in results:
|
|
posters += get_posters(result, False)
|
|
results = re.compile('<a href="(http://www.movieposterdb.com/poster/.+?)">', re.DOTALL).findall(html)
|
|
for result in results:
|
|
html = read_url(result, timeout=timeout, unicode=True)
|
|
posters.append(find_re(html, '"(http://www.movieposterdb.com/posters/.+?\.jpg)"'))
|
|
return posters
|
|
|
|
def get_url(id):
|
|
return "http://www.movieposterdb.com/movie/%s/" % id
|
|
|
|
if __name__ == '__main__':
|
|
print get_data('0060304')
|
|
print get_data('0133093')
|