Create a simple .m3u file from a folder heirarchy

create_playlist.py

import os
import sys
import pathlib

if len(sys.argv) < 2:
    print("Error - pass the folder(s) you want to index as argument(s)")
    exit(1)

directories = sys.argv[1:]

for directory in directories:
    playlist = os.path.join(directory, 'playlist.m3u')

    print("Using directory", directory)

    with open(playlist, "w") as f:
        for path in pathlib.Path(directory).rglob('*.*'):
            relative_path = os.path.relpath(path, directory)

            if not path.suffix in ['.mkv', '.mp4']:
                print("Skipping", relative_path)
                continue

            f.write(relative_path.replace('\\', '/') + '\n')

    print("Playlist written", playlist)