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


WAREHOUSE_MODEL_LIST = [
    models.Warehouse, models.Container,
    models.WarehouseDivisionLink,
    models.ContainerLocalisation
]

# TODO: associated documents


def warehouse_serialization(archive=False, return_empty_types=False,
                            archive_name=None, operation_queryset=None,
                            site_queryset=None, cr_queryset=None,
                            find_queryset=None):
    result_queryset = {}
    """
    if operation_queryset:
        operation_ids = operation_queryset.values_list("id", flat=True)
        result_queryset = {
            models.BaseFind.__name__:
                models.BaseFind.objects.filter(
                    context_record__operation_id__in=operation_ids),
            models.Find.__name__:
                models.Find.objects.filter(
                    base_finds__context_record__operation_id__in=operation_ids),
        }
    elif site_queryset:
        sites = site_queryset.values_list("id", flat=True)
        f_q = {
            "base_finds__context_record__operation__archaeological_site_id__in": sites
        }
        result_queryset = {
            models.BaseFind.__name__:
                models.BaseFind.objects.filter(
                    context_record__operation__archaeological_site_id__in=sites
                ),
            models.Find.__name__:
                models.Find.objects.filter(**f_q),
        }
    elif cr_queryset:
        cr_ids = cr_queryset.values_list("id", flat=True)
        result_queryset = {
            models.BaseFind.__name__:
                models.BaseFind.objects.filter(
                    context_record__in=cr_ids),
            models.Find.__name__:
                models.Find.objects.filter(
                    base_finds__context_record__in=cr_ids),
        }
    elif find_queryset:
        find_ids = find_queryset.values_list("id", flat=True)
        result_queryset = {
            models.BaseFind.__name__: models.BaseFind.objects.filter(
                find__id__in=find_ids),
            models.Find.__name__: find_queryset,
        }
    """
    result = generic_get_results(WAREHOUSE_MODEL_LIST, "warehouse",
                                 result_queryset=result_queryset)
    full_archive = archive_serialization(
        result, archive_dir="warehouse", archive=archive,
        return_empty_types=return_empty_types, archive_name=archive_name,
    )
    return full_archive