mirror of
https://github.com/pkkid/python-plexapi
synced 2024-09-20 13:51:57 +00:00
116 lines
4.1 KiB
Python
116 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
PlexServer
|
|
"""
|
|
import requests
|
|
from requests.status_codes import _codes as codes
|
|
from plexapi import BASE_HEADERS, TIMEOUT
|
|
from plexapi import log, utils
|
|
from plexapi import audio, video, playlist # noqa; required
|
|
from plexapi.compat import quote
|
|
from plexapi.client import Client
|
|
from plexapi.exceptions import BadRequest, NotFound
|
|
from plexapi.library import Library
|
|
from plexapi.myplex import MyPlexAccount
|
|
from plexapi.playqueue import PlayQueue
|
|
from xml.etree import ElementTree
|
|
|
|
TOTAL_QUERIES = 0
|
|
DEFAULT_BASEURI = 'http://localhost:32400'
|
|
|
|
|
|
class PlexServer(object):
|
|
|
|
def __init__(self, baseuri=None, token=None, session=None):
|
|
self.baseuri = baseuri or DEFAULT_BASEURI
|
|
self.token = token
|
|
self.session = session or requests.Session()
|
|
data = self._connect()
|
|
self.friendlyName = data.attrib.get('friendlyName')
|
|
self.machineIdentifier = data.attrib.get('machineIdentifier')
|
|
self.myPlex = bool(data.attrib.get('myPlex'))
|
|
self.myPlexMappingState = data.attrib.get('myPlexMappingState')
|
|
self.myPlexSigninState = data.attrib.get('myPlexSigninState')
|
|
self.myPlexSubscription = data.attrib.get('myPlexSubscription')
|
|
self.myPlexUsername = data.attrib.get('myPlexUsername')
|
|
self.platform = data.attrib.get('platform')
|
|
self.platformVersion = data.attrib.get('platformVersion')
|
|
self.transcoderActiveVideoSessions = int(data.attrib.get('transcoderActiveVideoSessions', 0))
|
|
self.updatedAt = int(data.attrib.get('updatedAt', 0))
|
|
self.version = data.attrib.get('version')
|
|
|
|
def __repr__(self):
|
|
return '<%s:%s>' % (self.__class__.__name__, self.baseuri)
|
|
|
|
def _connect(self):
|
|
try:
|
|
return self.query('/')
|
|
except Exception as err:
|
|
log.error('%s: %s', self.baseuri, err)
|
|
raise NotFound('No server found at: %s' % self.baseuri)
|
|
|
|
@property
|
|
def library(self):
|
|
return Library(self, self.query('/library/'))
|
|
|
|
def account(self):
|
|
data = self.query('/myplex/account')
|
|
return MyPlexAccount(self, data)
|
|
|
|
def clients(self):
|
|
items = []
|
|
for elem in self.query('/clients'):
|
|
items.append(Client(self, elem))
|
|
return items
|
|
|
|
def client(self, name):
|
|
for elem in self.query('/clients'):
|
|
if elem.attrib.get('name').lower() == name.lower():
|
|
return Client(self, elem)
|
|
raise NotFound('Unknown client name: %s' % name)
|
|
|
|
def createPlayQueue(self, item):
|
|
return PlayQueue.create(self, item)
|
|
|
|
def headers(self):
|
|
headers = BASE_HEADERS
|
|
if self.token:
|
|
headers['X-Plex-Token'] = self.token
|
|
return headers
|
|
|
|
def playlists(self):
|
|
return utils.listItems(self, '/playlists')
|
|
|
|
def playlist(self, title=None): # noqa
|
|
for item in self.playlists():
|
|
if item.title == title:
|
|
return item
|
|
raise NotFound('Invalid playlist title: %s' % title)
|
|
|
|
def query(self, path, method=None, **kwargs):
|
|
global TOTAL_QUERIES
|
|
TOTAL_QUERIES += 1
|
|
url = self.url(path)
|
|
method = method or self.session.get
|
|
log.info('%s %s', method.__name__.upper(), url)
|
|
response = method(url, headers=self.headers(), timeout=TIMEOUT, **kwargs)
|
|
if response.status_code not in [200, 201]:
|
|
codename = codes.get(response.status_code)[0]
|
|
raise BadRequest('(%s) %s' % (response.status_code, codename))
|
|
data = response.text.encode('utf8')
|
|
return ElementTree.fromstring(data) if data else None
|
|
|
|
def search(self, query, mediatype=None):
|
|
items = utils.listItems(self, '/search?query=%s' % quote(query))
|
|
if mediatype:
|
|
return [item for item in items if item.type == mediatype]
|
|
return items
|
|
|
|
def sessions(self):
|
|
return utils.listItems(self, '/status/sessions')
|
|
|
|
def url(self, path):
|
|
if self.token:
|
|
delim = '&' if '?' in path else '?'
|
|
return '%s%s%sX-Plex-Token=%s' % (self.baseuri, path, delim, self.token)
|
|
return '%s%s' % (self.baseuri, path)
|