Format - black: operation

1 files changed, 93 insertions, 69 deletions

diff --git a/archaeological_operations/serializers.py b/archaeological_operations/serializers.py
index 91b0b1b0f..5fee27a4e 100644
--- a/archaeological_operations/serializers.py
+++ b/archaeological_operations/serializers.py
@@ -1,13 +1,15 @@
 from django.db.models import Q
 
-from ishtar_common.serializers_utils import generic_get_results, \
-    archive_serialization
+from ishtar_common.serializers_utils import generic_get_results, archive_serialization
 from archaeological_operations import models
 
 
 OPERATION_MODEL_LIST = [
-    models.ArchaeologicalSite, models.Operation, models.RecordRelations,
-    models.Parcel, models.ParcelOwner
+    models.ArchaeologicalSite,
+    models.Operation,
+    models.RecordRelations,
+    models.Parcel,
+    models.ParcelOwner,
 ]
 
 # TODO: administrativ acts, associated documents
@@ -20,17 +22,30 @@ def generate_warehouse_queryset(ids):
 
     for container_key in ("container", "container_ref"):
         for warehouse_key in ("location", "responsible"):
-            q_s = Q(**
-                    {"operations__{}__{}__{}__id__in".format(
-                        base_query_key, container_key, warehouse_key): ids})
-            q_o = Q(**
-                    {"{}__{}__{}__id__in".format(
-                        base_query_key, container_key, warehouse_key): ids})
-            q_r = Q(**
-                    {"left_record__{}__{}__{}__id__in".format(
-                        base_query_key, container_key, warehouse_key): ids,
-                     "right_record__{}__{}__{}__id__in".format(
-                         base_query_key, container_key, warehouse_key): ids})
+            q_s = Q(
+                **{
+                    "operations__{}__{}__{}__id__in".format(
+                        base_query_key, container_key, warehouse_key
+                    ): ids
+                }
+            )
+            q_o = Q(
+                **{
+                    "{}__{}__{}__id__in".format(
+                        base_query_key, container_key, warehouse_key
+                    ): ids
+                }
+            )
+            q_r = Q(
+                **{
+                    "left_record__{}__{}__{}__id__in".format(
+                        base_query_key, container_key, warehouse_key
+                    ): ids,
+                    "right_record__{}__{}__{}__id__in".format(
+                        base_query_key, container_key, warehouse_key
+                    ): ids,
+                }
+            )
         if not q_archaeological_site:
             q_archaeological_site = q_s
             q_operation = q_o
@@ -41,81 +56,83 @@ def generate_warehouse_queryset(ids):
             q_record_relation |= q_r
 
     result_queryset = {
-        models.ArchaeologicalSite.__name__:
-            models.ArchaeologicalSite.objects.filter(
-                q_archaeological_site
-            ),
-        models.Operation.__name__: models.Operation.objects.filter(
-            q_operation),
-        models.RecordRelations.__name__:
-            models.RecordRelations.objects.filter(q_record_relation),
+        models.ArchaeologicalSite.__name__: models.ArchaeologicalSite.objects.filter(
+            q_archaeological_site
+        ),
+        models.Operation.__name__: models.Operation.objects.filter(q_operation),
+        models.RecordRelations.__name__: models.RecordRelations.objects.filter(
+            q_record_relation
+        ),
     }
     return result_queryset
 
 
