Browse Source

Made exporters return linked model IDs

Vsevolod Levitan 1 year ago
parent
commit
23e1e069c8
1 changed files with 6 additions and 1 deletions
  1. 6 1
      src/convert/model_converter.py

+ 6 - 1
src/convert/model_converter.py

@@ -1,3 +1,4 @@
+from src.models.abstract_reference import abstract_reference
 from src.convert.basic_converter import basic_converter
 from src.convert.converter import converter
 
@@ -11,5 +12,9 @@ class model_converter(converter):
                 continue
             attr = getattr(obj.__class__, name)
             if isinstance(attr, property):
-                properties[name] = basic_converter.convert(attr.fget(obj))
+                v = attr.fget(obj)
+                if issubclass(v.__class__, abstract_reference):
+                    properties[name] = f"{v.id}"
+                else:
+                    properties[name] = basic_converter.convert(v)
         return properties