SickRage/sickbeard/rssfeeds.py

from __future__ import with_statement

import os
import urllib

import sickbeard

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

from feedcache.cache import Cache

from sqliteshelf import SQLiteShelf


class RSSFeeds:
    def __init__(self, db_name='feeds'):
        try:
            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))

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

    def clearCache(self, age=None):
        try:
            Cache(self.rssDB).purge(age)
        finally:
            self.rssDB.close()

    def getFeed(self, url, post_data=None, request_headers=None, items=None):


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

        try:
            resp = Cache(self.rssDB).fetch(url, force_update=True, request_headers=request_headers)
        finally:
            self.rssDB.close()

        return resp