-def operation_serialization(archive=False, return_empty_types=False,
-                            archive_name=None, operation_queryset=None,
-                            site_queryset=None, cr_queryset=None,
-                            find_queryset=None, warehouse_queryset=None,
-                            get_queryset=False, no_geo=True,
-                            put_locks=False, lock_user=None):
+def operation_serialization(
+    archive=False,
+    return_empty_types=False,
+    archive_name=None,
+    operation_queryset=None,
+    site_queryset=None,
+    cr_queryset=None,
+    find_queryset=None,
+    warehouse_queryset=None,
+    get_queryset=False,
+    no_geo=True,
+    put_locks=False,
+    lock_user=None,
+):
     result_queryset = {}
     if operation_queryset:
         operation_ids = operation_queryset.values_list("id", flat=True)
         result_queryset = {
             models.Operation.__name__: operation_queryset,
-            models.RecordRelations.__name__:
-                models.RecordRelations.objects.filter(
-                    left_record__id__in=operation_ids,
-                    right_record__id__in=operation_ids,
-                ),
-            models.ArchaeologicalSite.__name__:
-                models.ArchaeologicalSite.objects.filter(
-                    operations__id__in=operation_ids
-                    )
+            models.RecordRelations.__name__: models.RecordRelations.objects.filter(
+                left_record__id__in=operation_ids,
+                right_record__id__in=operation_ids,
+            ),
+            models.ArchaeologicalSite.__name__: models.ArchaeologicalSite.objects.filter(
+                operations__id__in=operation_ids
+            ),
         }
     elif site_queryset:
         site_ids = site_queryset.values_list("id", flat=True)
         result_queryset = {
             models.ArchaeologicalSite.__name__: site_queryset,
             models.Operation.__name__: models.Operation.objects.filter(
-                archaeological_sites__id__in=site_ids),
-            models.RecordRelations.__name__:
-                models.RecordRelations.objects.filter(
-                    left_record__archaeological_sites__id__in=site_ids,
-                    right_record__archaeological_sites__id__in=site_ids,
-                ),
+                archaeological_sites__id__in=site_ids
+            ),
+            models.RecordRelations.__name__: models.RecordRelations.objects.filter(
+                left_record__archaeological_sites__id__in=site_ids,
+                right_record__archaeological_sites__id__in=site_ids,
+            ),
         }
     elif cr_queryset:
         cr_ids = cr_queryset.values_list("id", flat=True)
         result_queryset = {
-            models.ArchaeologicalSite.__name__:
-                models.ArchaeologicalSite.objects.filter(
-                    operations__context_record__id__in=cr_ids),
+            models.ArchaeologicalSite.__name__: models.ArchaeologicalSite.objects.filter(
+                operations__context_record__id__in=cr_ids
+            ),
             models.Operation.__name__: models.Operation.objects.filter(
-                context_record__id__in=cr_ids),
-            models.RecordRelations.__name__:
-                models.RecordRelations.objects.filter(
-                    left_record__context_record__id__in=cr_ids,
-                    right_record__context_record__id__in=cr_ids,
-                ),
+                context_record__id__in=cr_ids
+            ),
+            models.RecordRelations.__name__: models.RecordRelations.objects.filter(
+                left_record__context_record__id__in=cr_ids,
+                right_record__context_record__id__in=cr_ids,
+            ),
         }
     elif find_queryset:
         find_ids = find_queryset.values_list("id", flat=True)
         result_queryset = {
-            models.ArchaeologicalSite.__name__:
-                models.ArchaeologicalSite.objects.filter(
-                    operations__context_record__base_finds__find__id__in
-                    =find_ids),
+            models.ArchaeologicalSite.__name__: models.ArchaeologicalSite.objects.filter(
+                operations__context_record__base_finds__find__id__in=find_ids
+            ),
             models.Operation.__name__: models.Operation.objects.filter(
-                context_record__base_finds__find__id__in=find_ids),
-            models.RecordRelations.__name__:
-                models.RecordRelations.objects.filter(
-                    left_record__context_record__base_finds__find__id__in=
-                    find_ids,
-                    right_record__context_record__base_finds__find__id__in=
-                    find_ids,
-                ),
+                context_record__base_finds__find__id__in=find_ids
+            ),
+            models.RecordRelations.__name__: models.RecordRelations.objects.filter(
+                left_record__context_record__base_finds__find__id__in=find_ids,
+                right_record__context_record__base_finds__find__id__in=find_ids,
+            ),
         }
     elif warehouse_queryset:
         warehouse_ids = warehouse_queryset.values_list("id", flat=True)
@@ -123,8 +140,12 @@ def operation_serialization(archive=False, return_empty_types=False,
     if get_queryset:
         return result_queryset
 
-    result = generic_get_results(OPERATION_MODEL_LIST, "operations",
-                                 result_queryset=result_queryset, no_geo=no_geo)
+    result = generic_get_results(
+        OPERATION_MODEL_LIST,
+        "operations",
+        result_queryset=result_queryset,
+        no_geo=no_geo,
+    )
 
     if put_locks:
         for model in OPERATION_MODEL_LIST:
@@ -136,7 +157,10 @@ def operation_serialization(archive=False, return_empty_types=False,
             q.update(locked=True, lock_user=lock_user)
 
     full_archive = archive_serialization(
-        result, archive_dir="operations", archive=archive,
-        return_empty_types=return_empty_types, archive_name=archive_name,
+        result,
+        archive_dir="operations",
+        archive=archive,
+        return_empty_types=return_empty_types,
+        archive_name=archive_name,
     )
     return full_archive