heckyel/yt-local
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix errors when data directory doesn't exist already

Browse Source
This commit is contained in:
James Taylor 2018-07-31 16:44:21 -07:00
parent c67e22c148
commit eea034d5a9
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
youtube/local_playlist.py
View File

@ -9,6 +9,8 @@ with open('yt_local_playlist_template.html', 'r', encoding='utf-8') as file:
local_playlist_template = Template(file.read()) local_playlist_template = Template(file.read())
def add_to_playlist(name, video_info_list): def add_to_playlist(name, video_info_list):
if not os.path.exists(playlists_directory):
os.makedirs(playlists_directory)
with open(os.path.join(playlists_directory, name + ".txt"), "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: for info in video_info_list:
file.write(info + "\n") file.write(info + "\n")
@ -35,7 +37,11 @@ def get_local_playlist_page(name):
) )
def get_playlist_names(): def get_playlist_names():
for item in os.listdir(playlists_directory): try:
items = os.listdir(playlists_directory)
except FileNotFoundError:
return
for item in items:
name, ext = os.path.splitext(item) name, ext = os.path.splitext(item)
if ext == '.txt': if ext == '.txt':
yield name yield name