REFACTOR: MUSIC LIBRARYgit status #91

[botaradio.git] / media / url_from_playlist.py

diff --git a/media/url_from_playlist.py b/media/url_from_playlist.py
index 0cac0b3cddfb9f39dfd0370a695c38ce306a0287..11e101c99487c92370f364ec0190218eaf3ff6b1 100644 (file)
--- a/media/url_from_playlist.py
+++ b/media/url_from_playlist.py
@@ -2,7 +2,9 @@ import youtube_dl
 import constants
 import media
 import variables as var
-from media.url import URLItem
+import hashlib
+from media.item import item_builders, item_loaders, item_id_generators
+from media.url import URLItem, url_item_id_generator
 
 def get_playlist_info(bot, url, start_index=0, user=""):
     items = []
@@ -48,6 +50,23 @@ def get_playlist_info(bot, url, start_index=0, user=""):
 
     return items
 
+
+def playlist_url_item_builder(bot, **kwargs):
+    return PlaylistURLItem(bot,
+                           kwargs['url'],
+                           kwargs['title'],
+                           kwargs['playlist_url'],
+                           kwargs['playlist_title'])
+
+
+def playlist_url_item_loader(bot, _dict):
+    return PlaylistURLItem(bot, "", "", "", "", _dict)
+
+item_builders['url_from_playlist'] = playlist_url_item_builder
+item_loaders['url_from_playlist'] = playlist_url_item_loader
+item_id_generators['url_from_playlist'] = url_item_id_generator
+
+
 class PlaylistURLItem(URLItem):
     def __init__(self, bot, url, title, playlist_url, playlist_title, from_dict=None):
         if from_dict is None: