#!/usr/bin/env python

"""
Extracts description from datalist
"""

import os
import sys
import xml.etree.ElementTree as ET

tree = ET.parse(rf"{sys.argv[1]}")
root = tree.getroot()

for game in root.findall("game"):
    romName = game.find("rom").get("name")
    print(romName)
    textFile = os.path.splitext(f"{romName}")[0] + ".txt"
    os.makedirs("text", exist_ok=True)

    if game.find("description") is not None:
        with open(f"text/{textFile}", "w", encoding="utf-8") as f:
            f.write(game.find("description").text)