Explorar o código

Delegated JSON generation to json module in json_export

Vsevolod Levitan hai 1 ano
pai
achega
642a8901c9
Modificáronse 1 ficheiros con 3 adicións e 9 borrados
  1. 3 9
      src/export/strategies/json_export.py

+ 3 - 9
src/export/strategies/json_export.py

@@ -1,3 +1,4 @@
+import json
 from src.convert.converter_factory import converter_factory
 from src.export.strategies.export import export
 from src.models.abstract_reference import abstract_reference
@@ -22,16 +23,9 @@ class json_export(export):
             model (abstract_reference): модель, строку с которой нужно создать
         """
 
-        properties = converter_factory.create(model).convert(model)
-        res = "\n        {\n"
+        data = converter_factory.create(model).convert(model)
 
-        for key in properties:
-            val = properties[key]
-            res += f'            "{key}":"{val}",\n'
-
-        res += "        },"
-
-        return res
+        return json.dumps(data, sort_keys=True, indent=4)
 
     def postprocess(self, text):
         """Пост-обработка текста перед экспортом (по надобности)"""