diff --git a/main.py b/main.py index e465c79..118679e 100644 --- a/main.py +++ b/main.py @@ -38,9 +38,14 @@ def run_pls_command(playlist_id): return "" +def sanitize_for_filename(name): + """Sanitizes a string to be safe as a filename component.""" + return name.replace('/', '_').replace('\\', '_') + + def save_playlist_file(playlist_dir, playlist_name, content): """Saves the transformed playlist content to a file.""" - filename = f"{playlist_name}.m3u8" + filename = f"{sanitize_for_filename(playlist_name)}.m3u8" filepath = os.path.join(playlist_dir, filename) try: with open(filepath, 'w', encoding='utf-8') as f: @@ -52,7 +57,7 @@ def save_playlist_file(playlist_dir, playlist_name, content): def delete_playlist_file(playlist_dir, playlist_name): """Deletes a playlist file.""" - filename = f"{playlist_name}.m3u8" + filename = f"{sanitize_for_filename(playlist_name)}.m3u8" filepath = os.path.join(playlist_dir, filename) if os.path.exists(filepath): try: