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 def _parse_markdown_table(markdown_content): """ Parse markdown table content and return XML element with
structure """ lines = [line.strip() for line in markdown_content.strip().split('\n') if line.strip()] if not lines: return None # Create table element table = ET.Element('table') # Parse first row as header header_row = [cell.strip() for cell in lines[0].split('|') if cell.strip()] header = ET.SubElement(table, 'header') for cell in header_row: column = ET.SubElement(header, 'column') column.text = cell # Parse remaining rows (skip separator row if it exists) for line in lines[2:] if len(lines) > 1 and all(c in '-| ' for c in lines[1]) else lines[1:]: # Check if this is a separator row if all(c in '-| ' for c in line): continue row = ET.SubElement(table, 'row') row_cells = [cell.strip() for cell in line.split('|') if cell.strip()] for cell in row_cells: column = ET.SubElement(row, 'column') column.text = cell return table 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') for gb in geltungsbereich_sections: section_type = gb.abschnitttyp.abschnitttyp if gb.abschnitttyp else "text" if section_type in ('tabelle', 'table'): table = _parse_markdown_table(gb.inhalt) if table is not None: abschnitt_element = ET.SubElement(geltungsbereich_element, 'Abschnitt') abschnitt_element.set('typ', section_type) abschnitt_element.append(table) else: abschnitt_element = ET.SubElement(geltungsbereich_element, 'Abschnitt') abschnitt_element.set('typ', section_type) abschnitt_element.text = gb.inhalt einleitung_sections = dokument.einleitung_set.all().order_by('order') if einleitung_sections: einleitung_element = ET.SubElement(doc_element, 'Einleitung') for ei in einleitung_sections: section_type = ei.abschnitttyp.abschnitttyp if ei.abschnitttyp else "text" if section_type in ('tabelle', 'table'): table = _parse_markdown_table(ei.inhalt) if table is not None: abschnitt_element = ET.SubElement(einleitung_element, 'Abschnitt') abschnitt_element.set('typ', section_type) abschnitt_element.append(table) else: abschnitt_element = ET.SubElement(einleitung_element, 'Abschnitt') abschnitt_element.set('typ', section_type) abschnitt_element.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') for kt in kurztext_sections: section_type = kt.abschnitttyp.abschnitttyp if kt.abschnitttyp else "text" if section_type in ('tabelle', 'table'): table = _parse_markdown_table(kt.inhalt) if table is not None: abschnitt = ET.SubElement(kurztext_element, 'Abschnitt') abschnitt.set('typ', section_type) abschnitt.append(table) else: abschnitt = ET.SubElement(kurztext_element, 'Abschnitt') abschnitt.set('typ', section_type) abschnitt.text = kt.inhalt langtext_sections = vorgabe.vorgabelangtext_set.all().order_by('order') if langtext_sections: langtext_element = ET.SubElement(vorgabe_el, 'Langtext') for lt in langtext_sections: section_type = lt.abschnitttyp.abschnitttyp if lt.abschnitttyp else "text" if section_type in ('tabelle', 'table'): table = _parse_markdown_table(lt.inhalt) if table is not None: abschnitt = ET.SubElement(langtext_element, 'Abschnitt') abschnitt.set('typ', section_type) abschnitt.append(table) else: abschnitt = ET.SubElement(langtext_element, 'Abschnitt') abschnitt.set('typ', section_type) abschnitt.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')