Adrian A. Baumann
b0725fb385
Some checks failed
SonarQube Scan / SonarQube Trigger (push) Failing after 56s
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
from django.core.management.base import BaseCommand
|
|
import xml.etree.ElementTree as ET
|
|
from dokumente.models import Dokument
|
|
from dokumente.utils import build_dokument_xml_element, prettify_xml
|
|
|
|
|
|
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:
|
|
build_dokument_xml_element(dokument, root)
|
|
|
|
xml_str = ET.tostring(root, encoding='unicode', method='xml')
|
|
xml_output = 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)
|