124 lines
3.1 KiB
Python
124 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
#################################################################################################
|
|
|
|
import logging
|
|
|
|
from client import JellyfinClient
|
|
from helpers import has_attribute
|
|
|
|
#################################################################################################
|
|
|
|
class NullHandler(logging.Handler):
|
|
def emit(self, record):
|
|
print(self.format(record))
|
|
|
|
loghandler = NullHandler
|
|
LOG = logging.getLogger('Jellyfin')
|
|
|
|
#################################################################################################
|
|
|
|
def config(level=logging.INFO):
|
|
|
|
logger = logging.getLogger('Jellyfin')
|
|
logger.addHandler(Jellyfin.loghandler())
|
|
logger.setLevel(level)
|
|
|
|
def ensure_client():
|
|
|
|
def decorator(func):
|
|
def wrapper(self, *args, **kwargs):
|
|
|
|
if self.client.get(self.server_id) is None:
|
|
self.construct()
|
|
|
|
return func(self, *args, **kwargs)
|
|
|
|
return wrapper
|
|
return decorator
|
|
|
|
|
|
class Jellyfin(object):
|
|
|
|
''' This is your Jellyfinclient, you can create more than one. The server_id is only a temporary thing
|
|
to communicate with the JellyfinClient().
|
|
|
|
from jellyfin import Jellyfin
|
|
|
|
Jellyfin('123456').config.data['app']
|
|
|
|
# Permanent client reference
|
|
client = Jellyfin('123456').get_client()
|
|
client.config.data['app']
|
|
'''
|
|
|
|
# Borg - multiple instances, shared state
|
|
_shared_state = {}
|
|
client = {}
|
|
server_id = "default"
|
|
loghandler = loghandler
|
|
|
|
def __init__(self, server_id=None):
|
|
self.__dict__ = self._shared_state
|
|
self.server_id = server_id or "default"
|
|
|
|
def get_client(self):
|
|
return self.client[self.server_id]
|
|
|
|
@classmethod
|
|
def set_loghandler(cls, func=loghandler, level=logging.INFO):
|
|
|
|
for handler in logging.getLogger('Jellyfin').handlers:
|
|
if isinstance(handler, cls.loghandler):
|
|
logging.getLogger('Jellyfin').removeHandler(handler)
|
|
|
|
cls.loghandler = func
|
|
config(level)
|
|
|
|
def close(self):
|
|
|
|
if self.server_id not in self.client:
|
|
return
|
|
|
|
self.client[self.server_id].stop()
|
|
self.client.pop(self.server_id, None)
|
|
|
|
LOG.info("---[ STOPPED JELLYFINCLIENT: %s ]---", self.server_id)
|
|
|
|
@classmethod
|
|
def close_all(cls):
|
|
|
|
for client in cls.client:
|
|
cls.client[client].stop()
|
|
|
|
cls.client = {}
|
|
LOG.info("---[ STOPPED ALL JELLYFINCLIENTS ]---")
|
|
|
|
@classmethod
|
|
def get_active_clients(cls):
|
|
return cls.client
|
|
|
|
@ensure_client()
|
|
def __setattr__(self, name, value):
|
|
|
|
if has_attribute(self, name):
|
|
return super(Jellyfin, self).__setattr__(name, value)
|
|
|
|
setattr(self.client[self.server_id], name, value)
|
|
|
|
@ensure_client()
|
|
def __getattr__(self, name):
|
|
return getattr(self.client[self.server_id], name)
|
|
|
|
def construct(self):
|
|
|
|
self.client[self.server_id] = JellyfinClient()
|
|
|
|
if self.server_id == 'default':
|
|
LOG.info("---[ START JELLYFINCLIENT ]---")
|
|
else:
|
|
LOG.info("---[ START JELLYFINCLIENT: %s ]---", self.server_id)
|
|
|
|
|
|
config()
|