From 26620cbac85f1fa5057e54fa8d91c35d907c0e0f Mon Sep 17 00:00:00 2001 From: James Taylor Date: Sun, 8 Jul 2018 02:56:54 -0700 Subject: front-end support for adding to playlists --- youtube/local_playlist.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'youtube/local_playlist.py') diff --git a/youtube/local_playlist.py b/youtube/local_playlist.py index 057555a..5a4182b 100644 --- a/youtube/local_playlist.py +++ b/youtube/local_playlist.py @@ -1,13 +1,19 @@ -import os.path +import os import json playlists_directory = os.path.normpath("data/playlists") def add_to_playlist(name, video_info_list): - with open(os.path.join(playlists_directory, name), "a", encoding='utf-8') as file: + with open(os.path.join(playlists_directory, name + ".txt"), "a", encoding='utf-8') as file: for info in video_info_list: file.write(info + "\n") def get_playlist_page(name): - pass \ No newline at end of file + pass + +def get_playlist_names(): + for item in os.listdir(playlists_directory): + name, ext = os.path.splitext(item) + if ext == '.txt': + yield name \ No newline at end of file -- cgit v1.2.3