vgui-cicd/dokumente/management/commands/export_xml.py

from django.core.management.base import BaseCommand
import xml.etree.ElementTree as ET
from datetime import datetime
from dokumente.models import Dokument, Vorgabe, VorgabeKurztext, VorgabeLangtext, Checklistenfrage


class Command(BaseCommand):
    help = 'Export all dokumente as XML'

    def add_arguments(self, parser):
        parser.add_argument(
            '--output',
            type=str,
            help='Output file path (default: stdout)',
        )

    def handle(self, *args, **options):
        dokumente = Dokument.objects.filter(aktiv=True).prefetch_related(
            'autoren', 'pruefende', 'vorgaben__thema', 
            'vorgaben__referenzen', 'vorgaben__stichworte',
            'vorgaben__checklistenfragen', 'vorgaben__vorgabekurztext_set',
            'vorgaben__vorgabelangtext_set', 'geltungsbereich_set',
            'einleitung_set', 'changelog__autoren'
        ).order_by('nummer')

        root = ET.Element('Vorgabendokumente')

        for dokument in dokumente:
            doc_element = ET.SubElement(root, 'Vorgabendokument')
            
            ET.SubElement(doc_element, 'Typ').text = dokument.dokumententyp.name if dokument.dokumententyp else ""
            ET.SubElement(doc_element, 'Nummer').text = dokument.nummer
            ET.SubElement(doc_element, 'Name').text = dokument.name
            
            autoren_element = ET.SubElement(doc_element, 'Autoren')
            for autor in dokument.autoren.all():
                ET.SubElement(autoren_element, 'Autor').text = autor.name
            
            pruefende_element = ET.SubElement(doc_element, 'Pruefende')
            for pruefender in dokument.pruefende.all():
                ET.SubElement(pruefende_element, 'Pruefender').text = pruefender.name
            
            gueltigkeit_element = ET.SubElement(doc_element, 'Gueltigkeit')
            ET.SubElement(gueltigkeit_element, 'Von').text = dokument.gueltigkeit_von.strftime("%Y-%m-%d") if dokument.gueltigkeit_von else ""
            ET.SubElement(gueltigkeit_element, 'Bis').text = dokument.gueltigkeit_bis.strftime("%Y-%m-%d") if dokument.gueltigkeit_bis else None
            
            ET.SubElement(doc_element, 'SignaturCSO').text = dokument.signatur_cso
            
            geltungsbereich_sections = dokument.geltungsbereich_set.all().order_by('order')
            if geltungsbereich_sections:
                geltungsbereich_element = ET.SubElement(doc_element, 'Geltungsbereich')
                abschnitt_element = ET.SubElement(geltungsbereich_element, 'Abschnitt')
                for gb in geltungsbereich_sections:
                    section = ET.SubElement(abschnitt_element, 'Teil')
                    section.set('typ', gb.abschnitttyp.abschnitttyp if gb.abschnitttyp else "text")
                    section.text = gb.inhalt
            
            einleitung_sections = dokument.einleitung_set.all().order_by('order')
            if einleitung_sections:
                einleitung_element = ET.SubElement(doc_element, 'Einleitung')
                abschnitt_element = ET.SubElement(einleitung_element, 'Abschnitt')
                for ei in einleitung_sections:
                    section = ET.SubElement(abschnitt_element, 'Teil')
                    section.set('typ', ei.abschnitttyp.abschnitttyp if ei.abschnitttyp else "text")
                    section.text = ei.inhalt
            
            ET.SubElement(doc_element, 'Ziel').text = ""
            ET.SubElement(doc_element, 'Grundlagen').text = ""
            
            changelog_element = ET.SubElement(doc_element, 'Changelog')
            for cl in dokument.changelog.all().order_by('-datum'):
                entry = ET.SubElement(changelog_element, 'Eintrag')
                ET.SubElement(entry, 'Datum').text = cl.datum.strftime("%Y-%m-%d")
                autoren = ET.SubElement(entry, 'Autoren')
                for autor in cl.autoren.all():
                    ET.SubElement(autoren, 'Autor').text = autor.name
                ET.SubElement(entry, 'Aenderung').text = cl.aenderung
            
            anhaenge_element = ET.SubElement(doc_element, 'Anhaenge')
            ET.SubElement(anhaenge_element, 'Anhang').text = dokument.anhaenge
            
            ET.SubElement(doc_element, 'Verantwortlich').text = "Information Security Management BIT"
            ET.SubElement(doc_element, 'Klassifizierung').text = ""
            
            glossar_element = ET.SubElement(doc_element, 'Glossar')
            
            vorgaben_element = ET.SubElement(doc_element, 'Vorgaben')
            
            for vorgabe in dokument.vorgaben.all().order_by('order'):
                vorgabe_el = ET.SubElement(vorgaben_element, 'Vorgabe')
                
                ET.SubElement(vorgabe_el, 'Nummer').text = str(vorgabe.nummer)
                ET.SubElement(vorgabe_el, 'Titel').text = vorgabe.titel
                ET.SubElement(vorgabe_el, 'Thema').text = vorgabe.thema.name if vorgabe.thema else ""
                
                kurztext_sections = vorgabe.vorgabekurztext_set.all().order_by('order')
                if kurztext_sections:
                    kurztext_element = ET.SubElement(vorgabe_el, 'Kurztext')
                    abschnitt = ET.SubElement(kurztext_element, 'Abschnitt')
                    for kt in kurztext_sections:
                        teil = ET.SubElement(abschnitt, 'Teil')
                        teil.set('typ', kt.abschnitttyp.abschnitttyp if kt.abschnitttyp else "text")
                        teil.text = kt.inhalt
                
                langtext_sections = vorgabe.vorgabelangtext_set.all().order_by('order')
                if langtext_sections:
                    langtext_element = ET.SubElement(vorgabe_el, 'Langtext')
                    abschnitt = ET.SubElement(langtext_element, 'Abschnitt')
                    for lt in langtext_sections:
                        teil = ET.SubElement(abschnitt, 'Teil')
                        teil.set('typ', lt.abschnitttyp.abschnitttyp if lt.abschnitttyp else "text")
                        teil.text = lt.inhalt
                
                referenz_element = ET.SubElement(vorgabe_el, 'Referenzen')
                for ref in vorgabe.referenzen.all():
                    ref_text = f"{ref.name_nummer}: {ref.name_text}" if ref.name_text else ref.name_nummer
                    ET.SubElement(referenz_element, 'Referenz').text = ref_text
                
                vorgabe_gueltigkeit = ET.SubElement(vorgabe_el, 'Gueltigkeit')
                ET.SubElement(vorgabe_gueltigkeit, 'Von').text = vorgabe.gueltigkeit_von.strftime("%Y-%m-%d") if vorgabe.gueltigkeit_von else ""
                ET.SubElement(vorgabe_gueltigkeit, 'Bis').text = vorgabe.gueltigkeit_bis.strftime("%Y-%m-%d") if vorgabe.gueltigkeit_bis else None
                
                checklistenfragen_element = ET.SubElement(vorgabe_el, 'Checklistenfragen')
                for cf in vorgabe.checklistenfragen.all():
                    ET.SubElement(checklistenfragen_element, 'Frage').text = cf.frage
                
                stichworte_element = ET.SubElement(vorgabe_el, 'Stichworte')
                for stw in vorgabe.stichworte.all():
                    ET.SubElement(stichworte_element, 'Stichwort').text = stw.stichwort

        xml_str = ET.tostring(root, encoding='unicode', method='xml')
        xml_output = self._prettify_xml(xml_str)
        
        if options['output']:
            with open(options['output'], 'w', encoding='utf-8') as f:
                f.write(xml_output)
            self.stdout.write(self.style.SUCCESS(f'XML exported to {options["output"]}'))
        else:
            self.stdout.write(xml_output)

    def _prettify_xml(self, xml_string):
        import xml.dom.minidom
        dom = xml.dom.minidom.parseString(xml_string)
        return dom.toprettyxml(indent="  ", encoding="UTF-8").decode('utf-8')