from django.db.models import Q, QuerySet

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="")) \
    #     .only("id", "country_code_2").annotate(Count("country_code_2")).order_by("country_code_2")
    queryset = Cities.objects.raw("SELECT cities.id, cities.country_code_2 "
                                  "FROM cities "
                                  "WHERE NOT ((cities.country_code_2 IS NULL "
                                  "OR (cities.country_code_2 = '' "
                                  "AND cities.country_code_2 IS NOT NULL))) "
                                  "GROUP BY cities.country_code_2")

    queryset = [vars(query) for query in queryset]

    serialized_data = CountrySerializer(data=queryset, 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('id', '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("id", "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": [
    ],
    "tmp": [
        transfer_countries,
        # transfer_regions,
        # transfer_cities
    ]
}