From 9e6072526c0200e9c1d1723198f950a0039b9a2d Mon Sep 17 00:00:00 2001 From: Terry Geng Date: Sat, 7 Mar 2020 22:38:25 +0800 Subject: [PATCH] fix: web interface add url issue #91 --- interface.py | 4 ++-- media/playlist.py | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/interface.py b/interface.py index bbda09d..1cd43aa 100644 --- a/interface.py +++ b/interface.py @@ -189,7 +189,7 @@ def post(): elif 'add_url' in request.form: - music_wrapper = get_item_wrapper(var.bot, type='url', url=request.form['url']) + music_wrapper = get_item_wrapper(var.bot, type='url', url=request.form['url'], user=user) var.playlist.append(music_wrapper) log.info("web: add to playlist: " + music_wrapper.format_debug_string()) @@ -199,7 +199,7 @@ def post(): elif 'add_radio' in request.form: url = request.form['add_radio'] - music_wrapper = get_item_wrapper(var.bot, type='radio', url=url) + music_wrapper = get_item_wrapper(var.bot, type='radio', url=url, user=user) var.playlist.append(music_wrapper) log.info("cmd: add to playlist: " + music_wrapper.format_debug_string()) diff --git a/media/playlist.py b/media/playlist.py index 219a6c9..088dc58 100644 --- a/media/playlist.py +++ b/media/playlist.py @@ -91,6 +91,8 @@ class PlaylistItemWrapper: def get_item_wrapper(bot, **kwargs): item = var.library.get_item(bot, **kwargs) + if 'user' not in kwargs: + raise KeyError("Which user added this song?") return PlaylistItemWrapper(var.library, item.id, kwargs['type'], kwargs['user']) def get_item_wrapper_by_id(bot, id, user): -- 2.39.2