mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-18 07:15:13 -05:00
77ea8117c6
A recent update/change in RSS cache removed the cookies for custom RSS provider. This re-adds it.
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
import os
|
|
import threading
|
|
import urllib
|
|
import urlparse
|
|
import re
|
|
import time
|
|
import sickbeard
|
|
|
|
from sickbeard import logger
|
|
from sickbeard import encodingKludge as ek
|
|
from sickbeard.exceptions import ex
|
|
from lib.shove import Shove
|
|
from lib.feedcache import cache
|
|
|
|
feed_lock = threading.Lock()
|
|
|
|
class RSSFeeds:
|
|
def __init__(self, db_name):
|
|
try:
|
|
self.fs = self.fs = Shove('sqlite:///' + ek.ek(os.path.join, sickbeard.CACHE_DIR, db_name + '.db'), compress=True)
|
|
self.fc = cache.Cache(self.fs)
|
|
except Exception, e:
|
|
logger.log(u"RSS error: " + ex(e), logger.ERROR)
|
|
raise
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, type, value, tb):
|
|
self.fc = None
|
|
self.fs.close()
|
|
|
|
def clearCache(self, age=None):
|
|
if not self.fc:
|
|
return
|
|
|
|
self.fc.purge(age)
|
|
|
|
def getRSSFeed(self, url, post_data=None, request_headers=None):
|
|
if not self.fc:
|
|
return
|
|
|
|
with feed_lock:
|
|
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)
|
|
|
|
feed = self.fc.fetch(url, False, False, request_headers)
|
|
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
|