server.py

from library import Library
from client import Client
from media import Media
try:
    from pyplexlogger.logger import pyPlexLogger
except ImportError:
    pass
import base64

import urllib2
import xml.etree.ElementTree as ET
from xml.etree.ElementTree import XML

class Server(object):
    transcodeURL = '/video/:/transcode/segmented/start.m3u8?'
    transcode_private = base64.b64decode('k3U6GLkZOoNIoSgjDshPErvqMIFdE0xMTx8kgsrhnC0=')
    transcode_public = 'KQMIY6GATPC63AIMC4R2'

    def __init__(self, address, port=32400):
        try:
            self.l = pyPlexLogger('PlexAPI').logger
        except Exception:
            pass
        # TODO: clean up address, remove http:// etc
        
        # remove slash at end of address
        if address[-1] == '/':
            address = address[:-1]
        self.address = address
        self.port = int(port)

        
        
    def execute(self, path):
        self.log("execute %s" % path)
        if path[0] == '/':
            path = path[1:]
            
        # open url
        try: 
            urllib2.urlopen("http://%s:%d/%s" % (self.address, self.port, path)) 
        except urllib2.URLError, e:
            self.log("Reading %s failed. %s" % (path, e))

        
    def query(self, path):
        # print path
        self.log("Query %s" % path)
        if path[0] == '/':
            path = path[1:]
            
        # open url and get raw xml data
        try:
            response = urllib2.urlopen("http://%s:%d/%s" % (self.address, self.port, path))
        except urllib2.URLError, e:
            self.log("Reading %s failed. %s" % (path, e))
        
        # create element from xml data
        xmldata = response.read()
        self.log("Got response: %s" % xmldata)
        element = XML(xmldata)
        return element
    
    
    def __str__(self):
        return "<Server: %s:%d/>" % (self.address, self.port) 
    
    def __repr__(self):
        return "<Server: %s:%d/>" % (self.address, self.port) 
    
    def log(self, string):
        try:
            self.l.info(string)
        except Exception:
            print string
            pass

    def loadServer(self, elem):
        list = []
        ip = elem.attrib['host']
        if '.local' in ip:
            ip = elem.attrib['address']

        port = elem.attrib['port']

        
        return Server(ip, port)

    def getMedia(self, mediaPath):
        result = self.query(mediaPath)
        media = Media(result, self)
        return media.getMediaObject()
    
    @property
    def library(self):
        elem = self.query("/library")
        return Library(self)
    
    @property
    def clients(self):
        elem = self.query("/clients")
        clist = [Client(e, self) for e in elem]
        return clist

    @property
    def servers(self):
        elem = self.query("/servers")
        # get all other servers known by this server
        slist = [self.loadServer(e) for e in elem]
        # filter on None elements before returning
        return [f for f in slist if f != None]