81 lines
3.2 KiB
Python
81 lines
3.2 KiB
Python
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
|
|
]
|
|
}
|