Serializers: import action - manage partial export by type of data

1 files changed, 59 insertions, 45 deletions

diff --git a/ishtar_common/serializers.py b/ishtar_common/serializers.py
index e59c1aa3c..66f771e3c 100644
--- a/ishtar_common/serializers.py
+++ b/ishtar_common/serializers.py
@@ -214,51 +214,65 @@ def document_serialization(archive=False, return_empty_types=False,
 def full_serialization(operation_queryset=None, site_queryset=None,
                        cr_queryset=None, find_queryset=None,
                        warehouse_queryset=None, archive=True, no_geo=True,
-                       info=None):
-    # print("type")
-    archive_name = type_serialization(archive=archive, info=info)
-    # print("conf")
-    conf_serialization(archive=archive, archive_name=archive_name)
-    # print("importer")
-    importer_serialization(archive=archive, archive_name=archive_name)
-    # print("geo")
-    geo_serialization(archive=archive, archive_name=archive_name, no_geo=no_geo)
-    # print("directory")
-    directory_serialization(archive=archive, archive_name=archive_name)
-    # print("document")
-    document_serialization(
-        archive=archive,
-        archive_name=archive_name, operation_queryset=operation_queryset,
-        site_queryset=site_queryset, cr_queryset=cr_queryset,
-        find_queryset=find_queryset, warehouse_queryset=warehouse_queryset)
-    # print("operation")
-    operation_serialization(
-        archive=archive,
-        archive_name=archive_name, operation_queryset=operation_queryset,
-        site_queryset=site_queryset, cr_queryset=cr_queryset,
-        find_queryset=find_queryset, warehouse_queryset=warehouse_queryset,
-        no_geo=no_geo)
-    # print("cr")
-    cr_serialization(
-        archive=archive,
-        archive_name=archive_name, operation_queryset=operation_queryset,
-        site_queryset=site_queryset, cr_queryset=cr_queryset,
-        find_queryset=find_queryset, warehouse_queryset=warehouse_queryset,
-        no_geo=no_geo)
-    # print("find")
-    find_serialization(
-        archive=archive,
-        archive_name=archive_name, operation_queryset=operation_queryset,
-        site_queryset=site_queryset, cr_queryset=cr_queryset,
-        find_queryset=find_queryset, warehouse_queryset=warehouse_queryset,
-        no_geo=no_geo)
-    # print("warehouse")
-    warehouse_serialization(
-        archive=archive,
-        archive_name=archive_name, operation_queryset=operation_queryset,
-        site_queryset=site_queryset, cr_queryset=cr_queryset,
-        find_queryset=find_queryset, warehouse_queryset=warehouse_queryset,
-        no_geo=no_geo)
+                       info=None, export_types=True, export_conf=True,
+                       export_importers=True, export_geo=True, export_dir=True,
+                       export_docs=True, export_items=True):
+    archive_name = None
+    if export_types:
+        # print("type")
+        archive_name = type_serialization(archive=archive, info=info)
+    if export_conf:
+        # print("conf")
+        archive_name = conf_serialization(archive=archive,
+                                          archive_name=archive_name)
+    if export_importers:
+        # print("importer")
+        archive_name = importer_serialization(archive=archive,
+                                              archive_name=archive_name)
+    if export_geo:
+        # print("geo")
+        archive_name = geo_serialization(
+            archive=archive, archive_name=archive_name, no_geo=no_geo)
+    if export_dir:
+        # print("directory")
+        archive_name = directory_serialization(archive=archive,
+                                               archive_name=archive_name)
+    if export_docs:
+        # print("document")
+        archive_name = document_serialization(
+            archive=archive, archive_name=archive_name,
+            operation_queryset=operation_queryset, site_queryset=site_queryset,
+            cr_queryset=cr_queryset, find_queryset=find_queryset,
+            warehouse_queryset=warehouse_queryset)
+    if export_items:
+        # print("operation")
+        archive_name = operation_serialization(
+            archive=archive,
+            archive_name=archive_name, operation_queryset=operation_queryset,
+            site_queryset=site_queryset, cr_queryset=cr_queryset,
+            find_queryset=find_queryset, warehouse_queryset=warehouse_queryset,
+            no_geo=no_geo)
+        # print("cr")
+        cr_serialization(
+            archive=archive,
+            archive_name=archive_name, operation_queryset=operation_queryset,
+            site_queryset=site_queryset, cr_queryset=cr_queryset,
+            find_queryset=find_queryset, warehouse_queryset=warehouse_queryset,
+            no_geo=no_geo)
+        # print("find")
+        find_serialization(
+            archive=archive,
+            archive_name=archive_name, operation_queryset=operation_queryset,
+            site_queryset=site_queryset, cr_queryset=cr_queryset,
+            find_queryset=find_queryset, warehouse_queryset=warehouse_queryset,
+            no_geo=no_geo)
+        # print("warehouse")
+        warehouse_serialization(
+            archive=archive,
+            archive_name=archive_name, operation_queryset=operation_queryset,
+            site_queryset=site_queryset, cr_queryset=cr_queryset,
+            find_queryset=find_queryset, warehouse_queryset=warehouse_queryset,
+            no_geo=no_geo)
     return archive_name