fix test test_custom_enum_import / export

04a85e2c · Alexander Schrode · oroulet · 16333392 · 04a85e2c
Commit 04a85e2c authored Feb 27, 2022 by Alexander Schrode Committed by oroulet Feb 28, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

asyncua/common/xmlexporter.py asyncua/common/xmlexporter.py +4 -4

No files found.
--- a/asyncua/common/xmlexporter.py
+++ b/asyncua/common/xmlexporter.py
@@ -297,11 +297,11 @@ class XmlExporter:
            bname = bname.Value.Value
            sdef_el = Et.SubElement(obj_el, 'Definition')
            sdef_el.attrib['Name'] = bname.Name
-            if sdef.StructureType == ua.StructureType.Union:
-                sdef_el.attrib['IsUnion'] = 'true'
-            elif sdef.StructureType == ua.StructureType.StructureWithOptionalFields:
-                sdef_el.attrib['IsOptional'] = 'true'
            if isinstance(sdef, ua.StructureDefinition):
+                if sdef.StructureType == ua.StructureType.Union:
+                    sdef_el.attrib['IsUnion'] = 'true'
+                elif sdef.StructureType == ua.StructureType.StructureWithOptionalFields:
+                    sdef_el.attrib['IsOptional'] = 'true'
                self._structure_fields_to_etree(sdef_el, sdef)
            elif isinstance(sdef, ua.EnumDefinition):
                self._enum_fields_to_etree(sdef_el, sdef)