Extract audio from nine sols wwise

extract_audio.py

from pathlib import Path
import subprocess
import xml.etree.ElementTree as xml

dir = Path(
    "/home/jakob/.local/share/Steam/steamapps/common/Nine Sols/NineSols_Data/StreamingAssets/Audio/GeneratedSoundBanks/Windows"
)
out = Path("output")


soundbanks: list[xml.Element] = xml.parse(dir.joinpath("SoundbanksInfo.xml")).findall(
    "SoundBanks/SoundBank"
)

out.mkdir(exist_ok=True)
for soundbank in soundbanks:
    name = soundbank.findtext("ShortName")
    assert name is not None

    path = soundbank.findtext("Path")

    soundbank_out = out.joinpath(name)
    soundbank_out.mkdir(exist_ok=True)

    subprocess.run(
        [
            "bnk-extract",
            "--oggs-only",
            "--audio",
            dir.joinpath(path),
            "--output",
            soundbank_out,
        ]
    )

    for media in soundbank.findall("Media/File"):
        id = media.attrib["Id"]
        location = media.attrib["Location"]
        media_name = Path(media.findtext("ShortName")).name

        if location == "Memory":
            matches = list(soundbank_out.glob(f"{id}.*"))
            if len(matches) == 0:
                continue
            (file,) = matches
            file.rename(file.with_stem(media_name))
            pass
        else:
            path = media.findtext("Path")
            subprocess.run(
                [
                    "vgmstream-cli",
                    "-o",
                    soundbank_out.joinpath(media_name),
                    dir.joinpath(path),
                ],
                stdout=subprocess.DEVNULL,
            )