@ -3,33 +3,144 @@ import certifi
import json
import sickbeard
import time
import datetime
from sickbeard import logger
from exceptions import traktException , traktAuthException # , traktServerBusy
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 , ssl_verify = True , timeout = None ) :
def __init__ ( self , timeout = None ) :
self . session = requests . Session ( )
self . verify = ssl_verify and sickbeard . TRAKT_VERIFY and certifi . where ( )
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
}
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
def trakt_token ( self , trakt_pin = None , refresh = False , count = 0 ) :
if 3 < = count :
sickbeard . TRAKT_ACCESS_TOKEN = ' '
@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
elif 0 < count :
time . sleep ( 3 )
@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 ,
@ -38,62 +149,61 @@ class TraktAPI:
}
if refresh :
data [ ' grant_type ' ] = ' refresh_token '
data [ ' refresh_token ' ] = sickbeard . TRAKT_REFRESH_TOKEN
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 '
}
headers = { ' Content-Type ' : ' application/json ' }
resp = self . trakt_request ( ' oauth/token ' , data = data , headers = headers , url = self . auth_url , method = ' POST ' , count = count )
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 :
sickbeard . TRAKT_TOKEN = resp [ ' access_token ' ]
if ' refresh_token ' in resp :
sickbeard . TRAKT_REFRESH_TOKEN = resp [ ' refresh_token ' ]
return True
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 validate_account ( self ) :
resp = self . trakt_request ( ' users/settings ' )
return ' account ' in resp
def get_connected_user ( self ) :
def trakt_request ( self , path , data = None , headers = None , url = None , count = 0 , sleep_retry = 60 , send_oauth = None , * * kwargs ) :
if sickbeard . TRAKT_TOKEN :
response = ' Connected to Trakt user account: %s '
if sickbeard . TRAKT_CONNECTED_ACCOUNT and sickbeard . TRAKT_TOKEN == sickbeard . TRAKT_CONNECTED_ACCOUNT [ 1 ] and sickbeard . TRAKT_CONNECTED_ACCOUNT [ 0 ] :
return response % sickbeard . TRAKT_CONNECTED_ACCOUNT [ 0 ]
resp = self . trakt_request ( ' users/settings ' )
if ' user ' in resp :
sickbeard . TRAKT_CONNECTED_ACCOUNT = [ resp [ ' user ' ] [ ' username ' ] , sickbeard . TRAKT_TOKEN ]
return response % sickbeard . TRAKT_CONNECTED_ACCOUNT [ 0 ]
return ' Not connected to Trakt '
def trakt_request ( self , path , data = None , headers = None , url = None , method = ' GET ' , count = 0 ) :
if None is sickbeard . TRAKT_TOKEN :
logger . log ( u ' You must get a Trakt token. Check your Trakt settings ' , logger . WARNING )
count + = 1
if count > self . max_retrys :
return { }
# wait before retry
count > 1 and time . sleep ( sleep_retry )
headers = headers or self . headers
url = url or self . api_url
count + = 1
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 { }
headers [ ' Authorization ' ] = ' Bearer ' + sickbeard . TRAKT_TOKEN
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 ( method , url + path , headers = headers , timeout = self . timeout ,
data = json . dumps ( data ) if data else [ ] , verify = self . verify )
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 ( )
@ -104,25 +214,37 @@ class TraktAPI:
code = getattr ( e . response , ' status_code ' , None )
if not code :
if ' timed out ' in e :
logger . log ( u ' Timeout connecting to Trakt. Try to increase timeout value in Trakt settings ' , logger . WARNING )
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 , method , count = count )
elif 401 == code :
if self . trakt_token ( refresh = True , count = count ) :
sickbeard . save_config ( )
return self . trakt_request ( path , data , headers , url , method , count = count )
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 :
logger . log ( u ' Unauthorized. Please check your Trakt settings ' , logger . WARNING )
raise traktAuthException ( )
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. Try again later please ' , logger . WARNING )
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 :
@ -132,10 +254,12 @@ class TraktAPI:
# check and confirm Trakt call did not fail
if isinstance ( resp , dict ) and ' failure ' == resp . get ( ' status ' , None ) :
if ' message ' in resp :
raise t raktException( resp [ ' message ' ] )
raise T raktException( resp [ ' message ' ] )
if ' error ' in resp :
raise t raktException( resp [ ' error ' ] )
raise T raktException( resp [ ' error ' ] )
else :
raise t raktException( ' Unknown Error ' )
raise T raktException( ' Unknown Error ' )
if send_oauth and send_oauth in sickbeard . TRAKT_ACCOUNTS :
sickbeard . TRAKT_ACCOUNTS [ send_oauth ] . reset_auth_failure ( )
return resp
Prinz23
10 years ago
JackDandy
