diff options
Diffstat (limited to 'youtube')
| -rw-r--r-- | youtube/common.py | 9 | ||||
| -rw-r--r-- | youtube/local_playlist.py | 34 | ||||
| -rw-r--r-- | youtube/shared.css | 2 | ||||
| -rw-r--r-- | youtube/youtube.py | 16 |
4 files changed, 30 insertions, 31 deletions
diff --git a/youtube/common.py b/youtube/common.py index e4ab39a..46dfc8d 100644 --- a/youtube/common.py +++ b/youtube/common.py @@ -82,7 +82,7 @@ medium_video_item_template = Template(''' <span class="description">$description</span> <span class="badges">$badges</span> </div> - <input class="item-checkbox" type="checkbox" name="video_info_list" value="$video_info" form="playlist-add"> + <input class="item-checkbox" type="checkbox" name="video_info_list" value="$video_info" form="playlist-edit"> </div> ''') @@ -99,7 +99,7 @@ small_video_item_template = Template(''' <span class="views">$views</span> </div> - <input class="item-checkbox" type="checkbox" name="video_info_list" value="$video_info" form="playlist-add"> + <input class="item-checkbox" type="checkbox" name="video_info_list" value="$video_info" form="playlist-edit"> </div> ''') @@ -298,13 +298,12 @@ header_template = Template(''' </form> </div> <div id="header-right"> - <form id="playlist-add" action="/youtube.com/edit_playlist" method="post" target="_self"> - <input type="hidden" name="action" value="add"> + <form id="playlist-edit" action="/youtube.com/edit_playlist" method="post" target="_self"> <input name="playlist_name" id="playlist-name-selection" list="playlist-options" type="text"> <datalist id="playlist-options"> $playlists </datalist> - <button type="submit" id="playlist-add-button">Add to playlist</button> + <button type="submit" id="playlist-add-button" name="action" value="add">Add to playlist</button> <button type="reset" id="item-selection-reset">Clear selection</button> </form> <a href="/youtube.com/playlists" id="local-playlists">Local playlists</a> diff --git a/youtube/local_playlist.py b/youtube/local_playlist.py index 926c93f..5da3b4d 100644 --- a/youtube/local_playlist.py +++ b/youtube/local_playlist.py @@ -40,30 +40,18 @@ def get_playlist_names(): if ext == '.txt': yield name -''' - main{ - display:grid; - grid-template-columns: 3fr 1fr; - } - - header{ - grid-template-columns: 3fr 1fr; - } - - #left{ - grid-column: 1; - grid-row: 1; - - display: grid; - grid-template-columns: 1fr 800px; - grid-template-rows: 0fr 1fr 0fr; - } +def remove_from_playlist(name, video_info_list): + ids = [json.loads(video)['id'] for video in video_info_list] + with open(os.path.join(playlists_directory, name + ".txt"), 'r', encoding='utf-8') as file: + videos = file.read() + videos_in = videos.splitlines() + videos_out = [] + for video in videos_in: + if json.loads(video)['id'] not in ids: + videos_out.append(video) + with open(os.path.join(playlists_directory, name + ".txt"), 'w', encoding='utf-8') as file: + file.write("\n".join(videos_out) + "\n") - #right{ - grid-column: 2; - grid-row: 1; - } -''' def get_playlists_list_page(): page = '''<ul>\n''' list_item_template = Template(''' <li><a href="$url">$name</a></li>\n''') diff --git a/youtube/shared.css b/youtube/shared.css index dc1d121..670a1b3 100644 --- a/youtube/shared.css +++ b/youtube/shared.css @@ -74,7 +74,7 @@ address{ grid-template-rows: 1fr; width: 550px; } - #playlist-add{ + #playlist-edit{ display:contents; } #playlist-name-selection{ diff --git a/youtube/youtube.py b/youtube/youtube.py index aa23d06..a9bd3c4 100644 --- a/youtube/youtube.py +++ b/youtube/youtube.py @@ -57,8 +57,20 @@ def youtube(env, start_response): fields = urllib.parse.parse_qs(env['wsgi.input'].read().decode()) if fields['action'][0] == 'add': local_playlist.add_to_playlist(fields['playlist_name'][0], fields['video_info_list']) - - start_response('204 No Content', ()) + start_response('204 No Content', ()) + elif fields['action'][0] == 'remove': + try: + playlist_name = fields['playlist_page'][0] + except KeyError: + playlist_name = fields['playlist_name'][0] + local_playlist.remove_from_playlist(playlist_name, fields['video_info_list']) + start_response('200 OK', ()) + return local_playlist.get_playlist_page(playlist_name).encode() + else: + start_response('400 Bad Request', ()) + return b'400 Bad Request' + + else: start_response('404 Not Found', ()) return b'404 Not Found' |