mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-01 00:43:37 +00:00
b3be940d44
Add setting to Trakt notification to update collection with downloaded episode info. Add Most Watched, Collected during the last month on Trakt. Change Add from Trakt/"Shows:" with Anticipated, Popular views. Change improve robustness of Trakt communications. Change Trakt notifier logo. Change pep8 and cleanup.
265 lines
11 KiB
Python
265 lines
11 KiB
Python
import requests
|
|
import certifi
|
|
import json
|
|
import sickbeard
|
|
import time
|
|
import datetime
|
|
from sickbeard import logger
|
|
|
|
from exceptions import TraktException, TraktAuthException # , TraktServerBusy
|
|
|
|
|
|
class TraktAccount:
|
|
max_auth_fail = 9
|
|
|
|
def __init__(self, account_id=None, token='', refresh_token='', auth_fail=0, last_fail=None, token_valid_date=None):
|
|
self.account_id = account_id
|
|
self._name = ''
|
|
self.token = token
|
|
self.refresh_token = refresh_token
|
|
self.auth_fail = auth_fail
|
|
self.last_fail = last_fail
|
|
self.token_valid_date = token_valid_date
|
|
|
|
@property
|
|
def name(self):
|
|
if self.token and self.active:
|
|
if not self._name:
|
|
try:
|
|
resp = TraktAPI().trakt_request('users/settings', send_oauth=self.account_id, sleep_retry=20)
|
|
self.reset_auth_failure()
|
|
if 'user' in resp:
|
|
self._name = resp['user']['username']
|
|
except TraktAuthException:
|
|
self.inc_auth_failure()
|
|
self._name = ''
|
|
except TraktException:
|
|
pass
|
|
else:
|
|
self._name = ''
|
|
|
|
return self._name
|
|
|
|
def reset_name(self):
|
|
self._name = ''
|
|
|
|
@property
|
|
def active(self):
|
|
return self.auth_fail < self.max_auth_fail and self.token
|
|
|
|
@property
|
|
def needs_refresh(self):
|
|
return not self.token_valid_date or self.token_valid_date - datetime.datetime.now() < datetime.timedelta(days=3)
|
|
|
|
@property
|
|
def token_expired(self):
|
|
return self.token_valid_date and datetime.datetime.now() > self.token_valid_date
|
|
|
|
def reset_auth_failure(self):
|
|
if self.auth_fail != 0:
|
|
self.auth_fail = 0
|
|
self.last_fail = None
|
|
|
|
def inc_auth_failure(self):
|
|
self.auth_fail += 1
|
|
self.last_fail = datetime.datetime.now()
|
|
|
|
def auth_failure(self):
|
|
if self.auth_fail < self.max_auth_fail:
|
|
if self.last_fail:
|
|
time_diff = datetime.datetime.now() - self.last_fail
|
|
if self.auth_fail % 3 == 0:
|
|
if time_diff > datetime.timedelta(days=1):
|
|
self.inc_auth_failure()
|
|
sickbeard.save_config()
|
|
elif time_diff > datetime.timedelta(minutes=15):
|
|
self.inc_auth_failure()
|
|
if self.auth_fail == self.max_auth_fail or time_diff > datetime.timedelta(hours=6):
|
|
sickbeard.save_config()
|
|
else:
|
|
self.inc_auth_failure()
|
|
|
|
|
|
class TraktAPI:
|
|
max_retrys = 3
|
|
|
|
def __init__(self, timeout=None):
|
|
|
|
self.session = requests.Session()
|
|
self.verify = sickbeard.TRAKT_VERIFY and certifi.where()
|
|
self.timeout = timeout or sickbeard.TRAKT_TIMEOUT
|
|
self.auth_url = sickbeard.TRAKT_BASE_URL
|
|
self.api_url = sickbeard.TRAKT_BASE_URL
|
|
self.headers = {'Content-Type': 'application/json',
|
|
'trakt-api-version': '2',
|
|
'trakt-api-key': sickbeard.TRAKT_CLIENT_ID}
|
|
|
|
@staticmethod
|
|
def build_config_string(data):
|
|
return '!!!'.join('%s|%s|%s|%s|%s|%s' % (
|
|
value.account_id, value.token, value.refresh_token, value.auth_fail,
|
|
value.last_fail.strftime('%Y%m%d%H%M') if value.last_fail else '0',
|
|
value.token_valid_date.strftime('%Y%m%d%H%M%S') if value.token_valid_date else '0') for (key, value) in data.items())
|
|
|
|
@staticmethod
|
|
def read_config_string(data):
|
|
return dict((int(a.split('|')[0]), TraktAccount(
|
|
int(a.split('|')[0]), a.split('|')[1], a.split('|')[2], int(a.split('|')[3]),
|
|
datetime.datetime.strptime(a.split('|')[4], '%Y%m%d%H%M') if a.split('|')[4] != '0' else None,
|
|
datetime.datetime.strptime(a.split('|')[5], '%Y%m%d%H%M%S') if a.split('|')[5] != '0' else None)) for a in data.split('!!!') if data)
|
|
|
|
@staticmethod
|
|
def add_account(token, refresh_token, token_valid_date):
|
|
k = max(sickbeard.TRAKT_ACCOUNTS.keys() or [0]) + 1
|
|
sickbeard.TRAKT_ACCOUNTS[k] = TraktAccount(account_id=k, token=token, refresh_token=refresh_token, token_valid_date=token_valid_date)
|
|
sickbeard.save_config()
|
|
return k
|
|
|
|
@staticmethod
|
|
def replace_account(account, token, refresh_token, token_valid_date, refresh):
|
|
if account in sickbeard.TRAKT_ACCOUNTS:
|
|
sickbeard.TRAKT_ACCOUNTS[account].token = token
|
|
sickbeard.TRAKT_ACCOUNTS[account].refresh_token = refresh_token
|
|
sickbeard.TRAKT_ACCOUNTS[account].token_valid_date = token_valid_date
|
|
if not refresh:
|
|
sickbeard.TRAKT_ACCOUNTS[account].reset_name()
|
|
sickbeard.TRAKT_ACCOUNTS[account].reset_auth_failure()
|
|
sickbeard.save_config()
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
@staticmethod
|
|
def delete_account(account):
|
|
if account in sickbeard.TRAKT_ACCOUNTS:
|
|
sickbeard.TRAKT_ACCOUNTS.pop(account)
|
|
sickbeard.save_config()
|
|
return True
|
|
return False
|
|
|
|
def trakt_token(self, trakt_pin=None, refresh=False, count=0, account=None):
|
|
if self.max_retrys <= count:
|
|
return False
|
|
0 < count and time.sleep(3)
|
|
|
|
data = {
|
|
'client_id': sickbeard.TRAKT_CLIENT_ID,
|
|
'client_secret': sickbeard.TRAKT_CLIENT_SECRET,
|
|
'redirect_uri': 'urn:ietf:wg:oauth:2.0:oob'
|
|
}
|
|
|
|
if refresh:
|
|
if account and account in sickbeard.TRAKT_ACCOUNTS:
|
|
data['grant_type'] = 'refresh_token'
|
|
data['refresh_token'] = sickbeard.TRAKT_ACCOUNTS[account].refresh_token
|
|
else:
|
|
return False
|
|
else:
|
|
data['grant_type'] = 'authorization_code'
|
|
if trakt_pin:
|
|
data['code'] = trakt_pin
|
|
|
|
headers = {'Content-Type': 'application/json'}
|
|
|
|
try:
|
|
now = datetime.datetime.now()
|
|
resp = self.trakt_request('oauth/token', data=data, headers=headers, url=self.auth_url,
|
|
count=count, sleep_retry=0)
|
|
except (TraktAuthException, TraktException):
|
|
return False
|
|
|
|
if 'access_token' in resp and 'refresh_token' in resp and 'expires_in' in resp:
|
|
token_valid_date = now + datetime.timedelta(seconds=sickbeard.helpers.tryInt(resp['expires_in']))
|
|
if refresh or (not refresh and account and account in sickbeard.TRAKT_ACCOUNTS):
|
|
return self.replace_account(account, resp['access_token'], resp['refresh_token'], token_valid_date, refresh)
|
|
else:
|
|
return self.add_account(resp['access_token'], resp['refresh_token'], token_valid_date)
|
|
return False
|
|
|
|
def trakt_request(self, path, data=None, headers=None, url=None, count=0, sleep_retry=60, send_oauth=None, **kwargs):
|
|
|
|
count += 1
|
|
if count > self.max_retrys:
|
|
return {}
|
|
|
|
# wait before retry
|
|
count > 1 and time.sleep(sleep_retry)
|
|
|
|
headers = headers or self.headers
|
|
if send_oauth and send_oauth in sickbeard.TRAKT_ACCOUNTS:
|
|
if sickbeard.TRAKT_ACCOUNTS[send_oauth].active:
|
|
if sickbeard.TRAKT_ACCOUNTS[send_oauth].needs_refresh:
|
|
self.trakt_token(refresh=True, count=0, account=send_oauth)
|
|
if sickbeard.TRAKT_ACCOUNTS[send_oauth].token_expired:
|
|
return {}
|
|
headers['Authorization'] = 'Bearer %s' % sickbeard.TRAKT_ACCOUNTS[send_oauth].token
|
|
else:
|
|
return {}
|
|
|
|
kwargs = dict(headers=headers, timeout=self.timeout, verify=self.verify)
|
|
if data:
|
|
kwargs['data'] = json.dumps(data)
|
|
|
|
url = url or self.api_url
|
|
try:
|
|
resp = self.session.request(('GET', 'POST')['data' in kwargs.keys()],
|
|
url + path, **kwargs)
|
|
|
|
# check for http errors and raise if any are present
|
|
resp.raise_for_status()
|
|
|
|
# convert response to json
|
|
resp = resp.json()
|
|
except requests.RequestException as e:
|
|
code = getattr(e.response, 'status_code', None)
|
|
if not code:
|
|
if 'timed out' in e:
|
|
logger.log(u'Timeout connecting to Trakt', logger.WARNING)
|
|
# This is pretty much a fatal error if there is no status_code
|
|
# It means there basically was no response at all
|
|
else:
|
|
logger.log(u'Could not connect to Trakt. Error: {0}'.format(e), logger.WARNING)
|
|
elif 502 == code:
|
|
# Retry the request, Cloudflare had a proxying issue
|
|
logger.log(u'Retrying Trakt api request: %s' % path, logger.WARNING)
|
|
return self.trakt_request(path, data, headers, url, count=count, sleep_retry=sleep_retry, send_oauth=send_oauth)
|
|
elif 401 == code and path != 'oauth/token':
|
|
if send_oauth:
|
|
if sickbeard.TRAKT_ACCOUNTS[send_oauth].needs_refresh:
|
|
if self.trakt_token(refresh=True, count=count, account=send_oauth):
|
|
return self.trakt_request(path, data, headers, url, count=count, sleep_retry=sleep_retry, send_oauth=send_oauth)
|
|
else:
|
|
logger.log(u'Unauthorized. Please check your Trakt settings', logger.WARNING)
|
|
sickbeard.TRAKT_ACCOUNTS[send_oauth].auth_failure()
|
|
raise TraktAuthException()
|
|
else:
|
|
# sometimes the trakt server sends invalid token error even if it isn't
|
|
sickbeard.TRAKT_ACCOUNTS[send_oauth].auth_failure()
|
|
if count >= self.max_retrys:
|
|
raise TraktAuthException()
|
|
else:
|
|
return self.trakt_request(path, data, headers, url, count=count, sleep_retry=sleep_retry, send_oauth=send_oauth)
|
|
else:
|
|
raise TraktAuthException()
|
|
elif code in (500, 501, 503, 504, 520, 521, 522):
|
|
# http://docs.trakt.apiary.io/#introduction/status-codes
|
|
logger.log(u'Trakt may have some issues and it\'s unavailable. Trying again', logger.WARNING)
|
|
self.trakt_request(path, data, headers, url, count=count, sleep_retry=sleep_retry, send_oauth=send_oauth)
|
|
elif 404 == code:
|
|
logger.log(u'Trakt error (404) the resource does not exist: %s' % url + path, logger.WARNING)
|
|
else:
|
|
logger.log(u'Could not connect to Trakt. Code error: {0}'.format(code), logger.ERROR)
|
|
return {}
|
|
|
|
# check and confirm Trakt call did not fail
|
|
if isinstance(resp, dict) and 'failure' == resp.get('status', None):
|
|
if 'message' in resp:
|
|
raise TraktException(resp['message'])
|
|
if 'error' in resp:
|
|
raise TraktException(resp['error'])
|
|
else:
|
|
raise TraktException('Unknown Error')
|
|
|
|
if send_oauth and send_oauth in sickbeard.TRAKT_ACCOUNTS:
|
|
sickbeard.TRAKT_ACCOUNTS[send_oauth].reset_auth_failure()
|
|
return resp
|