XML export added

dokumente/management/commands/export_xml.py

@@ -0,0 +1,144 @@
+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')