pepupからダウンロードした医療費のxmlをcsvに変換するスクリプト

pepup.py

import xml.etree.ElementTree as ET
import csv

tree = ET.parse('pepup_download.xml')
root = tree.getroot()

csv_data = []
for child in root:
    if child.tag != '{http://xml.e-tax.nta.go.jp/XSD/kyotsu}WBD00000':
        continue

    for element in child:
        if element.tag != "{http://xml.e-tax.nta.go.jp/XSD/kyotsu}WBD00010":
            continue

        row = []
        for item in element:
            if item.tag == "{http://xml.e-tax.nta.go.jp/XSD/kyotsu}WBD00030":
                y = item.find("{http://xml.e-tax.nta.go.jp/XSD/general}yyyy").text
                m = item.find("{http://xml.e-tax.nta.go.jp/XSD/general}mm").text
                row.append(f"{y}/{int(m):02}")
            else:
                row.append(item.text)
        csv_data.append(row)

with open('output.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerows(csv_data)