Private GIT

Skip to content
Snippets Groups Projects
Commit 94478336 authored by Ruud's avatar Ruud
Browse files

RT cleanup

parent df53d0c5
Branches
Tags
No related merge requests found
......@@ -10,7 +10,7 @@ config = [{
'tab': 'automation',
'name': 'rottentomatoes_automation',
'label': 'Rottentomatoes',
'description': 'Imports movies from the rottentomatoes in theaters rss feed.',
'description': 'Imports movies from the rottentomatoes "in theaters"-feed.',
'options': [
{
'name': 'automation_enabled',
......
from couchpotato.core.helpers.rss import RSS
from couchpotato.core.helpers.variable import md5
from couchpotato.core.helpers.variable import tryInt
from couchpotato.core.logger import CPLog
from couchpotato.core.providers.automation.base import Automation
from xml.etree.ElementTree import QName
import datetime
import xml.etree.ElementTree as XMLTree
import re
log = CPLog(__name__)
......@@ -12,24 +11,17 @@ log = CPLog(__name__)
class Rottentomatoes(Automation, RSS):
interval = 1800
rss_url = 'http://www.rottentomatoes.com/syndication/rss/in_theaters.xml'
urls = {
'namespace': 'http://www.rottentomatoes.com/xmlns/rtmovie/',
'theater': 'http://www.rottentomatoes.com/syndication/rss/in_theaters.xml',
}
def getIMDBids(self):
if self.isDisabled():
return
movies = []
cache_key = 'rottentomatoes.%s' % md5(self.rss_url)
rss_data = self.getCache(cache_key, self.rss_url)
data = XMLTree.fromstring(rss_data)
if data:
namespace = 'http://www.rottentomatoes.com/xmlns/rtmovie/'
rating_tag = str(QName(namespace, 'tomatometer_percent'))
rss_movies = self.getElements(data, 'channel/item')
rss_movies = self.getRSSData(self.urls['theater'])
rating_tag = str(QName(self.urls['namespace'], 'tomatometer_percent'))
for movie in rss_movies:
......@@ -38,17 +30,15 @@ class Rottentomatoes(Automation, RSS):
if result:
log.info('Something smells...')
rating = int(self.getTextElement(movie, rating_tag))
log.info2('Something smells...')
rating = tryInt(self.getTextElement(movie, rating_tag))
name = result.group(0)
if rating < int(self.conf('tomatometer_percent')):
log.info('%s seems to be rotten...' % name)
if rating < tryInt(self.conf('tomatometer_percent')):
log.info2('%s seems to be rotten...' % name)
else:
log.info('You find %s fresh enough though, enqueuing: %s' % (rating, name))
log.info2('Found %s fresh enough movies, enqueuing: %s' % (rating, name))
year = datetime.datetime.now().strftime("%Y")
imdb = self.search(name, year)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment