from ishtar_common.serializers import generic_get_results, archive_serialization
from archaeological_operations import models


OPERATION_MODEL_LIST = [
    models.ArchaeologicalSite, models.Operation, models.Parcel,
    models.ParcelOwner
]


def operation_serialization(archive=False, return_empty_types=False,
                            archive_name=None, operation_queryset=None,
                            site_queryset=None):
    result_queryset = {}
    if operation_queryset:
        result_queryset = {
            models.Operation.__name__: operation_queryset,
            models.ArchaeologicalSite.__name__:
                models.ArchaeologicalSite.objects.filter(
                    operations__id__in=operation_queryset.values_list(
                        "id", flat=True)
                    )
        }
    elif site_queryset:
        result_queryset = {
            models.ArchaeologicalSite.__name__: site_queryset,
            models.Operation.__name__: models.Operation.objects.filter(
                archaeological_sites__id__in=site_queryset.values_list(
                    "id", flat=True))
        }

    result = generic_get_results(OPERATION_MODEL_LIST, "operations",
                                 result_queryset=result_queryset)
    full_archive = archive_serialization(
        result, archive_dir="operations", archive=archive,
        return_empty_types=return_empty_types, archive_name=archive_name,
    )
    return full_archive