PlexAccount.refresh() - Code Metrics - Inspection of "Merge branch 'release/1.0.2.0'" - fuzeman/Plex-Trakt-Scrobbler - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Push — master ( aeb165...d9ae97 )

by Dean

created 2016-04-15 10:51 UTC

PlexAccount.refresh() F

↳ Parent: Project

Complexity

Conditions

Size

Total Lines

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	1
CRAP Score	77.2264

Metric	Value
dl	0
loc	32
ccs	1
cts	18
cp	0.0556
rs	3
cc	9
crap	77.2264

from plugin.core.exceptions import AccountAuthenticationError
from plugin.models.core import db
from plugin.models.account import Account

from datetime import datetime, timedelta
from playhouse.apsw_ext import *
from plex import Plex
from urllib import urlencode
from urlparse import urlparse, parse_qsl
from xml.etree import ElementTree
import logging
import requests

REFRESH_INTERVAL = timedelta(days=1)

log = logging.getLogger(__name__)


class PlexAccount(Model):
    class Meta:
        database = db
        db_table = 'plex.account'

    account = ForeignKeyField(Account, 'plex_accounts', unique=True)

    key = IntegerField(null=True, unique=True)
    username = CharField(null=True, unique=True)

    title = CharField(null=True)
    thumb = TextField(null=True)

    refreshed_at = DateTimeField(null=True)

    def __init__(self, *args, **kwargs):
        super(PlexAccount, self).__init__(*args, **kwargs)

        self._basic_credential = None

    @property
    def account_id(self):
        return self._data.get('account')

    @property
    def basic(self):
        if self._basic_credential:
            return self._basic_credential

        return self.basic_credentials.first()

    @basic.setter
    def basic(self, value):
        self._basic_credential = value

    def authorization(self):
        # Basic
        basic = self.basic

        if basic:
            return self.basic_authorization(basic)

        # No account authorization available
        raise AccountAuthenticationError("Plex account hasn't been authenticated")

    def basic_authorization(self, basic_credential=None):
        if basic_credential is None:
            basic_credential = self.basic

        # Ensure token exists
        if basic_credential.token_server is None:
            raise AccountAuthenticationError("Plex account is missing the server token")

        log.debug('Using basic authorization for %r', self)

        # Return authorization context
        return Plex.configuration.authentication(basic_credential.token_server)

    def refresh(self, force=False, save=True):
        # Retrieve credentials
        basic = self.basic

        if not basic:
            return False

        # Check if refresh is required
        if self.refresh_required(basic):
            force = True

        # Only refresh account every `REFRESH_INTERVAL`
        if not force and self.refreshed_at:
            since_refresh = datetime.utcnow() - self.refreshed_at

            if since_refresh < REFRESH_INTERVAL:
                return False

        # Refresh account details
        if not self.refresh_details(basic):
            return False

        if not basic.refresh():
            return False

        # Store changes in database
        self.refreshed_at = datetime.utcnow()

        if save:
            self.save()

        return True

    def refresh_details(self, basic):
        if basic.token_plex == 'anonymous':
            return self.refresh_anonymous()

        # Fetch account details
        response = requests.get('https://plex.tv/users/account', headers={
            'X-Plex-Token': basic.token_plex
        })

        if not (200 <= response.status_code < 300):
            # Invalid response
            return False

        user = ElementTree.fromstring(response.content)

        # Update details
        self.username = user.attrib.get('username') or None

        self.title = user.attrib.get('title')
        self.thumb = user.attrib.get('thumb')

        # Update `key`
        if self.id == 1:
            # Use administrator `key`
            self.key = 1
        else:
            # Retrieve user id from plex.tv details
            try:
                user_id = int(user.attrib.get('id'))
            except Exception, ex:
                log.warn('Unable to cast user id to integer: %s', ex, exc_info=True)
                user_id = None

            # Update `key`
            self.key = user_id

        return True

    def refresh_anonymous(self):
        log.debug('Refreshing anonymous plex account')

        self.username = 'administrator'

        self.title = 'Administrator'
        self.thumb = None

        if self.id == 1:
            self.key = 1
        else:
            self.key = None

        return True

    def refresh_required(self, basic):
        if self.key is None:
            return True

        if self.title is None:
            return True

        if basic.token_server is None:
            return True

        return False

    def thumb_url(self, default=None, rating='pg', size=256):
        if not self.thumb:
            return None

        thumb = urlparse(self.thumb)

        if thumb.netloc.endswith('plex.tv'):
            return self.thumb

        if not thumb.netloc.endswith('gravatar.com'):
            return None

        result = 'https://secure.gravatar.com%s' % thumb.path

        if default is None:
            query = dict(parse_qsl(thumb.query))

            default = query.get('d') or query.get('default')

        return result + '?' + urlencode({
            'd': default,
            'r': rating,
            's': size
        })

    def to_json(self, full=False):
        result = {
            'id': self.id,
            'username': self.username,

            'title': self.title,
            'thumb_url': self.thumb_url()
        }

        if not full:
            return result

        # Merge authorization details
        result['authorization'] = {
            'basic': {'state': 'empty'}
        }

        # - Basic credentials
        basic = self.basic

        if basic is not None:
            result['authorization']['basic'] = basic.to_json(self)

        return result

    def __repr__(self):
        return '<PlexAccount username: %r>' % (
            self.username,
        )


