from django.db.models import Q, F

from transfer.serializers.location import CountrySerializer, RegionSerializer, CitySerializer
from transfer.models import Cities
from pprint import pprint


def transfer_countries():
    queryset = Cities.objects.exclude(Q(country_code_2__isnull=True) | Q(country_code_2=""))\
        .values_list("country_code_2", flat=True).distinct()

    serialized_data = CountrySerializer(data=list(queryset.values()), many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        pprint(f"Country serializer errors: {serialized_data.errors}")


def transfer_regions():
    regions_without_subregion_queryset = Cities.objects.\
        exclude(Q(subregion_code__isnull=False) |
                Q(region_code__isnull=True) |
                Q(country_code_2__isnull=True) |
                Q(region_code="") |
                Q(country_code_2="")).\
        values('region_code', 'country_code_2', 'subregion_code').distinct()

    serialized_without_subregion = RegionSerializer(data=list(regions_without_subregion_queryset.values()), many=True)
    if serialized_without_subregion.is_valid():
        serialized_without_subregion.save()
    else:
        pprint(f"Parent regions serializer errors: {serialized_without_subregion.errors}")

    regions_with_subregion_queryset = Cities.objects. \
        exclude(Q(subregion_code__isnull=True) |
                Q(subregion_code="") |
                Q(region_code__isnull=True) |
                Q(country_code_2__isnull=True) |
                Q(region_code="") |
                Q(country_code_2="")). \
        values('region_code', 'country_code_2', 'subregion_code').distinct()

    serialized_with_subregion = RegionSerializer(data=list(regions_with_subregion_queryset.values()), many=True)
    if serialized_with_subregion.is_valid():
        serialized_with_subregion.save()
    else:
        pprint(f"Child regions serializer errors: {serialized_with_subregion.errors}")


def transfer_cities():
    queryset = Cities.objects.exclude(Q(region_code__isnull=True) |
                Q(country_code_2__isnull=True) |
                Q(region_code="") |
                Q(country_code_2="")).only("name", "latitude", "longitude", "subregion_code",
                                           "country_code_2", "zip_code", "is_island")

    serialized_data = CitySerializer(data=list(queryset.values()), many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        pprint(f"City serializer errors: {serialized_data.errors}")


data_types = {
    "dictionaries": [
        transfer_regions,
    ],
    "tmp": [
        transfer_countries,
        # transfer_cities
    ]
}