import sys
from pathlib import Path

from extract_un_fil_udpipe import extract_un_fil

# 3232 : idées
# 3476 : cinéma


def parcours(dossier: Path, fichier_xml, fichier_txt, rubrique):
    print(f"on traite {dossier}")
    for sub in sorted(dossier.iterdir()):
        if sub.is_dir():
            parcours(sub, fichier_xml, fichier_txt, rubrique)
        if sub.is_file() and sub.name.endswith(".xml") and rubrique in sub.name  :
            print(f"on appelle extract_un_fil sur {sub}")
            extract_un_fil(sub, fichier_xml, fichier_txt)

def main():
    dossier = Path("data/2021")
    rubrique = sys.argv[1]

    with open(f"data/bao2/{rubrique}/result_py_{rubrique}.xml", "w") as fichier_xml:
        with open(f"data/bao2/{rubrique}/result_py_{rubrique}.txt", "w") as fichier_txt:
            header = "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n<corpus>\n"
            fichier_xml.write(header)
            parcours(dossier, fichier_xml, fichier_txt, rubrique)
            fichier_xml.write("</corpus>\n")

if __name__ == "__main__":
    main()