48 lines
1.8 KiB
Python
48 lines
1.8 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, strip_tags
|
|
|
|
|
|
def get_url(id=None, imdb=None):
|
|
if imdb:
|
|
url = "http://www.rottentomatoes.com/alias?type=imdbid&s=%s" % imdb
|
|
data = read_url(url)
|
|
if "movie_title" in data:
|
|
movies = re.compile('(/m/.*?/)').findall(data)
|
|
if movies:
|
|
return "http://www.rottentomatoes.com" + movies[0]
|
|
return None
|
|
|
|
def get_og(data, key):
|
|
return find_re(data, '<meta property="og:%s".*?content="(.*?)"' % key)
|
|
|
|
def get_data(url):
|
|
data = read_url(url)
|
|
r = {}
|
|
r['title'] = find_re(data, '<h1 class="movie_title">(.*?)</h1>')
|
|
if '(' in r['title']:
|
|
r['year'] = find_re(r['title'], r'\((\d*?)\)')
|
|
r['title'] = strip_tags(re.sub(r'\((\d*?)\)', '', r['title'])).strip()
|
|
r['summary'] = strip_tags(find_re(data, '<p id="movieSynopsis" class="movie_synopsis" itemprop="description">(.*?)</p>')).strip()
|
|
r['summary'] = r['summary'].replace('\t', ' ').replace('\n', ' ').replace(' ', ' ').replace(' ', ' ')
|
|
if not r['summary']:
|
|
r['summary'] = get_og(data, 'description')
|
|
|
|
meter = re.compile('<span id="all-critics-meter" class="meter(.*?)">(.*?)</span>').findall(data)
|
|
meter = [m for m in meter if m[1].isdigit()]
|
|
if meter:
|
|
r['tomatometer'] = meter[0][1]
|
|
r['rating'] = find_re(data, r'Average Rating: <span>([\d.]+)/10</span>')
|
|
r['user_score'] = find_re(data, r'<span class="meter popcorn numeric ">(\d+)</span>')
|
|
r['user_rating'] = find_re(data, r'Average Rating: ([\d.]+)/5')
|
|
poster = get_og(data, 'image')
|
|
if poster and not 'poster_default.gif' in poster:
|
|
r['posters'] = [poster]
|
|
for key in r.keys():
|
|
if not r[key]:
|
|
del r[key]
|
|
return r
|
|
|