From b2d839f996a6ff00491a41384130a72c95d24118 Mon Sep 17 00:00:00 2001 From: Robin Gohmert Date: Tue, 2 Sep 2025 14:36:50 +0200 Subject: [PATCH] Fixed IncludedNote.content field type (#69) * Fixed IncludedNote.content field type * Fixed formatting and tests --- drafthorse/models/note.py | 13 +++++++------ tests/conftest.py | 4 +--- 2 files changed, 8 insertions(+), 9 deletions(-) diff --git a/drafthorse/models/note.py b/drafthorse/models/note.py index b8df3b3..4e51a1f 100644 --- a/drafthorse/models/note.py +++ b/drafthorse/models/note.py @@ -1,15 +1,16 @@ -from .container import StringContainer from . import BASIC, COMFORT, EXTENDED, NS_RAM from .elements import Element -from .fields import MultiStringField, StringField +from .fields import StringField class IncludedNote(Element): - content_code = StringField(NS_RAM, "ContentCode", required=False, profile=EXTENDED) - content: StringContainer = MultiStringField( - NS_RAM, "Content", required=False, profile=BASIC + content_code: StringField = StringField( + NS_RAM, "ContentCode", required=False, profile=EXTENDED + ) + content: StringField = StringField(NS_RAM, "Content", required=False, profile=BASIC) + subject_code: StringField = StringField( + NS_RAM, "SubjectCode", required=False, profile=COMFORT ) - subject_code = StringField(NS_RAM, "SubjectCode", required=False, profile=COMFORT) class Meta: namespace = NS_RAM diff --git a/tests/conftest.py b/tests/conftest.py index b3d092d..5238a6f 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -22,9 +22,7 @@ def invoice_document(request): doc.header.issue_date_time = date.today() doc.header.languages.add("de") - note = IncludedNote() - note.content.add("Test Node 1") - doc.header.notes.add(note) + doc.header.notes.add(IncludedNote(content="Test Node 1")) doc.trade.agreement.seller.name = "Lieferant GmbH" doc.trade.settlement.payee.name = "Lieferant GmbH" -- 2.47.3