#16 Export transpilation traces

Author: alessiostalla

pylasu/emf/metamodel_builder.py

@@ -40,6 +40,9 @@ def provide_class(self, cls: type):
             eclass.eClass.ePackage = self.package
         return eclass
 
+    def generate(self):
+        return self.package
+
 
 def getannotations(cls):
     import inspect

pylasu/emf/model.py

@@ -16,9 +16,10 @@ def find_eclass_in_resource(cls: type, resource: Resource):
 def find_eclass(self: Resource, cls: type):
     eclass = find_eclass_in_resource(cls, self)
     if not eclass:
-        for r in (self.resource_set.resources if self.resource_set else []):
-            if r != self:
-                eclass = find_eclass_in_resource(cls, r)
+        for uri in (self.resource_set.resources if self.resource_set else {}):
+            resource = self.resource_set.resources[uri]
+            if resource != self:
+                eclass = find_eclass_in_resource(cls, resource)
                 if eclass:
                     return eclass
     return eclass

pylasu/playground/transpilation_trace.py

@@ -1,11 +1,12 @@
 from dataclasses import dataclass, field
+from io import BytesIO
 from typing import List
 
-from pyecore.resources import Resource
+from pyecore.resources import Resource, ResourceSet, URI
 
 from pylasu import StrumentaLanguageSupport as starlasu
 from pylasu.emf.model import to_eobject
-from pylasu.playground.transpilation_trace_ecore import TranspilationTrace as ETranspilationTrace
+from pylasu.playground.transpilation_trace_ecore import TranspilationTrace as ETranspilationTrace, JsonResource
 from pylasu.validation.validation import Result, Issue
 
 
@@ -25,3 +26,15 @@ def to_eobject(self, resource: Resource):
             target_result=starlasu.Result(root=to_eobject(self.target_result.root, resource, mappings)),
             generated_code=self.generated_code
         )
+
+    def save_as_json(self, name, *packages):
+        rset = ResourceSet()
+        rset.resource_factory['json'] = JsonResource
+        resource = rset.create_resource(URI(name))
+        for pkg in packages:
+            package_resource = rset.create_resource(URI(pkg.nsURI))
+            package_resource.contents.append(pkg)
+        resource.contents.append(self.to_eobject(resource))
+        with BytesIO() as out:
+            resource.save(out)
+            return out.getvalue().decode('utf-8')

pylasu/playground/transpilation_trace_ecore.py

@@ -48,7 +48,7 @@ def save_as_json(self, name, *packages):
         resource = rset.create_resource(URI(name))
         for pkg in packages:
             package_resource = rset.create_resource(URI(pkg.nsURI))
-            package_resource.contents.add(pkg)
+            package_resource.contents.append(pkg)
         resource.contents.append(self)
         with BytesIO() as out:
             resource.save(out)

tests/test_transpilation_trace.py

@@ -1,15 +1,13 @@
 import json
 import unittest
 
+from pyecore.ecore import EString, EAttribute, EInt
+
 import pylasu.StrumentaLanguageSupport as starlasu
 from pylasu.StrumentaLanguageSupport import ASTNode
 from pylasu.emf.metamodel_builder import MetamodelBuilder
-from pylasu.emf.model import to_eobject
 from pylasu.playground.transpilation_trace import TranspilationTrace
 from pylasu.playground.transpilation_trace_ecore import TranspilationTrace as ETranspilationTrace
-from pyecore.ecore import EString, EAttribute, EInt, MetaEClass
-from pyecore.resources import Resource
-
 from pylasu.validation.validation import Result
 from tests.fixtures import Box
 
@@ -75,14 +73,13 @@ def test_serialize_transpilation_issue(self):
         self.assertEqual(json.loads(expected), json.loads(tt.save_as_json("foo.json")))
 
     def test_serialize_transpilation_from_nodes(self):
-        res = Resource()
-        mmb = MetamodelBuilder("tests.fixtures", "https://strumenta.com/pylasu/test/fixtures", resource=res)
+        mmb = MetamodelBuilder("tests.fixtures", "https://strumenta.com/pylasu/test/fixtures")
         mmb.provide_class(Box)
 
         tt = TranspilationTrace(
             original_code="box(a)[foo, bar]", generated_code='<box name="a"><foo /><bar /></box>',
             source_result=Result(Box("a")),
-            target_result=Result(Box("a"))).to_eobject(res)
+            target_result=Result(Box("a")))
 
         expected = """{
             "eClass": "https://strumenta.com/kolasu/transpilation/v1#//TranspilationTrace",
@@ -91,5 +88,5 @@ def test_serialize_transpilation_from_nodes(self):
             "sourceResult": {"root": {"eClass": "https://strumenta.com/pylasu/test/fixtures#//Box"}},
             "targetResult": {"root": {"eClass": "https://strumenta.com/pylasu/test/fixtures#//Box"}}
         }"""
-        as_json = tt.save_as_json("foo.json")
+        as_json = tt.save_as_json("foo.json", mmb.generate())
         self.assertEqual(json.loads(expected), json.loads(as_json))