SickGear/sickbeard/rssfeeds.py

from __future__ import with_statement

import os
import urllib
import urlparse
import re
import shelve
import sickbeard

from sickbeard import logger
from sickbeard import encodingKludge as ek
from contextlib import closing
from lib.feedcache import cache
from sickbeard.exceptions import ex


class RSSFeeds:
    def __init__(self, db_name):
        self.db_name = ek.ek(os.path.join, sickbeard.CACHE_DIR, db_name)

    def clearCache(self, age=None):
        try:
            with closing(shelve.open(self.db_name, writeback=True)) as fs:
                fc = cache.Cache(fs)
                fc.purge(age)
        except Exception as e:
            logger.log(u"RSS Error: " + ex(e), logger.ERROR)
            logger.log(u"RSS cache file corrupted, please delete " + self.db_name, logger.ERROR)

    def getFeed(self, url, post_data=None, request_headers=None):
        parsed = list(urlparse.urlparse(url))
        parsed[2] = re.sub("/{2,}", "/", parsed[2])  # replace two or more / with one

        if post_data:
            url += urllib.urlencode(post_data)

        try:
            with closing(shelve.open(self.db_name, writeback=True)) as fs:
                fc = cache.Cache(fs)
                feed = fc.fetch(url, False, False, request_headers)
        except Exception as e:
            logger.log(u"RSS Error: " + ex(e), logger.ERROR)
            logger.log(u"RSS cache file corrupted, please delete " + self.db_name, logger.ERROR)
            feed = None

        if not feed:
            logger.log(u"RSS Error loading URL: " + url, logger.ERROR)
            return
        elif 'error' in feed.feed:
            logger.log(u"RSS ERROR:[%s] CODE:[%s]" % (feed.feed['error']['description'], feed.feed['error']['code']),
                       logger.DEBUG)
            return
        elif not feed.entries:
            logger.log(u"No RSS items found using URL: " + url, logger.WARNING)
            return

        return feed