Skip to content
Snippets Groups Projects
Commit 0ae7a41d authored by Torben Miny's avatar Torben Miny
Browse files

Merge branch 'feature/aasx_write_xml' into 'master'

adapter.aasx: Allow chosing between XML and JSON serialization when writing AASX packages

See merge request !51
parents 67fb6f51 c3836cc8
No related branches found
No related tags found
1 merge request!51adapter.aasx: Allow chosing between XML and JSON serialization when writing AASX packages
Pipeline #335057 passed
......@@ -30,7 +30,7 @@ import os
import re
from typing import Dict, Tuple, IO, Union, List, Set, Optional
from .xml import read_aas_xml_file
from .xml import read_aas_xml_file, write_aas_xml_file
from .. import model
from .json import read_aas_json_file, write_aas_json_file
import pyecma376_2
......@@ -293,11 +293,11 @@ class AASXWriter:
p.close()
# TODO allow to specify, which supplementary parts (submodels, conceptDescriptions) should be added to the package
# TODO allow to select JSON/XML serialization
def write_aas(self,
aas_id: model.Identifier,
object_store: model.AbstractObjectStore,
file_store: "AbstractSupplementaryFileContainer") -> None:
file_store: "AbstractSupplementaryFileContainer",
write_json: bool = False) -> None:
"""
Add an Asset Administration Shell with all included and referenced objects to the AASX package.
......@@ -312,9 +312,11 @@ class AASXWriter:
Submodels) from
:param file_store: SupplementaryFileContainer to retrieve supplementary files from, which are referenced by File
objects
:param write_json: If True, JSON parts are created for the AAS and each submodel in the AASX package file
instead of XML parts. Defaults to False.
"""
aas_friendly_name = self._aas_name_friendlyfier.get_friendly_name(aas_id)
aas_part_name = "/aasx/{0}/{0}.aas.json".format(aas_friendly_name)
aas_part_name = "/aasx/{0}/{0}.aas.{1}".format(aas_friendly_name, "json" if write_json else "xml")
self._aas_part_names.append(aas_part_name)
aas_friendlyfier = NameFriendlyfier()
......@@ -347,8 +349,11 @@ class AASXWriter:
# Write AAS part
logger.debug("Writing AAS {} to part {} in AASX package ...".format(aas.identification, aas_part_name))
with self.writer.open_part(aas_part_name, "application/json") as p:
with self.writer.open_part(aas_part_name, "application/json" if write_json else "application/xml") as p:
if write_json:
write_aas_json_file(io.TextIOWrapper(p, encoding='utf-8'), objects_to_be_written)
else:
write_aas_xml_file(p, objects_to_be_written)
# Create a AAS split part for each (available) submodel of the AAS
aas_split_part_names: List[str] = []
......@@ -359,8 +364,9 @@ class AASXWriter:
logger.warning("Skipping Submodel, since {} could not be resolved: {}".format(submodel_ref, e))
continue
submodel_friendly_name = aas_friendlyfier.get_friendly_name(submodel.identification)
submodel_part_name = "/aasx/{0}/{1}/{1}.submodel.json".format(aas_friendly_name, submodel_friendly_name)
self._write_submodel_part(file_store, submodel, submodel_part_name)
submodel_part_name = "/aasx/{0}/{1}/{1}.submodel.{2}".format(aas_friendly_name, submodel_friendly_name,
"json" if write_json else "xml")
self._write_submodel_part(file_store, submodel, submodel_part_name, write_json)
aas_split_part_names.append(submodel_part_name)
# Add relationships from AAS part to (submodel) split parts
......@@ -375,7 +381,7 @@ class AASXWriter:
aas_part_name)
def _write_submodel_part(self, file_store: "AbstractSupplementaryFileContainer",
submodel: model.Submodel, submodel_part_name: str) -> None:
submodel: model.Submodel, submodel_part_name: str, write_json: bool = False) -> None:
"""
Helper function for `write_aas()` to write an aas-spec-split part for a Submodel object and add the relevant
supplementary files.
......@@ -383,14 +389,19 @@ class AASXWriter:
:param file_store: The SupplementaryFileContainer to retrieve supplementary files from
:param submodel: The submodel to be written into the AASX package
:param submodel_part_name: OPC part name of the aas-spec-split part for this Submodel
:param write_json: If True, the submodel is written as a JSON file instead of an XML file to the given OPC
part. Defaults to False.
"""
logger.debug("Writing Submodel {} to part {} in AASX package ..."
.format(submodel.identification, submodel_part_name))
submodel_file_objects: model.DictObjectStore[model.Identifiable] = model.DictObjectStore()
submodel_file_objects.add(submodel)
with self.writer.open_part(submodel_part_name, "application/json") as p:
with self.writer.open_part(submodel_part_name, "application/json" if write_json else "application/xml") as p:
if write_json:
write_aas_json_file(io.TextIOWrapper(p, encoding='utf-8'), submodel_file_objects)
else:
write_aas_xml_file(p, submodel_file_objects)
# Write submodel's supplementary files to AASX file
submodel_file_names = []
......
......@@ -73,12 +73,14 @@ class AASXWriterTest(unittest.TestCase):
cp.creator = "PyI40AAS Testing Framework"
# Write AASX file
for write_json in (False, True):
with self.subTest(write_json=write_json):
fd, filename = tempfile.mkstemp(suffix=".aasx")
os.close(fd)
with aasx.AASXWriter(filename) as writer:
writer.write_aas(model.Identifier(id_='https://acplt.org/Test_AssetAdministrationShell',
id_type=model.IdentifierType.IRI),
data, files)
data, files, write_json=write_json)
writer.write_core_properties(cp)
# Read AASX file
......@@ -104,6 +106,7 @@ class AASXWriterTest(unittest.TestCase):
self.assertEqual(new_files.get_content_type("/TestFile.pdf"), "application/pdf")
file_content = io.BytesIO()
new_files.write_file("/TestFile.pdf", file_content)
self.assertEqual(hashlib.sha1(file_content.getvalue()).hexdigest(), "78450a66f59d74c073bf6858db340090ea72a8b1")
self.assertEqual(hashlib.sha1(file_content.getvalue()).hexdigest(),
"78450a66f59d74c073bf6858db340090ea72a8b1")
os.unlink(filename)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment