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):
    result = generic_get_results(OPERATION_MODEL_LIST, "operations")
    full_archive = archive_serialization(
        result, archive_dir="operations", archive=archive,
        return_empty_types=return_empty_types, archive_name=archive_name)
    return full_archive