front-end support for adding to playlists

1 files changed, 9 insertions, 3 deletions

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