SickRage/sickbeard/rssfeeds.py

from __future__ import with_statement

import os
import urllib
import urlparse
import re

import sickbeard

from sickbeard import logger
from sickbeard import encodingKludge as ek
from sickbeard.exceptions import ex

from contextlib import closing
from lib.feedcache import cache
from sqliteshelf import SQLiteShelf

class RSSFeeds:
    def __init__(self, db_name):
        db_name = ek.ek(os.path.join, sickbeard.CACHE_DIR, 'rss', db_name) + '.db'
        if not os.path.exists(os.path.dirname(db_name)):
            sickbeard.helpers.makeDir(os.path.dirname(db_name))

        try:
            self.rssDB = SQLiteShelf(db_name)
        except Exception as e:
            logger.log(u"RSS error: " + ex(e), logger.DEBUG)

    def clearCache(self, age=None):
        try:
            with closing(self.rssDB) as fs:
                fc = cache.Cache(fs)
                fc.purge(age)
        except Exception as e:
            logger.log(u"RSS error clearing cache: " + ex(e), logger.DEBUG)

    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(self.rssDB) as fs:
                fc = cache.Cache(fs)
                feed = fc.fetch(url, False, False, request_headers)

                if feed:
                    if 'entries' in feed:
                        return feed
                    elif 'error' in feed.feed:
                        err_code = feed.feed['error']['code']
                        err_desc = feed.feed['error']['description']

                        logger.log(
                            u"RSS ERROR:[%s] CODE:[%s]" % (err_desc, err_code), logger.DEBUG)
                else:
                    logger.log(u"RSS error loading url: " + url, logger.DEBUG)

        except Exception as e:
            logger.log(u"RSS error: " + ex(e), logger.DEBUG)