Mirrors/python-plexapi
2
0
Fork 0
mirror of https://github.com/pkkid/python-plexapi synced 2025-02-16 12:58:26 +00:00
Code Issues Projects Releases Packages Wiki Activity

Allow custom headers for MyPlexPinLogin

Browse source
This commit is contained in:
JonnyWong16 2020-12-06 15:54:48 -08:00
parent 9adf29f2c6
commit 966ac5515b
No known key found for this signature in database
GPG key ID: B1F1F9807184697A
1 changed files with 20 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
plexapi/myplex.py
View file

@ -1112,10 +1112,11 @@ class MyPlexPinLogin(object):
LINK = 'https://plex.tv/api/v2/pins/link' # put
POLLINTERVAL = 1
def __init__(self, session=None, requestTimeout=None):
def __init__(self, session=None, requestTimeout=None, headers=None):
super(MyPlexPinLogin, self).__init__()
self._session = session or requests.Session()
self._requestTimeout = requestTimeout or TIMEOUT
self.headers = headers
self._loginTimeout = None
self._callback = None
@ -1197,17 +1198,21 @@ class MyPlexPinLogin(object):
return False
def link(self, code=None):
def link(self, code=None, token=None):
if code is None:
code = self.pin
url = self.LINK
headers = {
headers = BASE_HEADERS.copy()
headers.update({
'Content-Type': 'application/x-www-form-urlencoded',
'X-Plex-Product': 'Plex SSO',
'X-Plex-Client-Identifier': CONFIG.get('header.identifier'),
'X-Plex-Token': CONFIG.get('auth.server_token')
}
})
token = token or CONFIG.get('auth.server_token')
if token:
headers['X-Plex-Token'] = token
data = {'code': code}
self._query(url, self._session.put, headers=headers, data=data)
@ -1265,10 +1270,18 @@ class MyPlexPinLogin(object):
finally:
self.finished = True
def _headers(self, **kwargs):
""" Returns dict containing base headers for all requests for pin login. """
headers = BASE_HEADERS.copy()
if self.headers:
headers.update(self.headers)
headers.update(kwargs)
return headers
def _query(self, url, method=None, headers=None, **kwargs):
method = method or self._session.get
log.debug('%s %s', method.__name__.upper(), url)
headers = headers or BASE_HEADERS.copy()
headers = headers or self._headers()
response = method(url, headers=headers, timeout=self._requestTimeout, **kwargs)
if not response.ok: # pragma: no cover
codename = codes.get(response.status_code)[0]