mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-01 00:43:37 +00:00
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
# coding=utf-8
|
|
#
|
|
# This file is part of SickGear.
|
|
#
|
|
|
|
from feedparser import feedparser
|
|
|
|
from sickbeard import helpers, logger
|
|
from sickbeard.exceptions import ex
|
|
|
|
|
|
class RSSFeeds:
|
|
|
|
def __init__(self, provider=None):
|
|
|
|
self.provider = provider
|
|
self.response = None
|
|
|
|
def _check_auth_cookie(self):
|
|
|
|
if self.provider:
|
|
return self.provider.check_auth_cookie()
|
|
return True
|
|
|
|
# noinspection PyUnusedLocal
|
|
def cb_response(self, r, *args, **kwargs):
|
|
self.response = dict(url=r.url, elapsed=r.elapsed, from_cache=r.from_cache)
|
|
return r
|
|
|
|
def get_feed(self, url, request_headers=None, **kwargs):
|
|
|
|
if not self._check_auth_cookie():
|
|
return
|
|
|
|
session = None
|
|
if self.provider and hasattr(self.provider, 'session'):
|
|
session = self.provider.session
|
|
|
|
response = helpers.getURL(url, headers=request_headers, session=session,
|
|
hooks=dict(response=self.cb_response), **kwargs)
|
|
if not response:
|
|
return
|
|
|
|
try:
|
|
feed = feedparser.parse(response)
|
|
feed['rq_response'] = self.response
|
|
if feed and 'entries' in feed:
|
|
return feed
|
|
|
|
if feed and '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)
|