m3uプレイリストの自動生成

#!python3
# -*- coding: utf-8 -*-
 
'''
m3u/m3u8ファイルに記述した自動生成ルールをもとにプレイリストを更新するスクリプト
'''
 
mediaExtensions = [
    '.wav',
    '.mp3',
    '.m4a',
    '.ogg',
    '.flac',
]
 
playlistExtensions = [
    '.m3u',
    '.m3u8',
]
 
prefix = '#rule:'
 
import os, sys
import json
import re
import traceback
 
def modifyPath(path):
    path = path.replace('\\', '/')
    return path
 
def hasExtension(name, extensions):
    name = name.lower()
    base, extension = os.path.splitext(name)
    return extension in extensions
 
def isPlaylist(name):
    return hasExtension(name, playlistExtensions)
 
def isMedia(name):
    return hasExtension(name, mediaExtensions)
 
def getRule(header):
    if not header.lower().startswith(prefix):
        return None
    rule = header[len(prefix):].strip()
    if len(rule) == 0:
        return {}
    return json.loads(rule)
 
def getParam(rule, key, defaultParam=None):
    return rule[key] if key in rule else defaultParam
 
def match(texts, patterns):
    if type(texts) is not list:
        texts = [texts]
    if type(patterns) is not list:
        patterns = [patterns]
    for text in texts:
        for pattern in patterns:
            if re.match(pattern, text):
                return True
    return False
 
def getList(basePath, rule):
    includeDirs = getParam(rule, 'includeDir', [])
    excludeDirs = getParam(rule, 'excludeDir', [])
    includeFiles = getParam(rule, 'include', [])
    excludeFiles = getParam(rule, 'exclude', [])
 
    playlist = []
 
    for root, dirs, files in os.walk(basePath):
        dirNames = os.path.relpath(root, basePath)
        dirNames = list(os.path.split(dirNames))
 
        if len(includeDirs) > 0 and not match(dirNames, includeDirs):
            continue
        if len(excludeDirs) > 0 and match(dirNames, excludeDirs):
            continue
 
        for file in files:
            if not isMedia(file):
                continue
            if len(includeFiles) > 0 and not match(file, includeFiles):
                continue
            if len(excludeFiles) > 0 and match(file, excludeFiles):
                continue
            path = os.path.join(root, file)
            path = os.path.relpath(path, basePath)
            path = modifyPath(path)
            playlist.append(path)
 
    return playlist
 
def updatePlaylist(path):
    # ルール取得
    with open(path, encoding='utf_8_sig') as file:
        content = file.read()
        header = content.splitlines()[0]
        rule = getRule(header)
        if rule is None:
            print('No rule: {}'.format(path))
            return
 
    # プレイリスト取得
    print('Process: {}'.format(path))
    print('  Rule: {}'.format(rule))
    root = os.path.dirname(path)
    playlist = getList(root, rule)
 
    # プレイリスト出力
    contents = []
    contents.append(header)
    for record in playlist:
        contents.append(record)
    contents = '\n'.join(contents)
    with open(path, 'w', encoding='utf_8_sig') as file:
        file.write(contents)
 
def main():
    os.chdir(os.path.dirname(__file__))
    paths = []
    for root, dirs, files in os.walk('.'):
        for file in files:
            if not isPlaylist(file):
                continue
            path = os.path.join(root, file)
            path = modifyPath(path)
            paths.append(path)
    for path in paths:
        updatePlaylist(path)
 
if __name__ == '__main__':
    try:
        main()
    except:
        traceback.print_exc()
    input('Press Enter...')