124 lines
3.0 KiB
Python
124 lines
3.0 KiB
Python
import logging
|
|
|
|
item_builders = {}
|
|
item_loaders = {}
|
|
item_id_generators = {}
|
|
|
|
|
|
def example_builder(**kwargs):
|
|
return BaseItem()
|
|
|
|
|
|
def example_loader(_dict):
|
|
return BaseItem(from_dict=_dict)
|
|
|
|
|
|
def example_id_generator(**kwargs):
|
|
return ""
|
|
|
|
|
|
item_builders['base'] = example_builder
|
|
item_loaders['base'] = example_loader
|
|
item_id_generators['base'] = example_id_generator
|
|
|
|
|
|
def dicts_to_items(music_dicts):
|
|
items = []
|
|
for music_dict in music_dicts:
|
|
type = music_dict['type']
|
|
items.append(item_loaders[type](music_dict))
|
|
return items
|
|
|
|
|
|
def dict_to_item(music_dict):
|
|
type = music_dict['type']
|
|
return item_loaders[type](music_dict)
|
|
|
|
class ValidationFailedError(Exception):
|
|
def __init__(self, msg = None):
|
|
self.msg = msg
|
|
|
|
class PreparationFailedError(Exception):
|
|
def __init__(self, msg = None):
|
|
self.msg = msg
|
|
|
|
class BaseItem:
|
|
def __init__(self, from_dict=None):
|
|
self.log = logging.getLogger("bot")
|
|
self.type = "base"
|
|
self.title = ""
|
|
self.path = ""
|
|
self.tags = []
|
|
self.keywords = ""
|
|
self.duration = 0
|
|
self.version = 0 # if version increase, wrapper will re-save this item
|
|
|
|
if from_dict is None:
|
|
self.id = ""
|
|
self.ready = "pending" # pending - is_valid() -> validated - prepare() -> yes, failed
|
|
else:
|
|
self.id = from_dict['id']
|
|
self.ready = from_dict['ready']
|
|
self.tags = from_dict['tags']
|
|
self.title = from_dict['title']
|
|
self.path = from_dict['path']
|
|
self.keywords = from_dict['keywords']
|
|
self.duration = from_dict['duration']
|
|
|
|
def is_ready(self):
|
|
return True if self.ready == "yes" else False
|
|
|
|
def is_failed(self):
|
|
return True if self.ready == "failed" else False
|
|
|
|
def validate(self):
|
|
raise ValidationFailedError(None)
|
|
|
|
def uri(self):
|
|
raise
|
|
|
|
def prepare(self):
|
|
return True
|
|
|
|
def add_tags(self, tags):
|
|
for tag in tags:
|
|
if tag and tag not in self.tags:
|
|
self.tags.append(tag)
|
|
self.version += 1
|
|
|
|
def remove_tags(self, tags):
|
|
for tag in tags:
|
|
if tag in self.tags:
|
|
self.tags.remove(tag)
|
|
self.version += 1
|
|
|
|
def clear_tags(self):
|
|
if len(self.tags) > 0:
|
|
self.tags = []
|
|
self.version += 1
|
|
|
|
def format_song_string(self, user):
|
|
return self.id
|
|
|
|
def format_current_playing(self, user):
|
|
return self.id
|
|
|
|
def format_title(self):
|
|
return self.title
|
|
|
|
def format_debug_string(self):
|
|
return self.id
|
|
|
|
def display_type(self):
|
|
return ""
|
|
|
|
def to_dict(self):
|
|
return {"type": self.type,
|
|
"id": self.id,
|
|
"ready": self.ready,
|
|
"title": self.title,
|
|
"path": self.path,
|
|
"tags": self.tags,
|
|
"keywords": self.keywords,
|
|
"duration": self.duration}
|