iTunes automation cleanup

This commit is contained in:
Ruud committed 2013-01-04 20:20:52 +01:00
1 parent da429f0cb8
commit 637b21cc68
2 files changed
+13 -13

No files matched your search

@@ -1,4 +1,4 @@
from .main import ITunes from .main import ITunes
def start(): def start():
return ITunes() return ITunes()
@@ -1,11 +1,11 @@
from couchpotato.core.helpers.rss import RSS from couchpotato.core.helpers.rss import RSS
from couchpotato.core.helpers.variable import md5, getImdb, splitString, tryInt from couchpotato.core.helpers.variable import md5, splitString, tryInt
from couchpotato.core.logger import CPLog from couchpotato.core.logger import CPLog
from couchpotato.core.providers.automation.base import Automation from couchpotato.core.providers.automation.base import Automation
import datetime
import xml.etree.ElementTree as XMLTree
from xml.etree.ElementTree import QName from xml.etree.ElementTree import QName
import datetime
import traceback import traceback
import xml.etree.ElementTree as XMLTree
log = CPLog(__name__) log = CPLog(__name__)
@@ -26,7 +26,7 @@ class ITunes(Automation, RSS):
namespace = 'http://www.w3.org/2005/Atom' namespace = 'http://www.w3.org/2005/Atom'
namespaceIM = 'http://itunes.apple.com/rss' namespaceIM = 'http://itunes.apple.com/rss'
index = -1 index = -1
for url in urls: for url in urls:
@@ -39,24 +39,24 @@ class ITunes(Automation, RSS):
rss_data = self.getCache(cache_key, url) rss_data = self.getCache(cache_key, url)
data = XMLTree.fromstring(rss_data) data = XMLTree.fromstring(rss_data)
if data is not None: if data is not None:
entry_tag = str(QName(namespace, 'entry')) entry_tag = str(QName(namespace, 'entry'))
rss_movies = self.getElements(data, entry_tag) rss_movies = self.getElements(data, entry_tag)
for movie in rss_movies: for movie in rss_movies:
name_tag = str(QName(namespaceIM, 'name')) name_tag = str(QName(namespaceIM, 'name'))
name = self.getTextElement( movie, name_tag ) name = self.getTextElement(movie, name_tag)
releaseDate_tag = str(QName(namespaceIM, 'releaseDate')) releaseDate_tag = str(QName(namespaceIM, 'releaseDate'))
releaseDateText = self.getTextElement(movie, releaseDate_tag) releaseDateText = self.getTextElement(movie, releaseDate_tag)
year = datetime.datetime.strptime(releaseDateText, '%Y-%m-%dT00:00:00-07:00').strftime("%Y") year = datetime.datetime.strptime(releaseDateText, '%Y-%m-%dT00:00:00-07:00').strftime("%Y")
imdb = self.search(name, year) imdb = self.search(name, year)
if imdb and self.isMinimalMovie(imdb): if imdb and self.isMinimalMovie(imdb):
movies.append(imdb['imdb']) movies.append(imdb['imdb'])
except: except:
log.error('Failed loading iTunes rss feed: %s %s', (url, traceback.format_exc())) log.error('Failed loading iTunes rss feed: %s %s', (url, traceback.format_exc()))