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')