1	1	from plugin.core.exceptions import AccountAuthenticationError
2	1	from plugin.models.core import db
3	1	from plugin.models.account import Account
4
5	1	from datetime import datetime, timedelta
6	1	from playhouse.apsw_ext import *
7	1	from plex import Plex
8	1	from urllib import urlencode
9	1	from urlparse import urlparse, parse_qsl
10	1	from xml.etree import ElementTree
11	1	import logging
12	1	import requests
13
14	1	REFRESH_INTERVAL = timedelta(days=1)
15
16	1	log = logging.getLogger(__name__)
17
18
19	1	class PlexAccount(Model):
20	1	class Meta:
21	1	database = db
22	1	db_table = 'plex.account'
23
24	1	account = ForeignKeyField(Account, 'plex_accounts', unique=True)
25
26	1	key = IntegerField(null=True, unique=True)
27	1	username = CharField(null=True, unique=True)
28
29	1	title = CharField(null=True)
30	1	thumb = TextField(null=True)
31
32	1	refreshed_at = DateTimeField(null=True)
33
34	1	def __init__(self, args, *kwargs):
35		super(PlexAccount, self).__init__(args, *kwargs)
36
37		self._basic_credential = None
38
39	1	@property
40		def account_id(self):
41		return self._data.get('account')
42
43	1	@property
44		def basic(self):
45		if self._basic_credential:
46		return self._basic_credential
47
48		return self.basic_credentials.first()
49
50	1	@basic.setter
51		def basic(self, value):
52		self._basic_credential = value
53
54	1	def authorization(self):
55		# Basic
56		basic = self.basic
57
58		if basic:
59		return self.basic_authorization(basic)
60
61		# No account authorization available
62		raise AccountAuthenticationError("Plex account hasn't been authenticated")
63
64	1	def basic_authorization(self, basic_credential=None):
65		if basic_credential is None:
66		basic_credential = self.basic
67
68		# Ensure token exists
69		if basic_credential.token_server is None:
70		raise AccountAuthenticationError("Plex account is missing the server token")
71
72		log.debug('Using basic authorization for %r', self)
73
74		# Return authorization context
75		return Plex.configuration.authentication(basic_credential.token_server)
76
77	1	def refresh(self, force=False, save=True):
78		# Retrieve credentials
79		basic = self.basic
80
81		if not basic:
82		return False
83
84		# Check if refresh is required
85		if self.refresh_required(basic):
86		force = True
87
88		# Only refresh account every `REFRESH_INTERVAL`
89		if not force and self.refreshed_at:
90		since_refresh = datetime.utcnow() - self.refreshed_at
91
92		if since_refresh < REFRESH_INTERVAL:
93		return False
94
95		# Refresh account details
96		if not self.refresh_details(basic):
97		return False
98
99		if not basic.refresh():
100		return False
101
102		# Store changes in database
103		self.refreshed_at = datetime.utcnow()
104
105		if save:
106		self.save()
107
108		return True
109
110	1	def refresh_details(self, basic):
111		if basic.token_plex == 'anonymous':
112		return self.refresh_anonymous()
113
114		# Fetch account details
115		response = requests.get('https://plex.tv/users/account', headers={
116		'X-Plex-Token': basic.token_plex
117		})
118
119		if not (200 <= response.status_code < 300):
120		# Invalid response
121		return False
122
123		user = ElementTree.fromstring(response.content)
124
125		# Update details
126		self.username = user.attrib.get('username') or None
127
128		self.title = user.attrib.get('title')
129		self.thumb = user.attrib.get('thumb')
130
131		# Update `key`
132		if self.id == 1:
133		# Use administrator `key`
134		self.key = 1
135		else:
136		# Retrieve user id from plex.tv details
137		try:
138		user_id = int(user.attrib.get('id'))
139		except Exception, ex:
140		log.warn('Unable to cast user id to integer: %s', ex, exc_info=True)
141		user_id = None
142
143		# Update `key`
144		self.key = user_id
145
146		return True
147
148	1	def refresh_anonymous(self):
149		log.debug('Refreshing anonymous plex account')
150
151		self.username = 'administrator'
152
153		self.title = 'Administrator'
154		self.thumb = None
155
156		if self.id == 1:
157		self.key = 1
158		else:
159		self.key = None
160
161		return True
162
163	1	def refresh_required(self, basic):
164		if self.key is None:
165		return True
166
167		if self.title is None:
168		return True
169
170		if basic.token_server is None:
171		return True
172
173		return False
174
175	1	def thumb_url(self, default=None, rating='pg', size=256):
176		if not self.thumb:
177		return None
178
179		thumb = urlparse(self.thumb)
180
181		if thumb.netloc.endswith('plex.tv'):
182		return self.thumb
183
184		if not thumb.netloc.endswith('gravatar.com'):
185		return None
186
187		result = 'https://secure.gravatar.com%s' % thumb.path
188
189		if default is None:
190		query = dict(parse_qsl(thumb.query))
191
192		default = query.get('d') or query.get('default')
193
194		return result + '?' + urlencode({
195		'd': default,
196		'r': rating,
197		's': size
198		})
199
200	1	def to_json(self, full=False):
201		result = {
202		'id': self.id,
203		'username': self.username,
204
205		'title': self.title,
206		'thumb_url': self.thumb_url()
207		}
208
209		if not full:
210		return result
211
212		# Merge authorization details
213		result['authorization'] = {
214		'basic': {'state': 'empty'}
215		}
216
217		# - Basic credentials
218		basic = self.basic
219
220		if basic is not None:
221		result['authorization']['basic'] = basic.to_json(self)
222
223		return result
224
225	1	def __repr__(self):
226		return '<PlexAccount username: %r>' % (
227		self.username,
228		)
229

fuzeman / Plex-Trakt-Scrobbler

Push — master ( aeb165...d9ae97 )

PlexAccount.refresh() F

Complexity

Size

Duplication

Code Coverage

Duplication Side-by-Side

Filter issues like